From dd7fa0f5b864f234aff849c0758964f664a67399 Mon Sep 17 00:00:00 2001 From: Mark Cornick Date: Thu, 18 Aug 2022 00:30:28 -0400 Subject: [PATCH] Add __name__ == "__main__" construct --- cue_to_meta.py | 99 ++++++++++++++++++++++++++------------------------ 1 file changed, 52 insertions(+), 47 deletions(-) diff --git a/cue_to_meta.py b/cue_to_meta.py index f108dec..75b6689 100755 --- a/cue_to_meta.py +++ b/cue_to_meta.py @@ -26,53 +26,58 @@ import re import sys -cue_file = sys.argv[1] -meta_file = re.sub(r"cue$", "meta", cue_file) -try: - with open(cue_file) as f: - cue = f.readlines() -except FileNotFoundError: - sys.stderr.write("{} not found\n".format(cue_file)) - sys.exit(1) -except IsADirectoryError: - sys.stderr.write("{} is a directory\n".format(cue_file)) - sys.exit(1) -except PermissionError: - sys.stderr.write("No permission to read {}\n".format(cue_file)) - sys.exit(1) -except OSError: - sys.stderr.write("OS error trying to read {}\n".format(cue_file)) - sys.exit(1) +def cue_to_meta(cue_file): + meta_file = re.sub(r"cue$", "meta", cue_file) -title = None -performer = None -timestamp = None + try: + with open(cue_file) as f: + cue = f.readlines() + except FileNotFoundError: + sys.stderr.write("{} not found\n".format(cue_file)) + sys.exit(1) + except IsADirectoryError: + sys.stderr.write("{} is a directory\n".format(cue_file)) + sys.exit(1) + except PermissionError: + sys.stderr.write("No permission to read {}\n".format(cue_file)) + sys.exit(1) + except OSError: + sys.stderr.write("OS error trying to read {}\n".format(cue_file)) + sys.exit(1) -try: - with open(meta_file, "w") as f: - for line in cue: - title_re = re.search(r'^ TITLE "(.*)"$', line) - performer_re = re.search(r'^ PERFORMER "(.*)"$', line) - index_re = re.search(r"^ INDEX 01 (.*)$", line) - if title_re: - title = title_re[1] - if performer_re: - performer = performer_re[1] - if index_re: - time_re = re.search(r"^(..):(..):..$", index_re[1]) - timestamp = "00{}{}".format(time_re[1], time_re[2]) - if title and performer and timestamp: - f.write("{}:{} - {}\n".format(timestamp, performer, title)) - title = None - performer = None - timestamp = None -except IsADirectoryError: - sys.stderr.write("{} is a directory\n".format(meta_file)) - sys.exit(1) -except PermissionError: - sys.stderr.write("No permission to write {}\n".format(meta_file)) - sys.exit(1) -except OSError: - sys.stderr.write("OS error trying to write {}\n".format(meta_file)) - sys.exit(1) + title = None + performer = None + timestamp = None + + try: + with open(meta_file, "w") as f: + for line in cue: + title_re = re.search(r'^ TITLE "(.*)"$', line) + performer_re = re.search(r'^ PERFORMER "(.*)"$', line) + index_re = re.search(r"^ INDEX 01 (.*)$", line) + if title_re: + title = title_re[1] + if performer_re: + performer = performer_re[1] + if index_re: + time_re = re.search(r"^(..):(..):..$", index_re[1]) + timestamp = "00{}{}".format(time_re[1], time_re[2]) + if title and performer and timestamp: + f.write("{}:{} - {}\n".format(timestamp, performer, title)) + title = None + performer = None + timestamp = None + except IsADirectoryError: + sys.stderr.write("{} is a directory\n".format(meta_file)) + sys.exit(1) + except PermissionError: + sys.stderr.write("No permission to write {}\n".format(meta_file)) + sys.exit(1) + except OSError: + sys.stderr.write("OS error trying to write {}\n".format(meta_file)) + sys.exit(1) + + +if __name__ == "__main__": + cue_to_meta(sys.argv[1])