增加命令行新建空白视图功能

2021-07-16 13:04:49 +08:00
parent 04a89144ba
commit 78b7dae880
7 changed files with 139 additions and 8 deletions
@@ -0,0 +1,26 @@
+import os
+
+import click
+
+from applications.common.script.initdb import init_db
+from applications.common.script.newmodular.new import NewViewModular
+
+
+def init_script(app):
+    @app.cli.command()
+    def init():
+        init_db()
+
+    @app.cli.command()
+    @click.option('--type', prompt="请输入类型", help='新增的类型')
+    @click.option('--name', prompt="请输入新增的名称", help='新增的名称')
+    def new(type,name):
+        if type == 'view':
+            if name.count('/') > 1:
+                print("目前只支持二级目录，多级目录需要蓝图嵌套，本命令暂不支持，请手动创建")
+                quit()
+            if type == "view" and os.path.exists(f"applications/view/{name}.py"):
+                print(f'已经存在视图模块{name}.py')
+                quit()
+            NewViewModular(name=name).new_view()
+
@@ -0,0 +1,60 @@
+from dotenv import dotenv_values
+import sqlparse
+import pymysql
+config = dotenv_values('.env')
+# MySql配置信息
+HOST = config.get('MYSQL_HOST') or '127.0.0.1'
+PORT = config.get('MYSQL_PORT') or 3306
+DATABASE = config.get('MYSQL_DATABASE') or 'PearAdminFlask'
+USERNAME = config.get('MYSQL_USERNAME') or 'root'
+PASSWORD = config.get('MYSQL_PASSWORD') or '123456'
+
+
+def is_exist_database():
+    db = pymysql.connect(host=HOST, port=int(PORT), user=USERNAME, password=PASSWORD, charset='utf8mb4')
+    cursor1 = db.cursor()
+    sql = "select * from information_schema.SCHEMATA WHERE SCHEMA_NAME = '%s'  ; " % DATABASE
+    res = cursor1.execute(sql)
+    db.close()
+    return res
+
+
+def init_database():
+    db = pymysql.connect(host=HOST, port=int(PORT), user=USERNAME, password=PASSWORD, charset='utf8mb4')
+    cursor1 = db.cursor()
+    sql = "CREATE DATABASE IF NOT EXISTS %s" % DATABASE
+    res = cursor1.execute(sql)
+    db.close()
+    return res
+
+
+def execute_fromfile(filename):
+    db = pymysql.connect(host=HOST, port=int(PORT), user=USERNAME, password=PASSWORD, database=DATABASE,
+                         charset='utf8mb4')
+    fd = open(filename, 'r', encoding='utf-8')
+    cursor = db.cursor()
+    sqlfile = fd.read()
+    sqlfile = sqlparse.format(sqlfile, strip_comments=True).strip()
+
+    sqlcommamds = sqlfile.split(';')
+
+    for command in sqlcommamds:
+        try:
+            cursor.execute(command)
+            db.commit()
+
+        except Exception as msg:
+
+            db.rollback()
+    db.close()
+
+
+def init_db():
+    if is_exist_database():
+        print('数据库已经存在,为防止误初始化，请手动删除 %s 数据库' % str(DATABASE))
+        return
+    if init_database():
+        print('数据库%s创建成功' % str(DATABASE))
+    execute_fromfile('test/pear.sql')
+    print('表创建成功')
+    print('欢迎使用pear-admin-flask,请使用 flask run 命令启动程序')
@@ -0,0 +1,92 @@
+import os
+
+import jinja2
+
+
+class NewViewModular():
+
+    def __init__(self, name):
+        self.name = name
+
+    def path_is_exists(self):
+        return os.path.exists(f"applications/view/{self.name.split('/')[0]}")
+
+    def new_dirs(self):
+        if not self.path_is_exists():
+            # 如果不存在则创建目录
+
+            # 创建目录操作函数
+            os.makedirs(f"applications/view/{self.name.split('/')[0]}")
+
+            print(self.name + ' 创建成功')
+            return True
+        else:
+            return False
+
+    def add_view(self):
+        # jinja渲染
+        templateLoader = jinja2.FileSystemLoader(searchpath='applications/common/script/newmodular/template/')
+        templateEnv = jinja2.Environment(loader=templateLoader)
+        TEMPLATE_FILE = "view"
+        template = templateEnv.get_template(TEMPLATE_FILE, )
+        templateVars = {"name": self.name}
+        outputText = template.render(templateVars)
+        with open(f"applications/view/{self.name}.py", "a", encoding='utf-8') as f:
+            f.write(outputText)
+            f.close()
+
+    def add_init(self):
+        # 判断__init__.py
+        if os.path.exists(f"applications/view/{self.name.split('/')[0]}/__init__.py"):
+            with open(f"applications/view/{self.name.split('/')[0]}/__init__.py", "r") as file:
+                lines = file.readlines()
+                print(lines)
+                file.close()
+                import_line_num = 0
+                for line in lines:
+                    if 'def' in line:
+                        import_line_num = lines.index(line) - 2
+            with open(f"applications/view/{self.name.split('/')[0]}/__init__.py", "w") as file:
+                lines.insert(import_line_num,
+                             f"from applications.view.{self.name.replace('/', '.')} import {self.name.replace('/', '_')}\n")
+                print(lines)
+                lines.insert(len(lines) - 1, f"    app.register_blueprint({self.name.replace('/', '_')})\n")
+                print(lines)
+                file.writelines(lines)
+                file.close()
+            return True
+        else:
+            templateLoader = jinja2.FileSystemLoader(searchpath='applications/common/script/newmodular/template/')
+            templateEnv = jinja2.Environment(loader=templateLoader)
+            TEMPLATE_FILE = "__init__"
+            template = templateEnv.get_template(TEMPLATE_FILE, )
+            templateVars = {"name": self.name}
+            output = template.render(templateVars)
+            with open(f"applications/view/{self.name.split('/')[0]}/__init__.py", "a", encoding='utf-8') as f:
+                f.write(output)
+                f.close()
+            self.add_root_init()
+            return True
+
+    def add_root_init(self):
+        with open(f"applications/view/__init__.py", "r") as file:
+            lines = file.readlines()
+            file.close()
+            import_line_num = 0
+            for line in lines:
+                if 'def' in line:
+                    import_line_num = lines.index(line) - 2
+        with open(f"applications/view/__init__.py", "w") as file:
+            lines.insert(
+                import_line_num,
+                f"from applications.view.{self.name.split('/')[0]} import register_{self.name.split('/')[0]}_views\n")
+            print(lines)
+            lines.insert(len(lines) - 1, f"    register_{self.name.split('/')[0]}_views(app)\n")
+            print(lines)
+            file.writelines(lines)
+            file.close()
+
+    def new_view(self):
+        self.new_dirs()
+        self.add_view()
+        self.add_init()
@@ -0,0 +1,6 @@
+from flask import Flask
+from applications.view.{{name.replace('/','.')}} import {{name.replace('/','_')}}
+
+
+def register_{{name.split('/')[0]}}_views(app: Flask):
+    app.register_blueprint({{name.replace('/','_')}})
@@ -0,0 +1,8 @@
+from flask import Blueprint
+
+{{name.replace('/','_')}} = Blueprint('{{name.replace('/','_')}}', __name__, url_prefix='/{{name}}')
+
+
+@{{name.replace('/','_')}}.route('/')
+def index():
+    return "这是{{name}}路由"