Take advantage of the tag_exports cache for get_all_tag_names.
This commit is contained in:
parent
4bcfd9958a
commit
4b5a52abb4
1 changed files with 15 additions and 8 deletions
|
@ -258,6 +258,7 @@ class PDBCacheManagerMixin:
|
||||||
def get_cached_tag_export(self, function, **kwargs):
|
def get_cached_tag_export(self, function, **kwargs):
|
||||||
if isinstance(function, str):
|
if isinstance(function, str):
|
||||||
function = getattr(tag_export, function)
|
function = getattr(tag_export, function)
|
||||||
|
if 'tags' in kwargs:
|
||||||
kwargs['tags'] = tuple(kwargs['tags'])
|
kwargs['tags'] = tuple(kwargs['tags'])
|
||||||
key = (function.__name__,) + helpers.dict_to_tuple(kwargs)
|
key = (function.__name__,) + helpers.dict_to_tuple(kwargs)
|
||||||
try:
|
try:
|
||||||
|
@ -1062,25 +1063,31 @@ class PDBTagMixin:
|
||||||
else:
|
else:
|
||||||
raise exceptions.TagExists(existing_tag)
|
raise exceptions.TagExists(existing_tag)
|
||||||
|
|
||||||
def get_all_tag_names(self):
|
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.
|
|
||||||
'''
|
|
||||||
query = 'SELECT name FROM tags'
|
query = 'SELECT name FROM tags'
|
||||||
rows = self.sql_select(query)
|
rows = self.sql_select(query)
|
||||||
names = [row[0] for row in rows]
|
names = [row[0] for row in rows]
|
||||||
return names
|
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'
|
query = 'SELECT name, mastername FROM tag_synonyms'
|
||||||
rows = self.sql_select(query)
|
rows = self.sql_select(query)
|
||||||
synonyms = {syn: tag for (syn, tag) in rows}
|
synonyms = {syn: tag for (syn, tag) in rows}
|
||||||
return synonyms
|
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):
|
def get_root_tags(self):
|
||||||
'''
|
'''
|
||||||
Yield Tags that have no parent.
|
Yield Tags that have no parent.
|
||||||
|
|
Loading…
Reference in a new issue