7731: Fix error tagging source in citation tree view

This commit is contained in:
Nick Hall 2014-05-22 14:04:07 +01:00
parent 77f652bb22
commit b7bb377939

View File

@ -568,13 +568,18 @@ class CitationTreeView(ListView):
all_links = all_links.union(links) all_links = all_links.union(links)
self.row_update(list(all_links)) self.row_update(list(all_links))
def add_tag(self, transaction, citation_handle, tag_handle): def add_tag(self, transaction, handle, tag_handle):
""" """
Add the given tag to the given citation. Add the given tag to the given source or citation.
""" """
citation = self.dbstate.db.get_citation_from_handle(citation_handle) citation = self.dbstate.db.get_citation_from_handle(handle)
if citation:
citation.add_tag(tag_handle) citation.add_tag(tag_handle)
self.dbstate.db.commit_citation(citation, transaction) self.dbstate.db.commit_citation(citation, transaction)
else:
source = self.dbstate.db.get_source_from_handle(handle)
source.add_tag(tag_handle)
self.dbstate.db.commit_source(source, transaction)
def get_default_gramplets(self): def get_default_gramplets(self):
""" """