danvics/ytdl-sub
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

separate parsers

Browse source
This commit is contained in:
Jesse Bannon 2021-11-27 17:17:51 +00:00
parent edcca6d745
commit f5738cc413
1 changed files with 18 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
ytdl_subscribe/main.py
View file

@ -12,21 +12,29 @@ parser.add_argument(
help="path to the config yaml, uses subscriptions.yaml if not provided", help="path to the config yaml, uses subscriptions.yaml if not provided",
default="subscriptions.yaml", default="subscriptions.yaml",
) )
parser.add_argument( subparsers = parser.add_subparsers(dest='subparser')
"-s", subscription_parser = subparsers.add_parser('s')
"--subscriptions", subscription_parser.add_argument(
"subscriptions",
metavar="SUB", metavar="SUB",
nargs="+", nargs="*",
help="specific subscriptions to download, downloads all if not provided", help="specific subscriptions to download, downloads all if not provided",
) )
download_parser = subparsers.add_parser('dl')
download_parser.add_argument("-p", "--preset", metavar="PRESET", help="which preset to apply")
download_parser.add_argument("-o", "--output", metavar="OUT", help="output path")
download_parser.add_argument("-r", "--overrides", metavar="NAME:VALUE", nargs="+", help="variable overrides formmatted as name:value")
if __name__ == "__main__": if __name__ == "__main__":
args = parser.parse_args() args = parser.parse_args()
subscriptions = parse_subscriptions(args.config) print(args)
if args.subscriptions:
subscriptions = [s for s in subscriptions if s.name in args.subscriptions]
for s in subscriptions: subscriptions = parse_subscriptions(args.config)
s.extract_info()
if args.subparser == 's':
if args.subscriptions:
subscriptions = [s for s in subscriptions if s.name in args.subscriptions]
for s in subscriptions:
s.extract_info()