diff --git a/dislocker_client.py b/dislocker_client.py index 27c39ce..8b7767d 100644 --- a/dislocker_client.py +++ b/dislocker_client.py @@ -125,6 +125,11 @@ def init(**kwargs): else: tkinter.messagebox.showerror(title=f"{app_name} | 登録時にエラー", message=f"登録時にエラーが発生しました。\nワンタイムパスワードが指定されていません。1個目の引数にPC番号、2個目の引数にワンタイムパスワードを指定して、もう一度お試しください。") return 2 + + if "host_url" in kwargs: + client_config["auth_host_url"] = str(kwargs["host_url"]) + else: + pass register_url = client_config["auth_host_url"] + "/register" register_json = { @@ -546,8 +551,14 @@ if __name__ == '__main__': stop = Stop() stop.run() - elif args[1] == "setup": - init_result = init(pc_number=args[2], onetime=args[3]) + elif args[1] == "setup": + if len(args) == 4: + init_result = init(pc_number=args[2], onetime=args[3]) + elif len(args) == 5: + init_result = init(pc_number=args[2], onetime=args[3], host_url=args[4]) + else: + print("引数エラー。") + error_msgbox = tkinter.messagebox.showerror(title=f"{app_name} | 引数エラー", message=f"引数が多すぎるか、少なすぎます。\n引数がPC番号、ワンタイムパスワード、ホストURLの順で正しく指定されているか確認してください。") if init_result == 1: warning_msgbox = tkinter.messagebox.showwarning(title=f"{app_name} | 多重起動エラー", message=f"すでに {app_name} は実行されています。\n正常に起動しない場合は、既に起動しているプロセスを終了してから、もう一度起動してみてください。") elif init_result == 2: