linting fixes on the markov models

This commit is contained in:
Brian S. Stephan 2023-02-16 16:17:49 -06:00
parent 1cf0364268
commit 8b3caabb57
Signed by: bss
GPG Key ID: 3DE06D3180895FCB
1 changed files with 7 additions and 16 deletions

View File

@ -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)