50 lines
1.7 KiB
Python
50 lines
1.7 KiB
Python
|
import uuid
|
||
|
|
from sqlalchemy.dialects.postgresql import UUID
|
||
|
|
from sqlalchemy import (
|
||
|
|
Column,
|
||
|
|
Integer,
|
||
|
|
String,
|
||
|
|
BigInteger,
|
||
|
|
TIMESTAMP,
|
||
|
|
ForeignKey,
|
||
|
|
Boolean,
|
||
|
|
)
|
||
|
|
from sqlalchemy.orm import relationship, Mapped, mapped_column
|
||
|
|
from app.db.database import Base
|
||
|
|
from datetime import datetime, timezone
|
||
|
|
|
||
|
|
|
||
|
|
class UserFile(Base):
|
||
|
|
__tablename__ = "user_files"
|
||
|
|
id = Column(Integer, primary_key=True, index=True)
|
||
|
|
tenant_id = Column(
|
||
|
|
UUID(as_uuid=True), ForeignKey("tenants.id"), nullable=False, index=True
|
||
|
|
)
|
||
|
|
user_id = Column(Integer, ForeignKey("users.id"), nullable=False, index=True)
|
||
|
|
filename = Column(String, nullable=False)
|
||
|
|
storage_provider = Column(String(50), nullable=False, default="b2")
|
||
|
|
s3_key = Column(String, nullable=False)
|
||
|
|
size = Column(BigInteger, default=0)
|
||
|
|
content_type = Column(String, nullable=True)
|
||
|
|
status = Column(String, default="uploaded")
|
||
|
|
is_deleted = Column(Boolean, default=False)
|
||
|
|
deleted_at = Column(TIMESTAMP, nullable=True)
|
||
|
|
created_at = Column(TIMESTAMP, default=lambda: datetime.now(timezone.utc))
|
||
|
|
|
||
|
|
user = relationship("User", back_populates="user_files")
|
||
|
|
|
||
|
|
|
||
|
|
class UserStorageUsage(Base):
|
||
|
|
__tablename__ = "user_storage_usage"
|
||
|
|
user_id = Column(Integer, ForeignKey("users.id"), primary_key=True, nullable=False)
|
||
|
|
total_bytes_used = Column(BigInteger, default=0)
|
||
|
|
total_files_count = Column(Integer, default=0)
|
||
|
|
max_bytes_quota = Column(BigInteger, default=1073741824)
|
||
|
|
last_updated = Column(
|
||
|
|
TIMESTAMP,
|
||
|
|
default=lambda: datetime.now(timezone.utc),
|
||
|
|
onupdate=lambda: datetime.now(timezone.utc),
|
||
|
|
)
|
||
|
|
|
||
|
|
user = relationship("User", back_populates="storage_usage")
|