ceda_ei
/
redacted.life
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

[SSG] Isolate argument parsing.

This commit is contained in:
Ceda EI 2020-01-27 17:14:38 +05:30
parent 5b13ce2a2f
commit 917d3addb6
1 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
nova.py
View File

@ -94,13 +94,13 @@ class EpisodeList(UserList):
class Episode: class Episode:
"Represents one episode of podcast" "Represents one episode of podcast"
def __init__(self, date, slug, title, show_notes, video_src, audo_src): def __init__(self, date, slug, title, show_notes, video_src, audio_src):
self.date = date self.date = date
self.slug = slug self.slug = slug
self.title = title.strip() self.title = title.strip()
self.show_notes = markdown.markdown(show_notes) self.show_notes = markdown.markdown(show_notes)
self.video = video_src self.video = video_src
self.audio = audo_src self.audio = audio_src
def render(self, template, thumbnail_src): def render(self, template, thumbnail_src):
"Renders the Episode with the given template" "Renders the Episode with the given template"
@ -118,15 +118,21 @@ class Episode:
subprocess.run(args, check=False) subprocess.run(args, check=False)
def main(): def parse_args():
"Main method" "Parses arguments"
root = path.dirname(sys.argv[0]).rstrip("/") + "/"
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("input_dir", help="Input directory") parser.add_argument("input_dir", help="Input directory")
parser.add_argument("output_dir", help="Output directory") parser.add_argument("output_dir", help="Output directory")
args = parser.parse_args() args = parser.parse_args()
input_dir = args.input_dir.rstrip("/") + "/" input_dir = args.input_dir.rstrip("/") + "/"
output_dir = args.output_dir.rstrip("/") + "/" output_dir = args.output_dir.rstrip("/") + "/"
return input_dir, output_dir
def main():
"Main method"
root = path.dirname(sys.argv[0]).rstrip("/") + "/"
input_dir, output_dir = parse_args()
# Input validation # Input validation
paths = [ paths = [
@ -139,8 +145,8 @@ def main():
print("Invalid Input", file=sys.stderr) print("Invalid Input", file=sys.stderr)
return return
if not path.isdir(args.output_dir): if not path.isdir(output_dir):
os.mkdir(args.output_dir) os.mkdir(output_dir)
env = jinja2.Environment( env = jinja2.Environment(
loader=jinja2.FileSystemLoader(root), loader=jinja2.FileSystemLoader(root),