Add tags attribute to models

chatterbot/ext/django_chatterbot/admin.py

@@ -1,5 +1,7 @@
 from django.contrib import admin
-from chatterbot.ext.django_chatterbot.models import Statement, Response, Conversation
+from chatterbot.ext.django_chatterbot.models import (
+    Statement, Response, Conversation, Tag
+)
 
 
 class StatementAdmin(admin.ModelAdmin):
@@ -14,9 +16,16 @@ class ResponseAdmin(admin.ModelAdmin):
 
 
 class ConversationAdmin(admin.ModelAdmin):
-    pass
+    list_display = ('id', )
+
+
+class TagAdmin(admin.ModelAdmin):
+    list_display = ('name', )
+    list_filter = ('name', )
+    search_fields = ('name', )
 
 
 admin.site.register(Statement, StatementAdmin)
 admin.site.register(Response, ResponseAdmin)
 admin.site.register(Conversation, ConversationAdmin)
+admin.site.register(Tag, TagAdmin)

chatterbot/ext/django_chatterbot/migrations/0009_tags.py

@@ -0,0 +1,35 @@
+# -*- coding: utf-8 -*-
+# Generated by Django 1.11a1 on 2017-07-07 00:12
+from __future__ import unicode_literals
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('django_chatterbot', '0008_update_conversations'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='Tag',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('name', models.SlugField()),
+            ],
+            options={
+                'abstract': False,
+            },
+        ),
+        migrations.AlterField(
+            model_name='statement',
+            name='text',
+            field=models.CharField(max_length=255, unique=True),
+        ),
+        migrations.AddField(
+            model_name='tag',
+            name='statements',
+            field=models.ManyToManyField(related_name='tags', to='django_chatterbot.Statement'),
+        ),
+    ]

chatterbot/ext/django_chatterbot/models.py

@@ -202,6 +202,29 @@ def __str__(self):
         return str(self.id)
 
 
+class AbstractBaseTag(models.Model):
+    """
+    The abstract base tag allows other models to
+    be created using the attributes that exist on the
+    default models.
+    """
+
+    name = models.SlugField(
+        max_length=50
+    )
+
+    statements = models.ManyToManyField(
+        'Statement',
+        related_name='tags'
+    )
+
+    class Meta:
+        abstract = True
+
+    def __str__(self):
+        return self.name
+
+
 class Statement(AbstractBaseStatement):
     """
     A statement represents a single spoken entity, sentence or
@@ -225,3 +248,10 @@ class Conversation(AbstractBaseConversation):
     A sequence of statements representing a conversation.
     """
     pass
+
+
+class Tag(AbstractBaseTag):
+    """
+    A label that categorizes a statement.
+    """
+    pass

chatterbot/storage/sql_storage.py

@@ -10,6 +10,20 @@
     from sqlalchemy.sql import func
     from sqlalchemy import Table, Column, Integer, String, DateTime, ForeignKey, PickleType
 
+    tag_association_table = Table(
+        'tag_association',
+        Base.metadata,
+        Column('tag_id', Integer, ForeignKey('tag.id')),
+        Column('statement_id', Integer, ForeignKey('StatementTable.id'))
+    )
+
+    class Tag(Base):
+        """
+        A tag that describes a statement.
+        """
+
+        name = Column(String)
+
     class StatementTable(Base):
         """
         StatementTable, placeholder for a sentence or phrase.
@@ -25,6 +39,12 @@ def get_statement(self):
 
         text = Column(String, unique=True)
 
+        tags = relationship(
+            'Tag',
+            secondary=lambda: tag_association_table,
+            backref='statements'
+        )
+
         extra_data = Column(PickleType)
 
         in_response_to = relationship(