Take advantage of the tag_exports cache for get_all_tag_names.

This commit is contained in:
voussoir 2020-09-17 21:22:54 -07:00
parent 4bcfd9958a
commit 4b5a52abb4

View file

@ -258,6 +258,7 @@ class PDBCacheManagerMixin:
def get_cached_tag_export(self, function, **kwargs):
if isinstance(function, str):
function = getattr(tag_export, function)
if 'tags' in kwargs:
kwargs['tags'] = tuple(kwargs['tags'])
key = (function.__name__,) + helpers.dict_to_tuple(kwargs)
try:
@ -1062,25 +1063,31 @@ class PDBTagMixin:
else:
raise exceptions.TagExists(existing_tag)
def get_all_tag_names(self):
'''
Return a list containing the names of all tags as strings.
Useful for when you don't want the overhead of actual Tag objects.
'''
def _get_all_tag_names(self):
query = 'SELECT name FROM tags'
rows = self.sql_select(query)
names = [row[0] for row in rows]
return names
def get_all_synonyms(self):
def get_all_tag_names(self):
'''
Return a dict mapping {synonym: mastertag} as strings.
Return a list containing the names of all tags as strings.
Useful for when you don't want the overhead of actual Tag objects.
'''
return self.get_cached_tag_export(self._get_all_tag_names)
def _get_all_synonyms(self):
query = 'SELECT name, mastername FROM tag_synonyms'
rows = self.sql_select(query)
synonyms = {syn: tag for (syn, tag) in rows}
return synonyms
def get_all_synonyms(self):
'''
Return a dict mapping {synonym: mastertag} as strings.
'''
return self.get_cached_tag_export(self._get_all_synonyms)
def get_root_tags(self):
'''
Yield Tags that have no parent.