mirror of https://github.com/tp4a/teleport
95 lines
2.9 KiB
Python
95 lines
2.9 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
# -*- coding: utf8 -*-
|
||
|
|
|
||
|
|
import os
|
||
|
|
import py_compile
|
||
|
|
import shutil
|
||
|
|
import sys
|
||
|
|
import time
|
||
|
|
import platform
|
||
|
|
|
||
|
|
from . import colorconsole as cc
|
||
|
|
|
||
|
|
rm_file_every_level = ['.pyc', '.pyo']
|
||
|
|
|
||
|
|
PY_VER = platform.python_version_tuple()
|
||
|
|
cpython_mid_name = 'cpython-{}{}'.format(PY_VER[0], PY_VER[1])
|
||
|
|
|
||
|
|
def make(tmp_path):
|
||
|
|
cc.v('Remove all old .pyc/.pyo files...')
|
||
|
|
clean_folder(tmp_path)
|
||
|
|
time.sleep(0.5)
|
||
|
|
cc.v('Compile all .py into .pyo...')
|
||
|
|
compile_files(tmp_path)
|
||
|
|
time.sleep(0.5)
|
||
|
|
cc.v('Remove all .py files...')
|
||
|
|
fix_pyo(tmp_path)
|
||
|
|
time.sleep(0.5)
|
||
|
|
cc.v('Remove all `__pycache__` folders...')
|
||
|
|
remove_cache(tmp_path)
|
||
|
|
|
||
|
|
|
||
|
|
def clean_folder(path):
|
||
|
|
for parent, dir_list, file_list in os.walk(path):
|
||
|
|
for d in dir_list:
|
||
|
|
clean_folder(os.path.join(parent, d))
|
||
|
|
|
||
|
|
for filename in file_list:
|
||
|
|
_, ext = os.path.splitext(filename)
|
||
|
|
# fileNameSplitList = filename.split(".")
|
||
|
|
# ext = fileNameSplitList[len(fileNameSplitList) - 1].lower()
|
||
|
|
if ext in rm_file_every_level:
|
||
|
|
os.remove(os.path.join(parent, filename))
|
||
|
|
|
||
|
|
|
||
|
|
def remove_cache(path):
|
||
|
|
for parent, dir_list, file_list in os.walk(path):
|
||
|
|
for d in dir_list:
|
||
|
|
d = d.lower()
|
||
|
|
if d == '__pycache__':
|
||
|
|
shutil.rmtree(os.path.join(parent, d))
|
||
|
|
continue
|
||
|
|
remove_cache(os.path.join(parent, d))
|
||
|
|
|
||
|
|
|
||
|
|
def compile_files(path):
|
||
|
|
for parent, dir_list, file_list in os.walk(path):
|
||
|
|
for d in dir_list:
|
||
|
|
compile_files(os.path.join(parent, d))
|
||
|
|
|
||
|
|
for filename in file_list:
|
||
|
|
_, ext = os.path.splitext(filename)
|
||
|
|
# fileNameSplitList = filename.split(".")
|
||
|
|
# ext = fileNameSplitList[len(fileNameSplitList) - 1].lower()
|
||
|
|
if ext == '.py':
|
||
|
|
compile_py(os.path.join(parent, filename))
|
||
|
|
|
||
|
|
|
||
|
|
def compile_py(filename):
|
||
|
|
py_compile.compile(filename, optimize=2)
|
||
|
|
|
||
|
|
|
||
|
|
def fix_pyo(path):
|
||
|
|
for parent, dir_list, file_list in os.walk(path):
|
||
|
|
for d in dir_list:
|
||
|
|
fix_pyo(os.path.join(parent, d))
|
||
|
|
|
||
|
|
for filename in file_list:
|
||
|
|
fileNameSplitList = filename.split(".")
|
||
|
|
ext = fileNameSplitList[len(fileNameSplitList) - 1].lower()
|
||
|
|
if ext == 'py':
|
||
|
|
os.remove(os.path.join(parent, filename))
|
||
|
|
elif ext == 'pyo':
|
||
|
|
cpython = fileNameSplitList[len(fileNameSplitList) - 2].lower()
|
||
|
|
if cpython == cpython_mid_name:
|
||
|
|
del fileNameSplitList[len(fileNameSplitList) - 2]
|
||
|
|
else:
|
||
|
|
continue
|
||
|
|
t_name = os.path.abspath(os.path.join(parent, '..', '.'.join(fileNameSplitList)))
|
||
|
|
f_name = os.path.join(parent, filename)
|
||
|
|
shutil.copy(f_name, t_name)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
make(sys.argv[1])
|