linting fixes on the markov models
This commit is contained in:
parent
1cf0364268
commit
8b3caabb57
|
@ -1,30 +1,22 @@
|
||||||
"""
|
"""Save brain pieces as markov chains for chaining."""
|
||||||
markov/models.py --- save brain pieces for chaining
|
|
||||||
|
|
||||||
"""
|
|
||||||
|
|
||||||
import logging
|
import logging
|
||||||
|
|
||||||
from django.db import models
|
from django.db import models
|
||||||
|
|
||||||
|
log = logging.getLogger(__name__)
|
||||||
log = logging.getLogger('markov.models')
|
|
||||||
|
|
||||||
|
|
||||||
class MarkovContext(models.Model):
|
class MarkovContext(models.Model):
|
||||||
|
|
||||||
"""Define contexts for Markov chains."""
|
"""Define contexts for Markov chains."""
|
||||||
|
|
||||||
name = models.CharField(max_length=200, unique=True)
|
name = models.CharField(max_length=200, unique=True)
|
||||||
|
|
||||||
def __str__(self):
|
def __str__(self):
|
||||||
"""String representation."""
|
"""Provide string representation."""
|
||||||
|
|
||||||
return "{0:s}".format(self.name)
|
return "{0:s}".format(self.name)
|
||||||
|
|
||||||
|
|
||||||
class MarkovTarget(models.Model):
|
class MarkovTarget(models.Model):
|
||||||
|
|
||||||
"""Define IRC targets that relate to a context, and can occasionally be talked to."""
|
"""Define IRC targets that relate to a context, and can occasionally be talked to."""
|
||||||
|
|
||||||
name = models.CharField(max_length=200, unique=True)
|
name = models.CharField(max_length=200, unique=True)
|
||||||
|
@ -33,13 +25,11 @@ class MarkovTarget(models.Model):
|
||||||
chatter_chance = models.IntegerField(default=0)
|
chatter_chance = models.IntegerField(default=0)
|
||||||
|
|
||||||
def __str__(self):
|
def __str__(self):
|
||||||
"""String representation."""
|
"""Provide string representation."""
|
||||||
|
|
||||||
return "{0:s} -> {1:s}".format(self.name, self.context.name)
|
return "{0:s} -> {1:s}".format(self.name, self.context.name)
|
||||||
|
|
||||||
|
|
||||||
class MarkovState(models.Model):
|
class MarkovState(models.Model):
|
||||||
|
|
||||||
"""One element in a Markov chain, some text or something."""
|
"""One element in a Markov chain, some text or something."""
|
||||||
|
|
||||||
_start1 = '__start1'
|
_start1 = '__start1'
|
||||||
|
@ -54,6 +44,8 @@ class MarkovState(models.Model):
|
||||||
context = models.ForeignKey(MarkovContext, on_delete=models.CASCADE)
|
context = models.ForeignKey(MarkovContext, on_delete=models.CASCADE)
|
||||||
|
|
||||||
class Meta:
|
class Meta:
|
||||||
|
"""Options for the model itself."""
|
||||||
|
|
||||||
index_together = [
|
index_together = [
|
||||||
['context', 'k1', 'k2'],
|
['context', 'k1', 'k2'],
|
||||||
['context', 'v'],
|
['context', 'v'],
|
||||||
|
@ -65,6 +57,5 @@ class MarkovState(models.Model):
|
||||||
unique_together = ('context', 'k1', 'k2', 'v')
|
unique_together = ('context', 'k1', 'k2', 'v')
|
||||||
|
|
||||||
def __str__(self):
|
def __str__(self):
|
||||||
"""String representation."""
|
"""Provide string representation."""
|
||||||
|
|
||||||
return "{0:s},{1:s} -> {2:s} (count: {3:d})".format(self.k1, self.k2, self.v, self.count)
|
return "{0:s},{1:s} -> {2:s} (count: {3:d})".format(self.k1, self.k2, self.v, self.count)
|
||||||
|
|
Loading…
Reference in New Issue