61 lines
2.2 KiB
Python
61 lines
2.2 KiB
Python
|
import time
|
||
|
|
import os
|
||
|
|
import shutil
|
||
|
|
from threading import Thread
|
||
|
|
from logger.logger import *
|
||
|
|
|
||
|
|
|
||
|
|
class PKG_Theard(Thread):
|
||
|
|
def __init__(self, disks, barcallback, donecallback):
|
||
|
|
self.disks = disks
|
||
|
|
self.barcallback = barcallback
|
||
|
|
self.donecallback = donecallback
|
||
|
|
Thread.__init__(self, daemon=True)
|
||
|
|
|
||
|
|
def run(self):
|
||
|
|
DISK = self.disks
|
||
|
|
info(f"[install]install to {DISK}")
|
||
|
|
self.barcallback(0)
|
||
|
|
if not os.path.exists(DISK+"/PEinjector"):
|
||
|
|
os.mkdir(DISK+"/PEinjector")
|
||
|
|
time.sleep(0.1)
|
||
|
|
self.barcallback(0.1)
|
||
|
|
info("[install]install main file")
|
||
|
|
if os.path.exists(DISK+"/PEinjector/PEinjector.7z"):
|
||
|
|
os.remove(DISK+"/PEinjector/PEinjector.7z")
|
||
|
|
time.sleep(0.1)
|
||
|
|
self.barcallback(0.2)
|
||
|
|
shutil.copyfile("software/peinjector/dist/PEinjector.7z",
|
||
|
|
DISK+"/PEinjector/PEinjector.7z")
|
||
|
|
time.sleep(0.1)
|
||
|
|
self.barcallback(0.4)
|
||
|
|
time.sleep(0.1)
|
||
|
|
info("[install]write info")
|
||
|
|
if not os.path.exists(DISK+"/PEinjector/disable.txt"):
|
||
|
|
with open(DISK+"/PEinjector/disable.txt", "w") as file:
|
||
|
|
pass
|
||
|
|
self.barcallback(0.5)
|
||
|
|
time.sleep(0.1)
|
||
|
|
shutil.copyfile("software/peinjector/version",
|
||
|
|
DISK+"/PEinjector/VERSION")
|
||
|
|
if not os.path.exists(DISK+"/PEinjector/install"):
|
||
|
|
os.mkdir(DISK+"/PEinjector/install")
|
||
|
|
self.barcallback(0.6)
|
||
|
|
if not os.path.exists(DISK+"/PEinjector/package"):
|
||
|
|
os.mkdir(DISK+"/PEinjector/package")
|
||
|
|
ns = 0
|
||
|
|
for i in os.listdir("software/peinjector/package"):
|
||
|
|
if os.path.exists(DISK+"/PEinjector/package/"+i):
|
||
|
|
shutil.rmtree(DISK+"/PEinjector/package/"+i)
|
||
|
|
shutil.copytree("software/peinjector/package/" +
|
||
|
|
i, DISK+"/PEinjector/package/"+i)
|
||
|
|
ns += 1
|
||
|
|
info(
|
||
|
|
f"[install]install pkg {ns}/{len(os.listdir('software/peinjector/package'))}")
|
||
|
|
self.barcallback(
|
||
|
|
0.6+ns/len(os.listdir("software/peinjector/package"))*0.4)
|
||
|
|
time.sleep(0.1)
|
||
|
|
info("[install]install finish")
|
||
|
|
self.barcallback(1)
|
||
|
|
self.donecallback()
|