修复找不到TEMP文件问题

This commit is contained in:
cxykevin 2024-02-10 22:00:46 +08:00
parent 9c0f122beb
commit 03611dbd8f
2 changed files with 3 additions and 3 deletions

View File

@ -110,8 +110,8 @@ def start(pkg_name: str, data_list: list, cmd: str) -> list:
def save_action(actions_list: list) -> None:
tmp_filename = "PEinjector.tmp"
log.info(f"save action list to {config.TEMP_DIR.replace(
'{TEMP}', os.environ['TEMP'].replace("\\", "/"))+'/'+tmp_filename}")
with open(config.TEMP_DIR.replace("{TEMP}", os.environ["TEMP"])+"/"+tmp_filename, "wb") as file:
'{TEMP}', "X:\\PEinjector")+'\\'+tmp_filename}")
with open(config.TEMP_DIR.replace("{TEMP}", "X:\\PEinjector")+"\\"+tmp_filename, "wb") as file:
pickle.dump(actions_list, file)
return tmp_filename

View File

@ -68,7 +68,7 @@ while 1:
# load action list
log.info("load action \"onload\"")
with open(config.TEMP_DIR.replace("{TEMP}", os.environ["TEMP"])+"/"+"PEinjector.tmp", "rb") as file:
with open(config.TEMP_DIR.replace("{TEMP}", "X:\\PEinjector")+"\\"+"PEinjector.tmp", "rb") as file:
lists = pickle.load(file)
alog = action.do_action(lists)
with open(config.ACTIONLOGPATH.replace("{DISK}", utils.find_disk()), "w") as file: