dog_training/business_objects/dog/company.py

"""
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
        }