Install.py

import win32api
import win32con
import ctypes
import sys
import shutil
import os
import stat
import winreg
import subprocess
from tkinter.ttk import *
import tkinter.filedialog
import tkinter as tk


def add_to_startup(filepath):
    key = winreg.OpenKey(
        winreg.HKEY_CURRENT_USER,
        r'Software\Microsoft\Windows\CurrentVersion\Run',
        0,
        winreg.KEY_ALL_ACCESS
    )
    winreg.SetValueEx(key, 'MyApp', 0, winreg.REG_SZ, filepath)
    winreg.CloseKey(key)


class mainThread:
    def __init__(self):
        self.dir = ""
        self.path_old = os.getcwd()
        self.root = tk.Tk()
        self.root.title("输入希沃目录。")
        x = int((self.root.winfo_screenwidth() - self.root.winfo_reqwidth()) / 2)
        y = int((self.root.winfo_screenheight() - self.root.winfo_reqheight()) / 2)
        self.root.geometry("500x75")
        self.root.geometry("+{}+{}".format(x, y))
        self.root.iconbitmap("ICON.ico")
        self.entry = Entry(self.root)
        self.entry.pack(fill=tk.X, padx=30)
        f1 = Frame(self.root)
        f1.pack(side=tk.BOTTOM, pady=5)
        button = Button(f1, text="确定", command=self.get)
        button.grid(column=0, row=0, padx=3)
        button_dir = Button(f1, text="浏览目录...", command=self.browse_dir)
        button_dir.grid(column=1, row=0, padx=3)
        self.root.mainloop()

        try:
            self.dir = self.dir + "/Main/Assets"
            if os.path.exists("D:\\Seewo"):
                pass

            else:
                os.makedirs("D:\\Seewo")

            shutil.copy("./SplashScreen.png", "D:/Seewo")
            shutil.copy("./Banner.png", "D:/Seewo")
            shutil.copy("./SplashScreen.png", self.dir)
            shutil.copy("./Banner.png",
            os.environ['USERPROFILE'] + "/AppData/Roaming/Seewo/EasiNote5/Resources/Banner/")
            os.chdir(r"C:\ProgramData\Microsoft\Windows\Start Menu\Programs\Startup/")
            os.chmod(r"C:\ProgramData\Microsoft\Windows\Start Menu\Programs\Startup/", stat.S_IWUSR)
            shutil.copy(self.path_old + "/check.exe", "D:/Seewo")
            add_to_startup(r'D:\Seewo\check.exe')
            subprocess.Popen("D:/Seewo/check.exe")
            os.chdir("D:\\Seewo")
            with open("config.cfg", "w") as w:
                w.write(self.dir)

            win32api.MessageBox(0, "完成。", "", win32con.MB_OK | win32con.MB_TOPMOST | win32con.MB_ICONINFORMATION)

        except FileExistsError as f:
            win32api.MessageBox(0, str(f), "", win32con.MB_OK | win32con.MB_TOPMOST)

        except Exception as e:
            win32api.MessageBox(0, str(e), "", win32con.MB_OK | win32con.MB_TOPMOST)


    def get(self):

        if self.entry.get() == "":
            win32api.MessageBox(0, "不允许空目录。", "",
                                win32con.MB_TASKMODAL | win32con.MB_TOPMOST | win32con.MB_ICONWARNING)
            return

        self.dir = self.entry.get()
        self.root.destroy()

    def browse_dir(self):
        self.dir = tkinter.filedialog.askdirectory(mustexist=True, initialdir="C:\Program Files (x86)\Seewo\EasiNote5")
        self.entry.insert(0, self.dir)


if __name__ == '__main__':
    try:
        ctypes.windll.shcore.SetProcessDpiAwareness(1)

    except Exception:
        pass

    try:
        Admin = ctypes.windll.shell32.IsUserAnAdmin()

    except Exception:
        Admin = False

    if not Admin:
        win32api.MessageBox(0, "此程序需要管理员权限。", "错误: 过低的权限", win32con.MB_ICONHAND | win32con.MB_TOPMOST)
        sys.exit()

    mainThread()