jan/masque
1
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

add more dict-like methods

Browse Source
This commit is contained in:
Jan Petykiewicz 2020-09-28 23:49:42 -07:00
parent b873a5ddf3
commit 03a359e446
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
masque/library/library.py
View File

@ -84,6 +84,11 @@ class Library:
def __getitem__(self, key: str) -> 'Pattern': def __getitem__(self, key: str) -> 'Pattern':
return self.get_primary(key) return self.get_primary(key)
def __iter__(self) -> Iterator[str]:
return self.keys()
def __contains__(self, key: str) -> bool:
return key in self.primary
def get_primary(self, key: str) -> 'Pattern': def get_primary(self, key: str) -> 'Pattern':
if self.enable_cache and key in self.cache: if self.enable_cache and key in self.cache:
@ -127,6 +132,15 @@ class Library:
raise LibraryError(f'Broken reference to {key} (tag {tag})') raise LibraryError(f'Broken reference to {key} (tag {tag})')
return pat return pat
def keys(self) -> Iterator[str]:
return self.primary.keys()
def values(self) -> Iterator['Pattern']:
return (self[key] for key in self.keys())
def items(self) -> Iterator[Tuple[str, 'Pattern']]:
return ((key, self[key]) for key in self.keys())
def __repr__(self) -> str: def __repr__(self) -> str:
return '<Library with keys ' + repr(list(self.primary.keys())) + '>' return '<Library with keys ' + repr(list(self.primary.keys())) + '>'