Add a base class for SQL Alchemy models

chatterbot/ext/sqlalchemy_app/models.py

@@ -0,0 +1,26 @@
+from sqlalchemy import Column, Integer
+from sqlalchemy.ext.declarative import (
+    declared_attr, declarative_base
+)
+
+
+class ModelBase(object):
+    """
+    An augmented base class for SqlAlchemy models.
+    """
+
+    @declared_attr
+    def __tablename__(cls):
+        """
+        Return the lowercase class name as the name of the table.
+        """
+        return cls.__name__.lower()
+
+    id = Column(
+        Integer,
+        primary_key=True,
+        autoincrement=True
+    )
+
+
+Base = declarative_base(cls=ModelBase)

chatterbot/storage/sql_storage.py

@@ -8,9 +8,7 @@
 Base = None
 
 try:
-    from sqlalchemy.ext.declarative import declarative_base
-
-    Base = declarative_base()
+    from chatterbot.ext.sqlalchemy_app.models import Base
 
     class StatementTable(Base):
         """
@@ -32,7 +30,6 @@ def get_statement_serialized(context):
             del params['text_search']
             return json.dumps(params)
 
-        id = Column(Integer, primary_key=True, autoincrement=True)
         text = Column(String, unique=True)
         extra_data = Column(PickleType)
 
@@ -60,7 +57,6 @@ def get_reponse_serialized(context):
             del params['text_search']
             return json.dumps(params)
 
-        id = Column(Integer, primary_key=True, autoincrement=True)
         text = Column(String)
         occurrence = Column(Integer, default=1)
         statement_text = Column(String, ForeignKey('StatementTable.text'))