Explorar o código

Add command line argument to csv printing script

Mario Hüttel %!s(int64=3) %!d(string=hai) anos
pai
achega
6be539f738
Modificáronse 1 ficheiros con 6 adicións e 1 borrados
  1. 6 1
      print_labels_from_csv.py

+ 6 - 1
print_labels_from_csv.py

@@ -2,9 +2,14 @@ import pandas as pd
 import shimatta_label.label_image as li
 import shimatta_label.brother_ql_wrapper as ql_wrapper
 import os
+import sys
 
 medir = os.path.dirname(__file__)
-example_data_path = os.path.join(medir, 'labels.csv')
+
+if len(sys.argv) > 2:
+    example_data_path = sys.argv[1]
+else:    
+    example_data_path = os.path.join(medir, 'labels.csv')
 
 df = pd.read_csv(example_data_path, converters={i: str for i in range(3)})