raw_parser: Prompt if filename not specified

master
Richard Meadows 2015-08-01 17:41:22 +01:00
rodzic 4d6979981f
commit 61a17fa643
1 zmienionych plików z 5 dodań i 7 usunięć

Wyświetl plik

@ -104,11 +104,10 @@ def extract_raw_data(line):
#-------------------------------------------------------------------------------
if len(sys.argv) != 2:
print 'Usage: python raw_parser.py rawdata.txt'
sys.exit()
file_name = sys.argv[1]
if len(sys.argv) == 2:
file_name = sys.argv[1]
else:
file_name = raw_input("File to read (rawdata.txt): ") or "rawdata.txt"
with open(file_name, 'r') as data_file:
data = []
@ -137,4 +136,3 @@ with open(file_name, 'r') as data_file:
# Print data
for datum in data:
print "%s: %s, %s" % ((str(datum['time']),) + datum['coords'][:2])