""" Project: PARTS Website Author: Edward Middleton-Smith Precision And Research Technology Systems Limited Technology: Business Objects Feature: Company Business Object """ # internal from business_objects.base import Base from business_objects.db_base import SQLAlchemy_ABC, Get_Many_Parameters_Base import lib.argument_validation as av from extensions import db from helpers.helper_app import Helper_App # external from dataclasses import dataclass from typing import ClassVar class Company(SQLAlchemy_ABC, Base): ATTR_ID_COMPANY: ClassVar[str] = 'id_company' FLAG_COMPANY: ClassVar[str] = 'company' NAME_ATTR_OPTION_VALUE: ClassVar[str] = ATTR_ID_COMPANY NAME_ATTR_OPTION_TEXT: ClassVar[str] = Base.FLAG_NAME __tablename__ = 'DOG_Company' __table_args__ = { 'extend_existing': True } id_company = db.Column(db.Integer, primary_key=True) name = db.Column(db.String(250)) website = db.Column(db.String(1000)) active = db.Column(db.Boolean) def __init__(self): self.id_company = 0 self.is_new = False super().__init__() @classmethod def from_db_company(cls, query_row): _m = f'{cls.__qualname__}.from_db_company' company = cls() company.id_company = query_row[0] company.name = query_row[1] company.website = query_row[2] company.active = av.input_bool(query_row[3], 'active', _m) # company.created_on = query_row[7] return company @classmethod def from_db_user(cls, query_row): _m = f'{cls.__qualname__}.from_db_user' company = cls() company.id_company = query_row[8] company.name = query_row[9] company.website = query_row[10] company.active = True return company @classmethod def from_json(cls, json): _m = 'Company.from_json' company = cls() if json is None: return company company.id_company = json.get(cls.ATTR_ID_COMPANY, -1) company.name = json[cls.FLAG_NAME] company.website = json[cls.FLAG_WEBSITE] company.active = av.input_bool(json[cls.FLAG_ACTIVE], cls.FLAG_ACTIVE, _m) return company def to_json(self): as_json = { **self.get_shared_json_attributes(self) , self.ATTR_ID_COMPANY: self.id_company , self.FLAG_NAME: self.name , self.FLAG_WEBSITE: self.website , self.FLAG_ACTIVE: self.active } return as_json def __repr__(self): return f''' {self.__class__.__name__}( {self.FLAG_COMPANY}: {self.id_company} {self.FLAG_NAME}: {self.name} {self.FLAG_WEBSITE}: {self.website} {self.FLAG_ACTIVE}: {self.active} ) ''' class Company_Temp(db.Model, Base): __tablename__ = 'DOG_Company_Temp' __table_args__ = { 'extend_existing': True } id_temp = db.Column(db.Integer, primary_key=True) id_company = db.Column(db.Integer) name = db.Column(db.String(250)) website = db.Column(db.String(1000)) active = db.Column(db.Boolean) guid: str = db.Column(db.String(36)) def __init__(self): super().__init__() @classmethod def from_company(cls, company): _m = 'Company_Temp.from_Company' temp = cls() temp.id_company = company.id_company temp.name = company.name temp.website = company.website temp.active = company.active return temp class Parameters_Company(Get_Many_Parameters_Base): get_all_company: bool get_inactive_company: bool ids_company: str names_company: str websites_company: str require_all_id_search_filters_met: bool require_any_id_search_filters_met: bool require_all_non_id_search_filters_met: bool require_any_non_id_search_filters_met: bool @classmethod def get_default(cls): return cls( get_all_company = True , get_inactive_company = False , ids_company = '' , names_company = '' , websites_company = '' , require_all_id_search_filters_met = True , require_any_id_search_filters_met = True , require_all_non_id_search_filters_met = False , require_any_non_id_search_filters_met = True ) @classmethod def from_json(cls, json): return cls( get_all_company = json.get('a_get_all_company', False) , get_inactive_company = json.get('a_get_inactive_company', False) , ids_company = json.get('a_ids_company', '') , names_company = json.get('a_names_company', '') , websites_company = json.get('a_websites_company', '') , require_all_id_search_filters_met = json.get('a_require_all_id_search_filters_met', True) , require_any_id_search_filters_met = json.get('a_require_any_id_search_filters_met', True) , require_all_non_id_search_filters_met = json.get('a_require_all_non_id_search_filters_met', False) , require_any_non_id_search_filters_met = json.get('a_require_any_non_id_search_filters_met', True) ) """ @classmethod def from_form_filters_company(cls, form): av.val_instance(form, 'form', 'Parameters_Company.from_form_filters_company', Filters_Company) has_filter_search_text = not (form.search.data == '' or form.search.data is None) active_only = av.input_bool(form.active_only.data, "active", "Parameters_Company.from_form_filters_company") filters = cls.get_default() filters.get_all_company = True filters.get_inactive_company = not active_only filters.ids_company = '' filters.names_company = form.search.data if has_filter_search_text else '' filters.notes_company = form.search.data if has_filter_search_text else '' return filters """ def to_json(self): return { 'a_get_all_company': self.get_all_company , 'a_get_inactive_company': self.get_inactive_company , 'a_ids_company': self.ids_company , 'a_names_company': self.names_company , 'a_websites_company': self.websites_company , 'a_require_all_id_search_filters_met': self.require_all_id_search_filters_met , 'a_require_any_id_search_filters_met': self.require_any_id_search_filters_met , 'a_require_all_non_id_search_filters_met': self.require_all_non_id_search_filters_met , 'a_require_any_non_id_search_filters_met': self.require_any_non_id_search_filters_met }