96 lines
2.9 KiB
Python
96 lines
2.9 KiB
Python
|
#!/usr/bin/python
|
|||
|
|
# -*- coding: utf-8 -*-
|
|||
|
|
# @version : 1.0
|
|||
|
|
# @Create Time : 2022/12/9 15:27
|
|||
|
|
# @File : main.py
|
|||
|
|
# @IDE : PyCharm
|
|||
|
|
# @desc : 简要说明
|
|||
|
|
|
|||
|
|
import datetime
|
|||
|
|
import os.path
|
|||
|
|
from application.settings import BASE_DIR
|
|||
|
|
|
|||
|
|
|
|||
|
|
class CreateApp:
|
|||
|
|
|
|||
|
|
APPS_ROOT = os.path.join(BASE_DIR, "apps")
|
|||
|
|
SCRIPT_DIR = os.path.join(BASE_DIR, 'scripts', 'create_app')
|
|||
|
|
|
|||
|
|
def __init__(self, path: str):
|
|||
|
|
"""
|
|||
|
|
:param path: app 路径，根目录为apps，填写apps后面路径即可，例子：vadmin/auth
|
|||
|
|
"""
|
|||
|
|
self.app_path = os.path.join(self.APPS_ROOT, path)
|
|||
|
|
self.path = path
|
|||
|
|
|
|||
|
|
def run(self):
|
|||
|
|
"""
|
|||
|
|
自动创建初始化 APP 结构，如何该路径已经存在，则不执行
|
|||
|
|
"""
|
|||
|
|
if self.exist(self.app_path):
|
|||
|
|
print(f"{self.app_path} 已经存在，无法自动创建，请删除后，重新执行。")
|
|||
|
|
return False
|
|||
|
|
print("开始生成 App 目录：", self.path)
|
|||
|
|
path = []
|
|||
|
|
for item in self.path.split("/"):
|
|||
|
|
path.append(item)
|
|||
|
|
self.create_pag(os.path.join(self.APPS_ROOT, *path))
|
|||
|
|
self.create_pag(os.path.join(self.app_path, "models"))
|
|||
|
|
self.create_pag(os.path.join(self.app_path, "params"))
|
|||
|
|
self.create_pag(os.path.join(self.app_path, "schemas"))
|
|||
|
|
self.generate_file("views.py")
|
|||
|
|
self.generate_file("crud.py")
|
|||
|
|
print("App 目录生成结束", self.app_path)
|
|||
|
|
|
|||
|
|
def create_pag(self, path: str) -> None:
|
|||
|
|
"""
|
|||
|
|
创建 python 包
|
|||
|
|
|
|||
|
|
:param path: 绝对路径
|
|||
|
|
"""
|
|||
|
|
if self.exist(path):
|
|||
|
|
return
|
|||
|
|
os.makedirs(path)
|
|||
|
|
now = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
|||
|
|
params = {
|
|||
|
|
"create_datetime": now,
|
|||
|
|
"filename": "__init__.py",
|
|||
|
|
"desc": "初始化文件"
|
|||
|
|
}
|
|||
|
|
self.create_file(os.path.join(path, "__init__.py"), "init.py", **params)
|
|||
|
|
|
|||
|
|
def generate_file(self, name: str) -> None:
|
|||
|
|
"""
|
|||
|
|
创建文件
|
|||
|
|
"""
|
|||
|
|
now = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
|||
|
|
params = {
|
|||
|
|
"create_datetime": now,
|
|||
|
|
}
|
|||
|
|
self.create_file(os.path.join(self.app_path, name), name, **params)
|
|||
|
|
|
|||
|
|
def create_file(self, filepath: str, name: str, **kwargs):
|
|||
|
|
"""
|
|||
|
|
创建文件
|
|||
|
|
"""
|
|||
|
|
with open(filepath, "w", encoding="utf-8") as f:
|
|||
|
|
content = self.__get_template(name)
|
|||
|
|
f.write(content.format(**kwargs))
|
|||
|
|
|
|||
|
|
@classmethod
|
|||
|
|
def exist(cls, path) -> bool:
|
|||
|
|
"""
|
|||
|
|
判断路径是否已经存在
|
|||
|
|
"""
|
|||
|
|
return os.path.exists(path)
|
|||
|
|
|
|||
|
|
def __get_template(self, name: str) -> str:
|
|||
|
|
"""
|
|||
|
|
获取模板内容
|
|||
|
|
"""
|
|||
|
|
template = open(os.path.join(self.SCRIPT_DIR, "template", name), 'r')
|
|||
|
|
content = template.read()
|
|||
|
|
template.close()
|
|||
|
|
return content
|
|||
|
|
|