__make_addon_zip.py

import os
from pathlib import Path
import zipfile
import shutil

parent_path = Path(__file__).parent


def get_tg_dir() -> Path:
    tg_dir = parent_path.joinpath(parent_path.name)
    if tg_dir.exists():
        shutil.rmtree(tg_dir)
    tg_dir.mkdir()

    return tg_dir


def copy_files() -> Path:
    tg_dir = get_tg_dir()
    sub_dir = tg_dir.joinpath(parent_path.name)
    sub_dir.mkdir()

    for root, dirs, files in os.walk(Path(__file__).parent):
        if '__pycache__' in dirs:
            pycache_folder = os.path.join(root, '__pycache__')
            shutil.rmtree(pycache_folder)
            print(f'Removed {pycache_folder}')

    for file in parent_path.glob('*'):
        if file.is_dir():
            if file.name == parent_path.name: continue
            if file.name.startswith('__') or file.name.startswith('.'): continue

            shutil.copytree(file, sub_dir.joinpath(file.name))

        elif file.is_file():
            if file.name == __file__: continue

            shutil.copy(file, sub_dir.joinpath(file.name))

    return tg_dir


def zip_dir() -> None:
    tg_dir = copy_files()

    zip_file = parent_path.joinpath(f'{parent_path.name}.zip')
    if zip_file.exists():
        os.remove(zip_file)
    # utf-8 encoding
    with zipfile.ZipFile(zip_file, 'w', zipfile.ZIP_DEFLATED, allowZip64=True) as zip:
        for root, dirs, files in os.walk(tg_dir):
            for file in files:
                # if file name is this file, skip
                if file == os.path.basename(__file__): continue
                zip.write(os.path.join(root, file), arcname=os.path.join(root, file).replace(str(tg_dir), ''))
    print(f'Zip file: {zip_file}')
    shutil.rmtree(tg_dir)
    print('Remove temp dir')


if __name__ == '__main__':
    copy_files()
    zip_dir()