начало БД

db/schemas/__init__.py

@@ -0,0 +1,6 @@
+from user import *
+from access import *
+from toolkit import *
+from categories import *
+from db.schemas.toolbox import *
+from stock import *

db/schemas/access.py

@@ -0,0 +1,53 @@
+from datetime import datetime
+from sqlalchemy import Boolean, Column, DateTime, Integer, String, Text
+from db import Base
+import utils
+
+
+class AccessLevel(Base):
+    __tablename__ = "access_level"
+
+    id = Column(Integer, primary_key=True, index=True)
+    title = Column(String, unique=True, index=True)
+    description = Column(Text)
+    created_at = Column(DateTime, default=datetime.now)
+    updated_at = Column(DateTime, default=datetime.now, onupdate=datetime.now)
+
+    # permissions
+    receiving_edit = Column(Boolean, default=False)
+    refund_request_edit = Column(Boolean, default=False)
+    refund_request_confirm = Column(Boolean, default=False)
+    debit_request_edit = Column(Boolean, default=False)
+    debit_request_confirm = Column(Boolean, default=False)
+    tools_creation = Column(Boolean, default=False)
+    tools_registration = Column(Boolean, default=False)
+    tools_registration_edit = Column(Boolean, default=False)
+    tools_edit = Column(Boolean, default=False)
+    tools_achievement = Column(Boolean, default=False)
+    users_creation = Column(Boolean, default=False)
+    users_edit = Column(Boolean, default=False)
+    users_disabling = Column(Boolean, default=False)
+    users_view = Column(Boolean, default=False)
+    available_own_toolbox = Column(Boolean, default=False)
+    view_all_toolboxes = Column(Boolean, default=False)
+    view_requests = Column(Boolean, default=False)
+    access_level_view = Column(Boolean, default=False)
+    access_level_edit = Column(Boolean, default=False)
+    manage_toolboxes = Column(Boolean, default=False)
+
+    def __init__(self, **kwargs):
+        for key, value in kwargs.items():
+            setattr(self, key, value)
+
+    def toDict(self):
+        return utils.toDict(self)
+
+    async def save(self):
+        from db import CRUD
+
+        return await CRUD.create(self, refresh=True)
+
+    async def edit(self, **kwargs):
+        from db import CRUD
+
+        return await CRUD.update(AccessLevel, self.id, **kwargs)

db/schemas/categories.py

@@ -0,0 +1,31 @@
+from datetime import datetime
+from sqlalchemy import Column, DateTime, Integer, String, Text
+from db import Base
+import utils
+
+
+class Category(Base):
+    __tablename__ = "categories"
+
+    id = Column(Integer, primary_key=True, index=True)
+    title = Column(String, unique=True, index=True)
+    description = Column(Text)
+    created_at = Column(DateTime, default=datetime.now)
+    updated_at = Column(DateTime, default=datetime.now, onupdate=datetime.now)
+
+    def __init__(self, **kwargs):
+        for key, value in kwargs.items():
+            setattr(self, key, value)
+
+    def toDict(self):
+        return utils.toDict(self)
+
+    async def save(self):
+        from db import CRUD
+
+        return await CRUD.create(self, refresh=True)
+
+    async def edit(id: int, **kwargs):
+        from db import CRUD
+
+        return await CRUD.update(Category, id, **kwargs)

db/schemas/stock.py

@@ -0,0 +1,45 @@
+from datetime import datetime
+from sqlalchemy import Column, DateTime, Float, ForeignKey, Integer
+from sqlalchemy.orm import relationship
+
+from db import Base
+import utils
+
+
+class Stock(Base):
+    __tablename__ = "stocks"
+
+    id = Column(Integer, primary_key=True, index=True)
+    toolkit_id = Column(
+        Integer, ForeignKey("toolkits.id", ondelete="CASCADE"), nullable=False
+    )
+    toolkit_data = relationship(
+        "Toolkit", cascade="all, delete-orphan", lazy="joined", uselist=False
+    )
+    toolbox_id = Column(
+        Integer, ForeignKey("toolboxes.id", ondelete="CASCADE"), nullable=False
+    )
+    toolbox_data = relationship(
+        "Toolbox", cascade="all, delete-orphan", lazy="joined", uselist=False
+    )
+    quantity = Column(Integer, nullable=False)
+    price = Column(Float, nullable=False)
+    created_at = Column(DateTime, default=datetime.now)
+    updated_at = Column(DateTime, default=datetime.now, onupdate=datetime.now)
+
+    def __init__(self, **kwargs):
+        for key, value in kwargs.items():
+            setattr(self, key, value)
+
+    def toDict(self):
+        return utils.toDict(self)
+
+    async def save(self):
+        from db import CRUD
+
+        return await CRUD.create(self, refresh=True)
+
+    async def edit(self, **kwargs):
+        from db import CRUD
+
+        return await CRUD.update(Stock, self.id, **kwargs)

