Merge pull request #500 from Deborah-Digges/add-dataset-name

Retriever CLI: Add dataset name to citation output
weecology · Jun 6, 2016 · e3600c1 · e3600c1
2 parents 177c4d7 + ee87098
commit e3600c1
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/__main__.py b/__main__.py
@@ -52,15 +52,15 @@ def main():
         elif args.command == 'citation':
             if args.dataset is None:
                 citation_path = os.path.join(os.path.split(__file__)[0], '../CITATION')
-                print "\nCitation for retriever:\n"
+                print("\nCitation for retriever:\n")
                 with open(citation_path) as citation_file:
-                    print citation_file.read()
+                    print(citation_file.read())
             else:
                 scripts = name_matches(script_list, args.dataset)
                 for dataset in scripts:
-
-                    print ("\nCitation:   {}".format(dataset.citation))
-                    print ("Description:   {}\n".format(dataset.description))
+                    print("\nDataset:  {}".format(dataset.name))
+                    print("Citation:   {}".format(dataset.citation))
+                    print("Description:   {}\n".format(dataset.description))
 
             return