75 lines
2.2 KiB
Python
75 lines
2.2 KiB
Python
import ctypes
|
|
import sys
|
|
import os
|
|
from time import sleep
|
|
import shelve
|
|
|
|
def get_path(relative_path):
|
|
try:
|
|
base_path = sys._MEIPASS # pyinstaller打包后的路径
|
|
except AttributeError:
|
|
base_path = os.path.abspath(".") # 当前工作目录的路径
|
|
|
|
return os.path.normpath(os.path.join(base_path, relative_path)) # 返回实际路径
|
|
|
|
path = os.getcwd()
|
|
with shelve.open('main.db') as db:
|
|
bin_name = db['bin_name']
|
|
if bin_name == None:
|
|
os.startfile(get_path('load.exe'))
|
|
db.close()
|
|
print(path)
|
|
|
|
def is_admin():
|
|
# 检查当前是否有管理员权限
|
|
try:
|
|
return ctypes.windll.shell32.IsUserAnAdmin()
|
|
except:
|
|
return False
|
|
|
|
if is_admin():
|
|
# 如果已经是管理员，执行主要的逻辑
|
|
print("已获得管理员权限")
|
|
print(f"当前模式(1为cs原版,2为csso):{bin_name}")
|
|
print("请选择模式")
|
|
print("1.改为csso 2.改回原版 3.退出")
|
|
ip = input('>>>')
|
|
|
|
# 逻辑系统bin2为csso文件 bin1为cs原版文件 bin为现在使用的游戏文件
|
|
if ip == '1':
|
|
if bin_name == '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'] = '1'
|
|
db.close()
|
|
print("完成!")
|
|
sleep(1.85)
|
|
sys.exit()
|
|
elif ip == '2':
|
|
if bin_name == '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'] = '2'
|
|
db.close()
|
|
print("完成!")
|
|
sleep(1.55)
|
|
sys.exit()
|
|
elif ip == '3':
|
|
sys.exit()
|
|
else:
|
|
print("输入错误!")
|
|
|
|
else:
|
|
# 如果没有管理员权限，重新以管理员权限运行脚本
|
|
print("请求管理员权限...")
|
|
# 使用 `ctypes` 模块提升权限，重新运行脚本
|
|
ctypes.windll.shell32.ShellExecuteW(None, "runas", sys.executable, " ".join(sys.argv), None, 1)
|