db/schemas/toolbox.py

@@ -0,0 +1,34 @@
+from datetime import datetime
+from sqlalchemy import Column, DateTime, ForeignKey, Integer, String, Text
+from db import Base
+import utils
+
+
+class Toolbox(Base):
+    __tablename__ = "toolboxes"
+
+    id = Column(Integer, primary_key=True, index=True)
+    title = Column(String, unique=True, index=True)
+    description = Column(Text)
+    owner_id = Column(
+        Integer, ForeignKey("users.id", ondelete="CASCADE"), nullable=True
+    )
+    created_at = Column(DateTime, default=datetime.now)
+    updated_at = Column(DateTime, default=datetime.now, onupdate=datetime.now)
+
+    def __init__(self, **kwargs):
+        for key, value in kwargs.items():
+            setattr(self, key, value)
+
+    def toDict(self):
+        return utils.toDict(self)
+
+    async def save(self):
+        from db import CRUD
+
+        return await CRUD.create(self, refresh=True)
+
+    async def edit(id: int, **kwargs):
+        from db import CRUD
+
+        return await CRUD.update(Toolbox, id, **kwargs)

db/schemas/toolkit.py

@@ -0,0 +1,45 @@
+from datetime import datetime
+from sqlalchemy import Column, DateTime, ForeignKey, Integer, String, Text
+from sqlalchemy.dialects.postgresql import JSONB
+from sqlalchemy.orm import relationship
+from db import Base
+import utils
+
+
+class Toolkit(Base):
+    __tablename__ = "toolkits"
+
+    id = Column(Integer, primary_key=True, index=True)
+    title = Column(String, unique=True, index=True)
+    description = Column(Text)
+    category_id = Column(Integer, ForeignKey("categories.id", ondelete="CASCADE"))
+    category_data = relationship(
+        "Category", cascade="all, delete-orphan", lazy="joined", uselist=False
+    )
+    image = Column(
+        JSONB, default={"main": "images/tools/default.png", "additional": []}
+    )
+    quantity_min = Column(Integer, nullable=True)
+    quantity_min_extra = Column(Integer, nullable=True)
+    external_link = Column(String, nullable=True)
+    created_at = Column(DateTime, default=datetime.now)
+    updated_at = Column(DateTime, default=datetime.now, onupdate=datetime.now)
+    refilled_at = Column(DateTime, default=datetime.now)
+    moved_at = Column(DateTime, default=datetime.now)
+
+    def __init__(self, **kwargs):
+        for key, value in kwargs.items():
+            setattr(self, key, value)
+
+    def toDict(self):
+        return utils.toDict(self)
+
+    async def save(self):
+        from db import CRUD
+
+        return await CRUD.create(self, refresh=True)
+
+    async def edit(id: int, **kwargs):
+        from db import CRUD
+
+        return await CRUD.update(Toolkit, id, **kwargs)

db/schemas/user.py

@@ -0,0 +1,35 @@
+from datetime import datetime
+from sqlalchemy import Boolean, Column, DateTime, ForeignKey, Integer, String
+from db import Base
+import utils
+
+
+class User(Base):
+    __tablename__ = "users"
+
+    id = Column(Integer, primary_key=True, index=True)
+    login = Column(String, unique=True, index=True)
+    username = Column(String, unique=True, index=True)
+    photo = Column(String, default="images/users/default.png")
+    hashed_password = Column(String)
+    access_level_id = Column(Integer, ForeignKey("access_level.id", ondelete="CASCADE"))
+    is_active = Column(Boolean, default=True)
+    created_at = Column(DateTime, default=datetime.now)
+    updated_at = Column(DateTime, default=datetime.now, onupdate=datetime.now)
+
+    def __init__(self, **kwargs):
+        for key, value in kwargs.items():
+            setattr(self, key, value)
+
+    def toDict(self):
+        return utils.toDict(self)
+
+    async def save(self) -> "User":
+        from db import CRUD
+
+        return await CRUD.create(self, refresh=True)
+
+    async def edit(self, **kwargs) -> "User":
+        from db import CRUD
+
+        return await CRUD.update(User, self.id, **kwargs)