93 lines
3.0 KiB
Python
93 lines
3.0 KiB
Python
"""
|
|
Project: PARTS Website
|
|
Author: Edward Middleton-Smith
|
|
Precision And Research Technology Systems Limited
|
|
|
|
Technology: Business Objects
|
|
Feature: Plant Business Object
|
|
|
|
Description:
|
|
Business object for plant
|
|
"""
|
|
|
|
# internal
|
|
import lib.argument_validation as av
|
|
from business_objects.address import Address
|
|
from business_objects.store.store_base import Store_Base
|
|
from extensions import db
|
|
# external
|
|
from typing import ClassVar
|
|
|
|
|
|
class Plant(db.Model, Store_Base):
|
|
NAME_ATTR_OPTION_VALUE: ClassVar[str] = Store_Base.ATTR_ID_PLANT
|
|
NAME_ATTR_OPTION_TEXT: ClassVar[str] = Store_Base.FLAG_NAME
|
|
__tablename__ = 'Shop_Plant'
|
|
id_plant = db.Column(db.Integer, primary_key=True)
|
|
code = db.Column(db.String(50))
|
|
name = db.Column(db.String(255))
|
|
id_address = db.Column(db.Integer)
|
|
id_user_manager = db.Column(db.Integer)
|
|
active = db.Column(db.Boolean)
|
|
|
|
# address = None
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
Store_Base.__init__(self)
|
|
self.address = None
|
|
@classmethod
|
|
def from_DB_storage_location(cls, query_row):
|
|
plant = cls()
|
|
plant.id_plant = query_row[1]
|
|
plant.id_address = query_row[2]
|
|
plant.address = Address.from_DB_storage_location(query_row)
|
|
return plant
|
|
@classmethod
|
|
def from_DB_plant(cls, query_row):
|
|
plant = cls()
|
|
plant.id_plant = query_row[0]
|
|
plant.id_address = query_row[1]
|
|
plant.address = Address.from_DB_plant(query_row)
|
|
plant.code = query_row[4]
|
|
plant.name = query_row[5]
|
|
plant.active = query_row[6]
|
|
return plant
|
|
@classmethod
|
|
def from_DB_stock_item(cls, query_row):
|
|
plant = cls()
|
|
plant.id_plant = query_row[5]
|
|
plant.code = query_row[10]
|
|
plant.name = query_row[11]
|
|
return plant
|
|
def __repr__(self):
|
|
return f'''
|
|
{self.ATTR_ID_PLANT}: {self.id_plant}
|
|
{self.FLAG_CODE}: {self.code}
|
|
{self.FLAG_NAME}: {self.name}
|
|
{self.FLAG_ADDRESS}: {self.address}
|
|
{self.ATTR_ID_USER_MANAGER}: {self.id_user_manager}
|
|
{self.FLAG_ACTIVE}: {self.active}
|
|
'''
|
|
def to_json(self):
|
|
return {
|
|
**self.get_shared_json_attributes(self),
|
|
self.ATTR_ID_PLANT: self.id_plant,
|
|
self.FLAG_CODE: self.code,
|
|
self.FLAG_NAME: self.name,
|
|
self.FLAG_ADDRESS: self.address.to_json(),
|
|
self.ATTR_ID_USER_MANAGER: self.id_user_manager,
|
|
self.FLAG_ACTIVE: 1 if av.input_bool(self.active, self.FLAG_ACTIVE, f'{self.__class__.__name__}.to_json') else 0
|
|
}
|
|
@classmethod
|
|
def from_json(cls, json):
|
|
print(f'{cls.__name__}.from_json: {json}')
|
|
plant = cls()
|
|
plant.id_plant = json[cls.ATTR_ID_PLANT],
|
|
plant.code = json[cls.FLAG_CODE],
|
|
plant.name = json[cls.FLAG_NAME],
|
|
plant.address = Address.from_json(json[cls.FLAG_ADDRESS]),
|
|
plant.id_user_manager = json[cls.ATTR_ID_USER_MANAGER],
|
|
plant.active = json[cls.FLAG_ACTIVE]
|
|
return plant
|