mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
31 lines
857 B
Python
31 lines
857 B
Python
import collections
|
|
|
|
from extras.constants import EXTRAS_FEATURES
|
|
|
|
|
|
class Registry(dict):
|
|
"""
|
|
Central registry for registration of functionality. Once a store (key) is defined, it cannot be overwritten or
|
|
deleted (although its value may be manipulated).
|
|
"""
|
|
def __getitem__(self, key):
|
|
try:
|
|
return super().__getitem__(key)
|
|
except KeyError:
|
|
raise KeyError(f"Invalid store: {key}")
|
|
|
|
def __setitem__(self, key, value):
|
|
if key in self:
|
|
raise KeyError(f"Store already set: {key}")
|
|
super().__setitem__(key, value)
|
|
|
|
def __delitem__(self, key):
|
|
raise TypeError("Cannot delete stores from registry")
|
|
|
|
|
|
# Initialize the global registry
|
|
registry = Registry()
|
|
registry['model_features'] = {
|
|
feature: collections.defaultdict(set) for feature in EXTRAS_FEATURES
|
|
}
|