# Compress the files
if sys.argv[1] == '-c':
print("Creating the archive..")
if not (ARCHIVE_NAME[len(ARCHIVE_NAME) - 2:] == 'gz'):
ARCHIVE_NAME += ".gz"
Converters.correct_last_line_file(GPS_FILE)
Converters.correct_last_line_file(CAN_FILE)
if not os.path.isfile(OUTPUT_CAN_FILE):
Converters.convert_canframe_file(CAN_FILE, OUTPUT_CAN_FILE)
if not os.path.isfile(OUTPUT_CAMERA_FILE):
Converters.convert_video_to_mp4(CAMERA_FILE, OUTPUT_CAMERA_FILE)
c['gps'] = [l for l in open(GPS_FILE)]
c['can'] = [l for l in open(OUTPUT_CAN_FILE)]
c['camera'] = open(OUTPUT_CAMERA_FILE, 'rb').read()
with gzip.open(ARCHIVE_NAME, 'wb') as f:
f.write(pickle.dumps(c))
print("Archive created in " + ARCHIVE_NAME)
# Decompress the file
else:
if not os.path.exists(OUTPUT_FOLDER):
os.makedirs(OUTPUT_FOLDER)
print("Extracting the archvive..")
with gzip.open(ARCHIVE_NAME, 'rb') as f:
