35 lines
1.5 KiB
Python
35 lines
1.5 KiB
Python
from sqlalchemy import Column, Integer, String, create_engine
|
|
from sqlalchemy.ext.declarative import declarative_base
|
|
from sqlalchemy.orm import sessionmaker
|
|
from sqlalchemy import Column, Integer, Text, ForeignKey, DateTime
|
|
from sqlalchemy.sql import func
|
|
|
|
SQLALCHEMY_DATABASE_URL = "sqlite:///./chepuhagram.db"
|
|
engine = create_engine(SQLALCHEMY_DATABASE_URL, connect_args={"check_same_thread": False})
|
|
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
|
|
Base = declarative_base()
|
|
Base.metadata.create_all(bind=engine)
|
|
|
|
class User(Base):
|
|
__tablename__ = "users"
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
first_name = Column(String(50), nullable=False, server_default="User")
|
|
last_name = Column(String(50), nullable=True)
|
|
username = Column(String, unique=True, index=True)
|
|
about = Column(String, nullable=True)
|
|
phone = Column(String(20), unique=True, nullable=True)
|
|
totp_secret = Column(String(32), nullable=True)
|
|
hashed_password = Column(String)
|
|
public_key = Column(String, nullable=True)
|
|
encrypted_private_key = Column(String, nullable=True)
|
|
fcm_token = Column(String, nullable=True)
|
|
|
|
class Message(Base):
|
|
__tablename__ = "messages"
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
sender_id = Column(Integer, ForeignKey("users.id"))
|
|
receiver_id = Column(Integer, ForeignKey("users.id"))
|
|
content = Column(Text)
|
|
timestamp = Column(DateTime(timezone=True), server_default=func.now())
|
|
|
|
Base.metadata.create_all(bind=engine) |