From 1532a06cb4cea0c19eb6612306f33ef29ee584a4 Mon Sep 17 00:00:00 2001 From: y1ndan Date: Mon, 4 Jan 2021 14:13:57 +0800 Subject: [PATCH] refactor: use argparse to get args --- genshin.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/genshin.py b/genshin.py index c199255..31641b0 100755 --- a/genshin.py +++ b/genshin.py @@ -1,6 +1,7 @@ #!/usr/bin/env python3 import requests +import argparse import json import uuid import logging @@ -272,10 +273,11 @@ def notify(sckey, status, message): if __name__ == '__main__': - secret = input().strip().split('#') - secret.append('') - cookie = secret[0] - sckey = secret[1] + parser = argparse.ArgumentParser() + parser.add_argument('-c', '--cookie', required=True, type=str, default='') + parser.add_argument('-s', '--sckey', type=str, default='') + args = parser.parse_args() + sckey = args.sckey - Sign(cookie).run() + Sign(args.cookie).run()