Add __name__ == "__main__" construct

This commit is contained in:
Mark Cornick 2022-08-18 00:30:28 -04:00
parent 7f7c3f03d1
commit dd7fa0f5b8
1 changed files with 52 additions and 47 deletions

View File

@ -26,30 +26,31 @@
import re import re
import sys import sys
cue_file = sys.argv[1]
meta_file = re.sub(r"cue$", "meta", cue_file)
try: def cue_to_meta(cue_file):
meta_file = re.sub(r"cue$", "meta", cue_file)
try:
with open(cue_file) as f: with open(cue_file) as f:
cue = f.readlines() cue = f.readlines()
except FileNotFoundError: except FileNotFoundError:
sys.stderr.write("{} not found\n".format(cue_file)) sys.stderr.write("{} not found\n".format(cue_file))
sys.exit(1) sys.exit(1)
except IsADirectoryError: except IsADirectoryError:
sys.stderr.write("{} is a directory\n".format(cue_file)) sys.stderr.write("{} is a directory\n".format(cue_file))
sys.exit(1) sys.exit(1)
except PermissionError: except PermissionError:
sys.stderr.write("No permission to read {}\n".format(cue_file)) sys.stderr.write("No permission to read {}\n".format(cue_file))
sys.exit(1) sys.exit(1)
except OSError: except OSError:
sys.stderr.write("OS error trying to read {}\n".format(cue_file)) sys.stderr.write("OS error trying to read {}\n".format(cue_file))
sys.exit(1) sys.exit(1)
title = None title = None
performer = None performer = None
timestamp = None timestamp = None
try: try:
with open(meta_file, "w") as f: with open(meta_file, "w") as f:
for line in cue: for line in cue:
title_re = re.search(r'^ TITLE "(.*)"$', line) title_re = re.search(r'^ TITLE "(.*)"$', line)
@ -67,12 +68,16 @@ try:
title = None title = None
performer = None performer = None
timestamp = None timestamp = None
except IsADirectoryError: except IsADirectoryError:
sys.stderr.write("{} is a directory\n".format(meta_file)) sys.stderr.write("{} is a directory\n".format(meta_file))
sys.exit(1) sys.exit(1)
except PermissionError: except PermissionError:
sys.stderr.write("No permission to write {}\n".format(meta_file)) sys.stderr.write("No permission to write {}\n".format(meta_file))
sys.exit(1) sys.exit(1)
except OSError: except OSError:
sys.stderr.write("OS error trying to write {}\n".format(meta_file)) sys.stderr.write("OS error trying to write {}\n".format(meta_file))
sys.exit(1) sys.exit(1)
if __name__ == "__main__":
cue_to_meta(sys.argv[1])