52 lines
1.9 KiB
Python
52 lines
1.9 KiB
Python
|
from time import time
|
||
|
|
from typing import List, Self
|
||
|
|
from uuid import UUID, uuid4
|
||
|
|
|
||
|
|
from sqlalchemy import BigInteger, ForeignKey
|
||
|
|
from sqlalchemy.orm import mapped_column, Mapped, relationship
|
||
|
|
from sqlalchemy.orm.attributes import InstrumentedAttribute
|
||
|
|
import sqlalchemy as sa
|
||
|
|
from pydantic import BaseModel
|
||
|
|
|
||
|
|
from materia_server.models.base import Base
|
||
|
|
from materia_server.models import database
|
||
|
|
|
||
|
|
|
||
|
|
class Repository(Base):
|
||
|
|
__tablename__ = "repository"
|
||
|
|
|
||
|
|
id: Mapped[int] = mapped_column(BigInteger, primary_key = True)
|
||
|
|
user_id: Mapped[UUID] = mapped_column(ForeignKey("user.id"))
|
||
|
|
capacity: Mapped[int] = mapped_column(BigInteger, nullable = False)
|
||
|
|
|
||
|
|
user: Mapped["User"] = relationship(back_populates = "repository")
|
||
|
|
directories: Mapped[List["Directory"]] = relationship(back_populates = "repository")
|
||
|
|
files: Mapped[List["File"]] = relationship(back_populates = "repository")
|
||
|
|
|
||
|
|
def to_dict(self) -> dict:
|
||
|
|
return { k: getattr(self, k) for k, v in Repository.__dict__.items() if isinstance(v, InstrumentedAttribute) }
|
||
|
|
|
||
|
|
async def create(self, db: database.Database):
|
||
|
|
async with db.session() as session:
|
||
|
|
session.add(self)
|
||
|
|
await session.commit()
|
||
|
|
|
||
|
|
async def update(self, db: database.Database):
|
||
|
|
async with db.session() as session:
|
||
|
|
await session.execute(sa.update(Repository).where(Repository.id == self.id).values(self.to_dict()))
|
||
|
|
await session.commit()
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
async def by_user_id(user_id: UUID, db: database.Database) -> Self | None:
|
||
|
|
async with db.session() as session:
|
||
|
|
return (await session.scalars(sa.select(Repository).where(Repository.user_id == user_id))).first()
|
||
|
|
|
||
|
|
|
||
|
|
class RepositoryInfo(BaseModel):
|
||
|
|
capacity: int
|
||
|
|
used: int
|
||
|
|
|
||
|
|
from materia_server.models.user import User
|
||
|
|
from materia_server.models.directory import Directory
|
||
|
|
from materia_server.models.file import File
|