diff --git a/markov/models.py b/markov/models.py index b23e150..9759889 100644 --- a/markov/models.py +++ b/markov/models.py @@ -1,30 +1,22 @@ -""" -markov/models.py --- save brain pieces for chaining - -""" - +"""Save brain pieces as markov chains for chaining.""" import logging from django.db import models - -log = logging.getLogger('markov.models') +log = logging.getLogger(__name__) class MarkovContext(models.Model): - """Define contexts for Markov chains.""" name = models.CharField(max_length=200, unique=True) def __str__(self): - """String representation.""" - + """Provide string representation.""" return "{0:s}".format(self.name) class MarkovTarget(models.Model): - """Define IRC targets that relate to a context, and can occasionally be talked to.""" name = models.CharField(max_length=200, unique=True) @@ -33,13 +25,11 @@ class MarkovTarget(models.Model): chatter_chance = models.IntegerField(default=0) def __str__(self): - """String representation.""" - + """Provide string representation.""" return "{0:s} -> {1:s}".format(self.name, self.context.name) class MarkovState(models.Model): - """One element in a Markov chain, some text or something.""" _start1 = '__start1' @@ -54,6 +44,8 @@ class MarkovState(models.Model): context = models.ForeignKey(MarkovContext, on_delete=models.CASCADE) class Meta: + """Options for the model itself.""" + index_together = [ ['context', 'k1', 'k2'], ['context', 'v'], @@ -65,6 +57,5 @@ class MarkovState(models.Model): unique_together = ('context', 'k1', 'k2', 'v') 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)