Added argparse
This commit is contained in:
parent
c5f28e9346
commit
3bd83ac53b
20
CDRM-Keys.py
20
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()
|
||||
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()
|
Loading…
Reference in New Issue
Block a user