Feat: Companies and User permissions added.

business_objects/dog/company.py

@@ -0,0 +1,183 @@
+"""
+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[6]
+        company.name = query_row[7]
+        company.website = query_row[8]
+        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
+        }