Added argparse

This commit is contained in:
CDM-Project 2024-09-16 20:34:51 -04:00
parent c5f28e9346
commit 3bd83ac53b

View File

@ -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()
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()