75 lines
2.5 KiB
Python
75 lines
2.5 KiB
Python
import uuid
|
|
from datetime import datetime, timezone
|
|
|
|
from sqlalchemy import Column, DateTime, ForeignKey, Integer, String
|
|
from sqlalchemy.orm import relationship
|
|
|
|
from app.database import Base
|
|
|
|
|
|
def _utcnow() -> datetime:
|
|
return datetime.now(timezone.utc)
|
|
|
|
|
|
class User(Base):
|
|
__tablename__ = "users"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
username = Column(String(64), unique=True, nullable=False, index=True)
|
|
password_hash = Column(String(128), nullable=False)
|
|
created_at = Column(DateTime, default=_utcnow)
|
|
|
|
api_keys = relationship("ApiKey", back_populates="user", cascade="all, delete-orphan")
|
|
profiles = relationship("Profile", back_populates="user", cascade="all, delete-orphan")
|
|
|
|
|
|
class ApiKey(Base):
|
|
__tablename__ = "api_keys"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
user_id = Column(Integer, ForeignKey("users.id", ondelete="CASCADE"), nullable=False)
|
|
key = Column(String(64), unique=True, nullable=False, index=True)
|
|
created_at = Column(DateTime, default=_utcnow)
|
|
|
|
user = relationship("User", back_populates="api_keys")
|
|
|
|
@staticmethod
|
|
def generate_key() -> str:
|
|
return uuid.uuid4().hex
|
|
|
|
|
|
class Profile(Base):
|
|
__tablename__ = "profiles"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
user_id = Column(Integer, ForeignKey("users.id", ondelete="CASCADE"), nullable=False)
|
|
nickname = Column(String(64), nullable=False)
|
|
created_at = Column(DateTime, default=_utcnow)
|
|
|
|
user = relationship("User", back_populates="profiles")
|
|
postcards = relationship("Postcard", back_populates="profile", cascade="all, delete-orphan")
|
|
|
|
|
|
class Postcard(Base):
|
|
__tablename__ = "postcards"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
profile_id = Column(Integer, ForeignKey("profiles.id", ondelete="CASCADE"), nullable=False)
|
|
card_number = Column(String(64), nullable=False)
|
|
status = Column(String(16), nullable=False) # sent / delivered / received
|
|
# sent & delivered
|
|
recipient_name = Column(String(64))
|
|
country = Column(String(64))
|
|
send_time = Column(DateTime)
|
|
arrival_time = Column(DateTime)
|
|
# received
|
|
sender_name = Column(String(64))
|
|
receive_time = Column(DateTime)
|
|
# images
|
|
image_front = Column(String(256))
|
|
image_back = Column(String(256))
|
|
created_at = Column(DateTime, default=_utcnow)
|
|
updated_at = Column(DateTime, default=_utcnow, onupdate=_utcnow)
|
|
|
|
profile = relationship("Profile", back_populates="postcards")
|