62 lines
2.0 KiB
Python
62 lines
2.0 KiB
Python
|
#!/usr/bin/env python
|
|||
|
|
# -*- coding: UTF-8 -*-
|
|||
|
|
"""
|
|||
|
|
@Project ：IoD_data_analysis_tool
|
|||
|
|
@File ：project_util.py
|
|||
|
|
@IDE ：PyCharm
|
|||
|
|
@Author ：rengengchen
|
|||
|
|
@Time ：2022/9/15 9:45
|
|||
|
|
"""
|
|||
|
|
import compileall
|
|||
|
|
import os
|
|||
|
|
import re
|
|||
|
|
import shutil
|
|||
|
|
from os.path import join
|
|||
|
|
|
|||
|
|
from lib.analysis_package.utils.file_util import walk
|
|||
|
|
|
|||
|
|
re_pyc = re.compile(r'cpython-\d+\.')
|
|||
|
|
|
|||
|
|
|
|||
|
|
def compile_project(source, target=None):
|
|||
|
|
"""
|
|||
|
|
编译项目为pyc文件到指定目录
|
|||
|
|
@param source: 项目路径
|
|||
|
|
@param target: 编译文件存放路径
|
|||
|
|
"""
|
|||
|
|
source = os.path.abspath(source)
|
|||
|
|
if target is None:
|
|||
|
|
target = source
|
|||
|
|
else:
|
|||
|
|
target = os.path.abspath(target)
|
|||
|
|
compileall.compile_dir(source)
|
|||
|
|
pycache_paths = set()
|
|||
|
|
if target == source:
|
|||
|
|
for root, dirname, filename in walk(source):
|
|||
|
|
if root[-11:] == '__pycache__':
|
|||
|
|
pycache_paths.add(root)
|
|||
|
|
shutil.move(join(root, filename), join(root, '../', re_pyc.sub('', filename)))
|
|||
|
|
if filename and filename.endswith('py'):
|
|||
|
|
os.remove(join(root, filename))
|
|||
|
|
else:
|
|||
|
|
if target is None:
|
|||
|
|
target = join(source, 'dist')
|
|||
|
|
len_t = len(target)
|
|||
|
|
for root, dirname, filename in walk(source):
|
|||
|
|
t_root = root.replace(source, target)
|
|||
|
|
if target == root[:len_t]:
|
|||
|
|
continue
|
|||
|
|
if dirname and dirname != '__pycache__':
|
|||
|
|
t_root = join(t_root, dirname)
|
|||
|
|
if not os.path.exists(t_root) and join(source, dirname) != target:
|
|||
|
|
os.makedirs(t_root)
|
|||
|
|
elif filename and not filename.endswith('py'):
|
|||
|
|
if root[-11:] == '__pycache__':
|
|||
|
|
pycache_paths.add(root)
|
|||
|
|
t_root = t_root[:-11]
|
|||
|
|
shutil.move(join(root, filename), join(t_root, re_pyc.sub('', filename)))
|
|||
|
|
else:
|
|||
|
|
shutil.copyfile(join(root, filename), join(t_root, filename))
|
|||
|
|
for p in pycache_paths:
|
|||
|
|
os.rmdir(p)
|