diff --git a/CDRM-Keys.py b/CDRM-Keys.py index 6b890b5..505262e 100644 --- a/CDRM-Keys.py +++ b/CDRM-Keys.py @@ -1,8 +1,20 @@ # Import dependencies import CDRM_Modules +import argparse -# Run startup checks -CDRM_Modules.Startup_Checks.run_startup_checks(startup=True) +def setup_arg_parser(): + parser = argparse.ArgumentParser(description="CDRM-Keys") + parser.add_argument('--pssh', action='store_true', help='Parse PSSH from MPD/HLS') + return parser -# Run main menu -CDRM_Modules.Main_Menu.main_menu() \ No newline at end of file +def main(): + parser = setup_arg_parser() + args = parser.parse_args() + + if args.pssh: + CDRM_Modules.Get_PSSH.get_pssh(mpd_or_m3u8_url=input("MPD/HLS URL: ")) + else: + CDRM_Modules.Main_Menu.main_menu() + +if __name__ == "__main__": + main() \ No newline at end of file