Dislocker/dislocker_client_shutdown.py

import os
import json
import tkinter.messagebox
import tkinter
import subprocess
import requests
import tkinter
import sys
import time
import shutil

app_name = "Dislocker"
dislocker_dir = os.path.dirname(os.path.abspath(sys.argv[0]))

os.chdir(dislocker_dir)

sp_startupinfo = subprocess.STARTUPINFO()
sp_startupinfo.dwFlags = subprocess.STARTF_USESHOWWINDOW
sp_startupinfo.wShowWindow = subprocess.SW_HIDE

resource_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), "resource")
config_dir_path = "./config/"
client_config_path = config_dir_path + "client.json"
if not os.path.isfile(client_config_path):
    msgbox = tkinter.messagebox.showerror(title=f"{app_name} | エラー", message=f"設定ファイルが正しく構成されていません。")

elif os.path.isfile(client_config_path):
    with open(client_config_path, "r") as r:
        client_config = json.load(r)

def init(**kwargs):
    task_exist = subprocess.run('tasklist /fi "IMAGENAME eq dislocker_shutdown.exe"', startupinfo=sp_startupinfo, stdout=subprocess.PIPE, text=True)
    if 'dislocker_shutdown.exe' in task_exist.stdout:
        task_count = task_exist.stdout.count("dislocker_shutdown.exe")
        if task_count == 1:
            pass
        else:
            return 1

    if client_config["initial"] == True:
        msgbox = tkinter.messagebox.showerror(title=f"{app_name} | エラー", message=f"設定ファイルが正しく構成されていません。")
        return 2
    else:
        return 0


class App(tkinter.Tk):
    def __init__(self):
        super().__init__()
        
        if client_config["testing"] == True:
            pass
        else:
            pass

        self.geometry("160x100")
        
        self.title(f"{app_name} | シャットダウンを監視")
        self.iconbitmap(default=resource_path + '\\icon\\dislocker_shutdown.ico')

        self.protocol("WM_SAVE_YOURSELF", self.handler_close)

        self.frame = tkinter.Frame(self)
        self.frame.grid(row=0, column=0, sticky='nsew')

        self.withdraw()
    
    def delete_appdata(self, **kwargs):
        process_name = kwargs["process_name"]
        dir_path = kwargs["dir_path"]

        if not os.path.exists(dir_path):
            print(f"エラー: 指定されたディレクトリ {dir_path} が存在しません。")
            return 1

        i = 0
        i_max = 10
        result = 1
        while i != i_max:
            i += 1
            try:
                # プロセスの終了
                subprocess.run(f'taskkill /f /t /im {process_name}', startupinfo=sp_startupinfo, stdout=subprocess.PIPE, text=True)
                print(f"{process_name} を終了しました。")
                time.sleep(0.1)
                # ディレクトリの削除
                shutil.rmtree(dir_path)
                if os.path.isdir(dir_path):
                    pass
                else:
                    print(f"{dir_path} を削除しました。")
                    result = 0
                    i = i_max

            except subprocess.CalledProcessError as e:
                print(f"プロセス終了エラー: {e}")

            except PermissionError as e:
                print(f"権限エラー: {e}")

            except Exception as e:
                print("エラーが発生しました。\nエラー内容:")
                print(f"エラータイプ: {e.__class__.__name__}")
                print(f"エラー引数: {e.args}")
                print(f"エラーメッセージ: {str(e)}")

    def handler_close(self):
        print("停止処理を実行。")
        stop_url = client_config["auth_host_url"] + "/stop"
        stop_json = {
            "pc_number": int(client_config["pc_number"]),
            "pc_uuid": str(client_config["pc_uuid"]),
            "pc_token": str(client_config["pc_token"])
        }
        try:
            responce = requests.post(stop_url, json=stop_json)
            if responce.status_code == 200:
                print("停止処理は成功しました。")
            elif responce.status_code == 401:
                print("認証に失敗しました。")
            else:
                print("内部エラーにより停止処理に失敗しました。")
        except:
            print("ネットワークエラーにより停止処理に失敗しました。")
        
        print("データ削除を実行。")
        try:
            if client_config["eraser"] == True:
                appdata_local = os.path.expandvars("%LOCALAPPDATA%")
                appdata_roaming = os.path.expandvars("%APPDATA%")
                epic_del = self.delete_appdata(process_name="EpicGamesLauncher.exe", dir_path=f"{appdata_local}\\EpicGamesLauncher\\Saved")
                chrome_del = self.delete_appdata(process_name="chrome.exe", dir_path=f"{appdata_local}\\Google\\Chrome\\User Data")
                discord_del = self.delete_appdata(process_name="discord.exe", dir_path=f"{appdata_roaming}\\discord")
                steam_del = self.delete_appdata(process_name="steam.exe", dir_path=f"{appdata_local}\\Steam")
                ea_del = self.delete_appdata(process_name="EADesktop.exe", dir_path=f"{appdata_local}\\Electronic Arts")
                riot_del = self.delete_appdata(process_name="RiotClientServices.exe", dir_path=f"{appdata_local}\\Riot Games\\Riot Client")
                """
                for i in client_config['erase_data'].keys():
                    if i == 'example':
                        pass
                    else:
                        erase_process_name = client_config['erase_data'][i]['process_name']
                        erase_dir_path = client_config['erase_data'][i]['dir_path']
                        erase_del = self.delete_appdata(process_name=erase_process_name, dir_path=erase_dir_path)
                """

            else:
                print("削除処理をスキップ。")
        except:
            print("データ削除に失敗しました。")
        
        self.destroy()
    
    def icon(self):
        self.iconify()


if __name__ == '__main__':
    init_result = init()
    if init_result == 1:
        warning_msgbox = tkinter.messagebox.showwarning(title=f"{app_name} | 多重起動エラー", message=f"すでに {app_name} は実行されています。\n正常に起動しない場合は、既に起動しているプロセスを終了してから、もう一度起動してみてください。")
    elif init_result == 2:
        pass
    else:
        app = App()
        app.mainloop()