Move to saner defaults:

- fix idiot path concat problem i had before
- use @mrled style main and parsed.
master
jowj 4 years ago
parent 835c4c872d
commit 361194c4be

@ -1,40 +1,55 @@
import os
import sys
import argparse
import subprocess
import shutil
from pathlib import Path
import logging
if __name__ == '__main__':
MUSIC_DEST =
VIDEO_DEST =
logger = logging.getLogger(__name__)
def main(*args, **kwargs):
"""Main program entrypoint
"""
parser = argparse.ArgumentParser()
parser.add_argument('--type', '-t', help="music or video?", type=str)
parser.add_argument('--url', '-u', help="link?", type=str)
parser.add_argument("--debug", "-d", action="store_true")
parser.add_argument("--logfile", "-l", type=str) # should this be File?
parsed = parser.parse_args()
loglevel = logging.INFO
PARSER = argparse.ArgumentParser()
PARSER.add_argument('-t', '--type', help="aud or vid?", type=str)
PARSER.add_argument('-u', '--url', help="link?", type=str)
if parsed.debug:
loglevel = logging.DEBUG
logger.setLevel(loglevel)
ARGS = PARSER.parse_args()
if parsed.logfile:
fh = logging.FileHandler(parsed.logfile)
fh.setLevel(loglevel)
fh.setFormatter(logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s'))
logger.addHandler(fh)
logger.debug("Configured logging, nice")
TEMP_PATH = os.getcwd() + '/temp'
os.mkdir(TEMP_PATH)
os.chdir(TEMP_PATH)
temp_path = os.getcwd() + '/temp'
if os.path.exists(temp_path):
os.chdir(temp_path)
else:
os.mkdir(temp_path)
os.chdir(temp_path)
if ARGS.type == 'music':
if parsed.type == 'music':
# this will download a video an convert to just mp3
URL = ARGS.url
DEST_ROOT = MUSIC_DEST
url = parsed.url
subprocess.call(["youtube-dl", "-i", "--extract-audio",
"--audio-format", "mp3", "--yes-playlist",
"-o", "%(playlist_title)s/%(title)s.%(ext)s",
URL])
if ARGS.type == 'video':
url])
if parsed.type == 'video':
# this actually downloads the videos in a playlist to seperate files.
URL = ARGS.url
DEST_ROOT = VIDEO_DEST
url = parsed.url
subprocess.call(["youtube-dl", "-i", "-f"
"mp4", "-o", "%(playlist_title)s/%(title)s.%(ext)s",
URL])
url])
PLAYLIST = os.listdir(".")
DEST = DEST_ROOT + "/" + PLAYLIST[0]
shutil.move(PLAYLIST[0], DEST)
os.rmdir(TEMP_PATH)
if __name__ == '__main__':
sys.exit(main(*sys.argv))

Loading…
Cancel
Save