优化model

chat/models/user.py

@@ -1,7 +1,7 @@
 from sqlalchemy import Column, Integer, String, DateTime, Boolean
-from sqlalchemy.ext.declarative import declarative_base
 
-Base = declarative_base()
+from db.session import Base
+
 
 class AuthToken(Base):
     __tablename__ = 'authtoken_token'
@@ -12,6 +12,7 @@ class AuthToken(Base):
 
 class DjangoUser(Base):
     __tablename__ = 'system_users'
+    
     id = Column(Integer, primary_key=True)
     username = Column(String(150), nullable=False)
     email = Column(String(254))