68 lines
2.2 KiB
Python
68 lines
2.2 KiB
Python
|
from ..errors import Errors
|
||
|
|
|
||
|
|
|
||
|
|
class SimpleFrozenDict(dict):
|
||
|
|
"""Simplified implementation of a frozen dict, mainly used as default
|
||
|
|
function or method argument (for arguments that should default to empty
|
||
|
|
dictionary). Will raise an error if user or Weasel attempts to add to dict.
|
||
|
|
"""
|
||
|
|
|
||
|
|
def __init__(self, *args, error: str = Errors.E001, **kwargs) -> None:
|
||
|
|
"""Initialize the frozen dict. Can be initialized with pre-defined
|
||
|
|
values.
|
||
|
|
|
||
|
|
error (str): The error message when user tries to assign to dict.
|
||
|
|
"""
|
||
|
|
super().__init__(*args, **kwargs)
|
||
|
|
self.error = error
|
||
|
|
|
||
|
|
def __setitem__(self, key, value):
|
||
|
|
raise NotImplementedError(self.error)
|
||
|
|
|
||
|
|
def pop(self, key, default=None):
|
||
|
|
raise NotImplementedError(self.error)
|
||
|
|
|
||
|
|
def update(self, other):
|
||
|
|
raise NotImplementedError(self.error)
|
||
|
|
|
||
|
|
|
||
|
|
class SimpleFrozenList(list):
|
||
|
|
"""Wrapper class around a list that lets us raise custom errors if certain
|
||
|
|
attributes/methods are accessed. Mostly used for properties like
|
||
|
|
Language.pipeline that return an immutable list (and that we don't want to
|
||
|
|
convert to a tuple to not break too much backwards compatibility). If a user
|
||
|
|
accidentally calls nlp.pipeline.append(), we can raise a more helpful error.
|
||
|
|
"""
|
||
|
|
|
||
|
|
def __init__(self, *args, error: str = Errors.E002) -> None:
|
||
|
|
"""Initialize the frozen list.
|
||
|
|
|
||
|
|
error (str): The error message when user tries to mutate the list.
|
||
|
|
"""
|
||
|
|
self.error = error
|
||
|
|
super().__init__(*args)
|
||
|
|
|
||
|
|
def append(self, *args, **kwargs):
|
||
|
|
raise NotImplementedError(self.error)
|
||
|
|
|
||
|
|
def clear(self, *args, **kwargs):
|
||
|
|
raise NotImplementedError(self.error)
|
||
|
|
|
||
|
|
def extend(self, *args, **kwargs):
|
||
|
|
raise NotImplementedError(self.error)
|
||
|
|
|
||
|
|
def insert(self, *args, **kwargs):
|
||
|
|
raise NotImplementedError(self.error)
|
||
|
|
|
||
|
|
def pop(self, *args, **kwargs):
|
||
|
|
raise NotImplementedError(self.error)
|
||
|
|
|
||
|
|
def remove(self, *args, **kwargs):
|
||
|
|
raise NotImplementedError(self.error)
|
||
|
|
|
||
|
|
def reverse(self, *args, **kwargs):
|
||
|
|
raise NotImplementedError(self.error)
|
||
|
|
|
||
|
|
def sort(self, *args, **kwargs):
|
||
|
|
raise NotImplementedError(self.error)
|