track args

Author: ahalev

src/expfig/goodybag/figure_tracker.py

@@ -14,7 +14,8 @@ def track_savefig(fname, *args, show=False, tracker_file=None, **kwargs):
         raise ImportError("matplotlib must be installed to use 'savefig'")
 
     plt.savefig(fname, *args, **kwargs)
-    track_save_to(fname, tracker_file)
+
+    track_save_to(script, fname)
 
     if show:
         plt.show()
@@ -24,27 +25,28 @@ def track_savetable(table, fname, print_out=False, tracker_file=None):
     with open(fname, 'w') as f:
         f.write(table)
 
-    track_save_to(fname, tracker_file)
+    track_save_to(script, fname)
 
     if print_out:
         print(table)
 
 
 def track_save_to(fname, tracker_file=None):
-    save_script_result(sys.argv[0], fname, tracker_file)
+    script, *args = sys.argv
+    save_script_result(script, fname, args, tracker_file)
     return fname
 
 
-def save_script_result(script, fname, tracker_file=None):
+def save_script_result(script, fname, args=None, tracker_file=None):
     if tracker_file is None:
         tracker_file = os.path.join(os.getcwd(), 'figure_tracker.csv')
 
     write_header = not os.path.exists(tracker_file)
 
     with open(tracker_file, 'a') as csv_f:
-        writer = csv.DictWriter(csv_f, ('script', 'figure'))
+        writer = csv.DictWriter(csv_f, ('script', 'figure', 'args'))
 
         if write_header:
             writer.writeheader()
 
-        writer.writerow({'script': script, 'figure': fname})
+        writer.writerow({'script': script, 'figure': fname, 'args': ' '.join(args or [])})