handle missing cites

McTavishLab · Nov 13, 2019 · 7ee8afd · 7ee8afd
1 parent f97a109
commit 7ee8afd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/physcraper/opentree_helpers.py b/physcraper/opentree_helpers.py
@@ -104,7 +104,8 @@ def get_tree_from_synth(ott_ids, label_format="name", citation="cites.txt"):
         query = {"ot:studyId":study}
         new_cite = oti.find_studies(query_dict = query, verbose=True)
         #print new_cite[0].keys()
-        cites = cites + '\n' + to_string(new_cite[0]['ot:studyPublicationReference']) + '\n' + new_cite[0]['ot:studyPublication']
+        if new_cite:
+            cites = cites + '\n' + to_string(new_cite[0]['ot:studyPublicationReference']) + '\n' + new_cite[0]['ot:studyPublication']
   #  cites = cites + '\n' +phylesystemref + synthref
     with open(citation,'w') as citfile:
         citfile.write(cites)