diff --git a/change.py b/change.py index 0048a18..7c6ff11 100644 --- a/change.py +++ b/change.py @@ -2,14 +2,14 @@ import ctypes import sys import os from time import sleep +import shelve path = os.getcwd() -t = False +with shelve.open('main.db') as db: + bin_name = db['bin_name'] + db.close() print(path) -def exit1(): - sys.exit() - def is_admin(): # 检查当前是否有管理员权限 try: @@ -19,26 +19,41 @@ def is_admin(): if is_admin(): # 如果已经是管理员,执行主要的逻辑 - while t==False: + while True: print("已获得管理员权限") + print(f"当前模式(1为cs原版,2为csso):{bin_name}") print("请选择模式") print("1.改为csso 2.改回原版 3.退出") ip = input(int('>>>')) # 逻辑系统bin2为csso文件 bin1为cs原版文件 bin为现在使用的游戏文件 - if ip == 1: + if ip == int(1): + if bin_name == int(1): + print('错误,你已是原版') + sleep(1.5) + sys.exit() os.system(f"ren {path}/bin {path}/bin1") os.system(f"ren {path}/bin2 {path}/bin") + with shelve.open('main.db') as db: + db['bin_name'] = int(1) + db.close() print("完成!") sleep(1.85) - exit1() - elif ip == 2: + sys.exit() + elif ip == int(2): + if bin_name == int(2): + print("错误,你已是csso!") + sleep(1.5) + sys.exit() os.system(f"ren {path}/bin {path}/bin2") os.system(f"ren {path}/bin1 {path}/bin") + with shelve.open('main.db') as db: + db['bin_name'] = int(2) + db.close() print("完成!") sleep(1.55) - exit1() - elif ip == 3: + sys.exit() + elif ip == int(3): sys.exit() else: print("输入错误!")