|
r"""_summary_ |
|
-*- coding: utf-8 -*- |
|
|
|
Module : data.utils.base_company |
|
|
|
File Name : base_company.py |
|
|
|
Description : The base class of the factory class, used to register and reflect specific classes |
|
|
|
Creation Date : 2024-10-29 |
|
|
|
Author : Frank Kang([email protected]) |
|
""" |
|
import threading |
|
from typing import Any |
|
from typing_extensions import override |
|
|
|
|
|
class BaseCompany(object): |
|
"""_summary_ |
|
|
|
The base class of the factory class, used to register and reflect specific classes. Use singleton mode, so it is necessary to maintain consistency in the path when importing and changing classes |
|
|
|
For example: |
|
``` |
|
base_company = BaseCompany.get() |
|
|
|
# Of course, you can also obtain the singleton using the following methods |
|
base_company = BaseCompany() |
|
|
|
entity = base_company[registered_name] |
|
``` |
|
""" |
|
_instance = None |
|
_lock = threading.Lock() |
|
|
|
def __new__(cls, *args, **kwargs): |
|
with cls._lock: |
|
if cls._instance is None: |
|
cls._instance = super(BaseCompany, cls).__new__( |
|
cls, *args, **kwargs) |
|
cls._instance.__init__() |
|
return cls._instance |
|
|
|
def __init__(self): |
|
self.entities = {} |
|
|
|
def init_factory(self): |
|
"""_summary_ |
|
|
|
Used for initializing singleton |
|
""" |
|
self.entities = {} |
|
|
|
@classmethod |
|
def get(cls, *args, **kwargs): |
|
"""_summary_ |
|
|
|
Method for obtaining singleton classes |
|
|
|
For example: |
|
``` |
|
base_company = BaseCompany.get() |
|
entity = base_company[registered_name] |
|
``` |
|
|
|
Returns: |
|
BaseCompany: singleton |
|
""" |
|
if cls._instance is None: |
|
cls.__new__(cls, *args, **kwargs) |
|
return cls._instance |
|
|
|
def register(self, entity_name: str, entity: Any) -> bool: |
|
"""_summary_ |
|
|
|
Register the entity, which is called by the automatic registrar. Please do not call it yourself. Each name can only be registered once |
|
|
|
Args: |
|
entity_name (str): Name used for registration |
|
entity (Any): Registered entity |
|
|
|
Returns: |
|
bool: Registration success returns true, failure returns false |
|
""" |
|
if entity_name not in self.entities: |
|
self.entities[entity_name] = entity |
|
return True |
|
else: |
|
return False |
|
|
|
def delete(self, entity_name: str) -> bool: |
|
"""_summary_ |
|
|
|
Remove registered entities, please use with caution |
|
|
|
Args: |
|
entity_name (str): The registered name of the registered entity |
|
|
|
Returns: |
|
bool: Success in deletion returns true, failure returns false |
|
""" |
|
if entity_name in self.entities: |
|
self.entities[entity_name] = None |
|
del self.entities[entity_name] |
|
return True |
|
else: |
|
return False |
|
|
|
def __getitem__(self, key): |
|
return self.entities[key] |
|
|
|
def __len__(self): |
|
return len(self.entities) |
|
|
|
@override |
|
def __repr__(self) -> str: |
|
return "BaseCompany" |
|
|
