46 lines
1.5 KiB
Python
46 lines
1.5 KiB
Python
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)
|