新功能(一键创建app命令): 一键创建app,并注册到settings和urls中
创建命令: python manage.py createapp app名 python manage.py createapp app01 app02 ...pull/28/head
parent
d8bf287c85
commit
5ce22a5a90
|
@ -0,0 +1,75 @@
|
||||||
|
import logging
|
||||||
|
import os
|
||||||
|
import shutil
|
||||||
|
|
||||||
|
from django.core.management.base import BaseCommand
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
from application.settings import BASE_DIR
|
||||||
|
|
||||||
|
|
||||||
|
class Command(BaseCommand):
|
||||||
|
"""
|
||||||
|
创建App命令:
|
||||||
|
python manage.py createapp app名
|
||||||
|
python manage.py createapp app01 app02 ...
|
||||||
|
"""
|
||||||
|
|
||||||
|
def add_arguments(self, parser):
|
||||||
|
parser.add_argument('app_name', nargs='*', type=str, )
|
||||||
|
|
||||||
|
def handle(self, *args, **options):
|
||||||
|
app_name = options.get('app_name')
|
||||||
|
for name in app_name:
|
||||||
|
app_path = os.path.join(BASE_DIR, "apps", name)
|
||||||
|
# 判断app是否存在
|
||||||
|
if os.path.exists(app_path):
|
||||||
|
print(f"创建失败,App {name} 已存在!")
|
||||||
|
break
|
||||||
|
source_path = os.path.join(BASE_DIR, "apps", "vadmin", "template")
|
||||||
|
target_path = app_path
|
||||||
|
if not os.path.exists(target_path):
|
||||||
|
# 如果目标路径不存在原文件夹的话就创建
|
||||||
|
os.makedirs(target_path)
|
||||||
|
if os.path.exists(source_path):
|
||||||
|
# 如果目标路径存在原文件夹的话就先删除
|
||||||
|
shutil.rmtree(target_path)
|
||||||
|
shutil.copytree(source_path, target_path)
|
||||||
|
# 修改app中的apps 内容
|
||||||
|
content = f"""from django.apps import AppConfig
|
||||||
|
|
||||||
|
|
||||||
|
class {name.capitalize()}Config(AppConfig):
|
||||||
|
name = '{name}'
|
||||||
|
verbose_name = "{name}App"
|
||||||
|
"""
|
||||||
|
with open(os.path.join(app_path, "apps.py"), 'w', encoding='UTF-8') as f:
|
||||||
|
f.write(content)
|
||||||
|
f.close()
|
||||||
|
# 注册app到 settings.py 中
|
||||||
|
injection(os.path.join(BASE_DIR, "application", "settings.py"), f" 'apps.{name}',\n", "INSTALLED_APPS",
|
||||||
|
"]")
|
||||||
|
|
||||||
|
# 注册app到 urls.py 中
|
||||||
|
injection(os.path.join(BASE_DIR, "application", "urls.py"),
|
||||||
|
f" re_path(r'^{name}/', include('apps.{name}.urls')),\n", "urlpatterns = [",
|
||||||
|
"]")
|
||||||
|
|
||||||
|
print(f"创建 {name} App成功")
|
||||||
|
|
||||||
|
|
||||||
|
def injection(file_path, insert_content, startswith, endswith):
|
||||||
|
with open(file_path, "r+", encoding="utf-8") as f:
|
||||||
|
data = f.readlines()
|
||||||
|
with open(file_path, 'w', encoding='UTF-8') as f1:
|
||||||
|
is_INSTALLED_APPS = False
|
||||||
|
is_insert = False
|
||||||
|
for content in data:
|
||||||
|
# 判断文件是否 INSTALLED_APPS 开头
|
||||||
|
if not is_insert and content.startswith(startswith):
|
||||||
|
is_INSTALLED_APPS = True
|
||||||
|
if not is_insert and content.startswith(endswith) and is_INSTALLED_APPS:
|
||||||
|
# 给前一行插入数据
|
||||||
|
content = insert_content + content
|
||||||
|
is_insert = True
|
||||||
|
f1.writelines(content)
|
|
@ -12,7 +12,7 @@ logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
class Command(BaseCommand):
|
class Command(BaseCommand):
|
||||||
"""
|
"""
|
||||||
项目初始化命令: python manage.py initialization
|
项目初始化命令: python manage.py init
|
||||||
"""
|
"""
|
||||||
|
|
||||||
def customSql(self, sql_list, model_name, table_name, is_yes):
|
def customSql(self, sql_list, model_name, table_name, is_yes):
|
||||||
|
|
|
@ -0,0 +1 @@
|
||||||
|
urlpatterns = []
|
Loading…
Reference in New Issue