84 lines
2.4 KiB
Python
84 lines
2.4 KiB
Python
from datetime import datetime
|
|
from sqlalchemy import Column, DateTime, Float, ForeignKey, Integer, String
|
|
from sqlalchemy.orm import relationship
|
|
|
|
from db import Base, CRUD
|
|
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,
|
|
single_parent=True,
|
|
)
|
|
toolbox_id = Column(
|
|
Integer, ForeignKey("toolboxes.id", ondelete="CASCADE"), nullable=False
|
|
)
|
|
toolbox_data = relationship(
|
|
"Toolbox",
|
|
cascade="all, delete-orphan",
|
|
lazy="joined",
|
|
uselist=False,
|
|
single_parent=True,
|
|
)
|
|
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():
|
|
try:
|
|
setattr(self, key, value)
|
|
except Exception:
|
|
pass
|
|
|
|
def toDict(self):
|
|
return utils.toDict(self)
|
|
|
|
async def save(self):
|
|
return await CRUD.create(self, refresh=True)
|
|
|
|
async def edit(self, **kwargs):
|
|
return await CRUD.update(Stock, self.id, **kwargs)
|
|
|
|
|
|
class Placement(Base):
|
|
__tablename__ = "placements"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
toolbox_id = Column(
|
|
Integer, ForeignKey("toolboxes.id", ondelete="CASCADE"), nullable=False
|
|
)
|
|
toolkit_id = Column(
|
|
Integer, ForeignKey("toolkits.id", ondelete="CASCADE"), nullable=False
|
|
)
|
|
placement = Column(String, 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():
|
|
try:
|
|
setattr(self, key, value)
|
|
except Exception:
|
|
pass
|
|
|
|
def toDict(self):
|
|
return utils.toDict(self)
|
|
|
|
async def save(self) -> "Placement":
|
|
return await CRUD.create(self, refresh=True)
|
|
|
|
async def edit(self, placement: str):
|
|
return await CRUD.update(Placement, self.id, placement=placement)
|