initのテスト

dislocker_client.py

@@ -14,6 +14,39 @@ import threading
 import signal
 import sys
 
+def init(**kwargs):
+    app_name = "Dislocker"
+    if "pc_number" in kwargs:
+        pc_number = int(kwargs["pc_number"])
+    else:
+        pc_number = 1
+    config_dir_path = "./config/"
+    client_config_path = config_dir_path + "client.json"
+    if not os.path.isfile(client_config_path):
+        if not os.path.isdir(config_dir_path):
+            os.mkdir(config_dir_path)
+
+        client_config = {
+            "initial": 1,
+            "auth_host_url": "http://localhost",
+            "pc_number": 1,
+            "master_password_hash": "",
+            "testing": 0
+        }
+
+    elif os.path.isfile(client_config_path):
+        with open(client_config_path, "r") as r:
+            client_config = json.load(r)
+
+    if client_config["initial"] == 1:
+        master_password = master_password_gen()
+        msgbox = tkinter.messagebox.showinfo(title=f"{app_name} | 初回起動を検出", message=f"初回起動のようです。\nマスターパスワードを記録しておいてください。\nこれ以降二度と表示されることはないでしょう。\n\n{master_password["password"]}\n\nまた、認証先サーバーの接続先を指定してください。ロックを解除できなくなります。")
+        client_config["master_password_hash"] = master_password["password_hash"]
+        client_config["initial"] = 0
+        client_config["pc_number"] = pc_number
+        with open(client_config_path, "w") as w:
+            json.dump(client_config, w, indent=4)
+
 class App(customtkinter.CTk):
     def __init__(self):
         super().__init__()
@@ -42,7 +75,7 @@ class App(customtkinter.CTk):
             keyboard.block_key(i)
 
     def block_taskmgr(self):
-        block = subprocess.run(['reg', 'add', 'HKEY_CURRENT_USER\\Software\\Microsoft\\Windows\\CurrentVersion\\Policies\\System', '/v', 'DisableTaskMgr', '/t', 'REG_DWORD', '/d', 'f', '1'])
+        block = subprocess.run(['reg', 'add', 'HKEY_CURRENT_USER\\Software\\Microsoft\\Windows\\CurrentVersion\\Policies\\System', '/v', 'DisableTaskMgr', '/t', 'REG_DWORD', '/d', '1', '/f'])
         print(block)
 
     def unlock_taskmgr(self):
@@ -289,47 +322,19 @@ def master_password_gen():
     result = {"password": password, "password_hash": password_hash}
     return result
 
-app_name = "Dislocker"
 
-config_dir_path = "./config/"
-client_config_path = config_dir_path + "client.json"
-if not os.path.isfile(client_config_path):
-    if not os.path.isdir(config_dir_path):
-        os.mkdir(config_dir_path)
-
-    client_config = {
-        "initial": 1,
-        "auth_host_url": "http://localhost",
-        "pc_number": 1,
-        "master_password_hash": "",
-        "testing": 0
-
-    }
-    with open(client_config_path, "w") as w:
-        json.dump(client_config, w, indent=4)
-elif os.path.isfile(client_config_path):
-    with open(client_config_path, "r") as r:
-        client_config = json.load(r)
-        
-        
-if client_config["initial"] == 1:
-    master_password = master_password_gen()
-    msgbox = tkinter.messagebox.showinfo(title=f"{app_name} | 初回起動を検出", message=f"初回起動のようです。\nマスターパスワードを記録しておいてください。\nこれ以降二度と表示されることはないでしょう。\n\n{master_password["password"]}\n\nまた、認証先サーバーの接続先を指定してください。ロックを解除できなくなります。")
-    client_config["master_password_hash"] = master_password["password_hash"]
-    client_config["initial"] = 0
-    with open(client_config_path, "w") as w:
-        json.dump(client_config, w, indent=4)
-
-else:
 if __name__ == '__main__':
     args = sys.argv
     if len(args) >= 2:
         if args[1] == "stop":
             monitor = Monitor()
             monitor.signal_handler()
+        elif args[1] == "setup": 
+            init(pc_number=args[2])
         else:
             print("引数エラー。")
     else:
+        init()
         app = App()
         app.protocol("WM_DELETE_WINDOW", app.handler_close)
         app.mainloop()