File size: 3,109 Bytes
e17c9f2 479f67b e17c9f2 479f67b e17c9f2 479f67b e17c9f2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 |
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"
|