152 lines
5.8 KiB
Python
152 lines
5.8 KiB
Python
import os
|
|
import json
|
|
import tkinter.messagebox
|
|
import customtkinter
|
|
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("100x100")
|
|
|
|
self.title(f"{app_name} | てすと")
|
|
self.iconbitmap(default=resource_path + '\\icon\\dislocker.ico')
|
|
|
|
self.protocol("WM_DELETE_WINDOW", self.handler_close)
|
|
|
|
self.frame = tkinter.Frame(self)
|
|
self.frame.grid(row=0, column=0, sticky='nsew')
|
|
|
|
self.iconify()
|
|
|
|
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("停止処理を実行。")
|
|
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")
|
|
else:
|
|
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("ネットワークエラーにより停止処理に失敗しました。")
|
|
finally:
|
|
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()
|