update

tools/README.md

+# 代码生成工具
+
+使用Python Jinja2模板引擎，自动生成代码。
+

tools/config.json

@@ -2,7 +2,8 @@
     "application": {},
     "controllers": [
         {
-            "className": "",
+            "className": "CacheDataResource",
+            "modelName": "cache_data",
             "post": {},
             "delete": {},
             "get": {},
@@ -49,8 +50,8 @@
             ]
         },
         {
-            "tableName": "cache_data",
             "className": "CacheDataModel",
+            "tableName": "cache_data",
             "fields": [
                 {
                     "name": "id",
@@ -104,13 +105,15 @@
             ]
         }
     ],
-    "resources": [
+    "views": [
         {
             "post": {},
             "delete": {},
             "get": {},
             "put": {},
-            "router": ""
+            "name": "",
+            "path": "/test/<string:id>",
+            "className": ""
         }
     ],
     "utils": []

tools/gen_code.py

+import json
+
+def handleResources():
+    pass
+
+def handleController():
+    # 根据模型字段自动从models列表中取出file name信息和class name信息
+    pass
+
+def handleModel():
+    # 将所有有默认值的字段分为一组，没有默认值的字段分为另一组
+    # 生成模板代码时，无默认值的字段在前，有默认值的字段字在后
+    pass
+
+def hanleRender():
+    pass
+
+def readConfig():
+    result = {}
+    with open("config.json", "r+") as f:
+        result = json.loads(f.read())
+    return result
+
+if __name__ == "__main__":
+    result = readConfig()
+    print(result)
\ No newline at end of file

tools/requirements.txt

+Jinja2==2.11.3
+MarkupSafe==1.1.1

tools/resources/controllers/__init__.py

+# -*- coding: utf-8 -*-
\ No newline at end of file

tools/resources/models/__init__.py

+# -*- coding: utf-8 -*-
+
+from sqlalchemy import Column, Integer, String
+from sqlalchemy.ext.declarative import declared_attr
+from sqlalchemy.ext.declarative import declarative_base
+from flask_sqlalchemy import SQLAlchemy
+
+class Base(object):
+    @declared_attr
+    def __tablename__(cls):
+        return cls.__name__.lower()
+
+    __table_args__ = {'mysql_engine': 'InnoDB'}
+
+    id = Column(Integer, primary_key=True)
+
+Base = declarative_base(cls=Base)
+
+class MyMixin(object):
+    @declared_attr
+    def __tablename__(cls):
+        return cls.__name__.lower()
+
+    __table_args__ = {'mysql_engine': 'InnoDB'}
+    __mapper_args__= {'always_refresh': True}
+
+    id = Column(Integer, primary_key=True)
+
+class MyModel(MyMixin, Base):
+    name = Column(String(1000))
+
+db = SQLAlchemy()
\ No newline at end of file

tools/resources/models/base.py

+# -*- coding: utf-8 -*-
+
+import uuid
+from datetime import datetime
+from . import db
+
+class BaseModel(db.Model):
+    create_date = db.Column(db.DateTime, default=datetime.now)
+    create_by = db.Column(db.String(64))
+    update_date = db.Column(db.DateTime, default=datetime.now)
+    update_by = db.Column(db.String(64))
+    remarks = db.Column(db.String(255), default="")
+    del_flag = db.Column(db.BOOLEAN, default=0)
+
+class PrimaryModel(BaseModel):
+    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
+    uuid = db.Column(db.String(64), primary_key=True, default=uuid.uuid1)
+
+class AutoBaseModel(BaseModel):
+    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
+    sort = db.Column(db.Integer, default=0)
+
+class UuidBaseModel(BaseModel):
+    id = db.Column(db.String(64), primary_key=True, default=uuid.uuid1)
+    sort = db.Column(db.Integer, default=0)
+
+class SortBaseModel(BaseModel):
+    sort = db.Column(db.Integer, default=0)
\ No newline at end of file

tools/resources/utils/__init__.py

+# -*- coding: utf-8 -*-
\ No newline at end of file

tools/resources/views/__init__.py

+# -*- coding: utf-8 -*-
+
+from flask import Blueprint
+from flask_restful import Api
+
+from . import .
+
+api_v1 = Blueprint('api_v1', __name__)
+
+api = Api(api_v1)
+
+api.add_resource(area.AreaListResource, '/area/', endpoint='area')
+api.add_resource(area.AreaResource, '/area/<int:id>')
+api.add_resource(cache_data.CacheDataListResource, '/cache', endpoint='cache')
+api.add_resource(company_contact.ContactListResource, '/companyPhone/')
+api.add_resource(company_contact.ContactResource, '/companyPhone/<int:id>')
+api.add_resource(company.CompanyListResource, '/company/')
+api.add_resource(company.ExtendCompanyResource, '/company/openApi')
+api.add_resource(company.CompanyResource, '/company/<string:id>')
+api.add_resource(dictionary.DictListResource, '/dictionary/')
+api.add_resource(dictionary.ExtendDictResource, '/dictionary/groupBy')
+api.add_resource(dictionary.DictResource, '/dictionary/<int:id>')
+api.add_resource(equipment_attrs.AttrsListResource, '/equipmentAttributes/')
+api.add_resource(equipment_attrs.AttrsResource, '/equipmentAttributes/<string:id>')
+api.add_resource(equipment_attrs.AttrValuesListResource, '/equipmentAttributes/attr/')
+api.add_resource(equipment_attrs.AttrValuesResource, '/equipmentAttributes/attr/<string:id>')
+api.add_resource(equipment_login.LoginListResource, '/equipmentLoginLog/')
+api.add_resource(equipment_msg.MessageListResource, '/equipmentMessage/')
+api.add_resource(equipment_type.TypeListResource, '/equipmentTypes/')
+api.add_resource(equipment_type.TypeResource, '/equipmentTypes/<string:id>')
+api.add_resource(equipment_phone.PhoneListResource, '/equipmentPhone/')
+api.add_resource(equipment_phone.PhoneResource, '/equipmentPhone/<int:id>')
+api.add_resource(terminal_code.CodeListResource, '/equipmentTerminalCode/')
+api.add_resource(equipment_operate.OperateLogListResource, '/terminalLog/')
+api.add_resource(equipment_parameters.ParameterListResource, '/equipmentParameter/')
+api.add_resource(equipment_parameters.ParameterResource, '/equipmentParameter/<int:id>')
+api.add_resource(equipment_parameters.CustomParametersResource, '/parameters/custom/')
+api.add_resource(equipment_history_data.HistoryDataResource, '/historyData/<string:id>/')
+api.add_resource(equipment_history_data.HistoryChartResource, '/historyData/chart/<string:id>/')
+api.add_resource(equipment_now_data.NowDataListResource, '/equipmentNowData/')
+api.add_resource(equipment_now_data.NowDataResource, '/equipmentNowData/<string:id>')
+api.add_resource(equipment_now_data.ChartDataResource, '/equipmentNowData/chart/<string:id>')
+api.add_resource(equipment.EquipmentListResource, '/equipment/')
+api.add_resource(equipment.EquipmentResource, '/equipment/<string:id>')
+api.add_resource(equipment.ExtendEquipmentResource, '/equipment/exists')
+api.add_resource(equipment.EquipmentsConfig, '/equipment/config')
+api.add_resource(equipment.LabEquipmentResource, '/equipment/getLabEquipment')
+api.add_resource(equipment.TerminalCodeResource, '/equipment/getTerminalCode')
+api.add_resource(user.UserListResource, '/user/')
+api.add_resource(user.UserResource, '/user/<string:id>')
+api.add_resource(user.ExtendUserResource, '/user/updateOperation/<string:id>')
+api.add_resource(user_weixin.UserWeiXinListResource, '/userWeixin/')
+api.add_resource(user_weixin.UserWeiXinResource, '/userWeixin/<string:id>')
+api.add_resource(warning_report.WarningReportListResource, '/warningReport/')
+api.add_resource(warning_type.WarningTypeListResource, '/warningTypes/')
+api.add_resource(warning_type.WarningTypeResource, '/warningTypes/<string:id>')
+api.add_resource(firmware.FirmwareListResource, '/firmware/')
+api.add_resource(firmware.FirmwareResource, '/firmware/<int:id>')
+api.add_resource(menu.MenuListResource, '/menu/')
+api.add_resource(menu.MenuResource, '/menu/<string:id>')
+api.add_resource(operation.OperationListResource, '/operation/')
+api.add_resource(operation.OperationResource, '/operation/<string:id>')
+api.add_resource(operation.UserOperationResource, '/operation/getUserOperation/<string:user>')
+api.add_resource(operation.MenuOperationResource, '/operation/getMenuOperation/')
+api.add_resource(role.RoleListResource, '/role/')
+api.add_resource(role.RoleResource, '/role/<string:id>')
+api.add_resource(role_user.RoleUserListResource, '/roleUser/')
+api.add_resource(role_permission.RolePermissionListResource, '/rolePermission/')
+api.add_resource(role_permission.RolePermissionResource, '/rolePermission/<string:id>')
+api.add_resource(role_permission.ExtendRolePermission, '/rolePermission/type/<string:type>')
+api.add_resource(permission.PermissionListResource, '/permission/')
+api.add_resource(permission.PermissionResource, '/permission/<string:id>')
+api.add_resource(permission_related.RelatedListResource, '/permissionRelated/')
+api.add_resource(permission_related.ExtendRelatedResource, '/permissionRelated/batch/')
+api.add_resource(permission_related.RelatedResource, '/permissionRelated/<string:id>')
+api.add_resource(public.UserInfoResource, '/public/login')
+api.add_resource(public.FileResource, '/public/fileRequest/')
+api.add_resource(public.WeixinLogin, '/public/weixin/login/')
+api.add_resource(public.BackstageResource, '/public/equipmentMap')
+api.add_resource(public.SystemResource, '/public/dashboard')
+api.add_resource(public.LogoutResource, '/public/logout')
+api.add_resource(containers.ContainerListResource, '/containers/')
+api.add_resource(synchronous.CompanySyncResource, '/synchronous/company/')
+api.add_resource(synchronous.EquipmentSyncResource, '/synchronous/equipment/')
+api.add_resource(synchronous.CompanyThirdPartySyncResource, '/synchronous/thirdparty/company/')
+api.add_resource(synchronous.EquipmentThirdPartySyncResource, '/synchronous/thirdparty/equipment/')
+api.add_resource(synchronous.UserSyncResource, '/synchronous/user/')
+api.add_resource(report.ReportResource, '/report/')
+api.add_resource(openapi.APIEquipmentResource, '/openapi/equipment/')
\ No newline at end of file

tools/templates/controller.tpl

+from models import db
+from models.{{ file_name }} import {{ class_name }}
+
+class {{ className }}(object):
+    def __init__(self):
+        super({{ className }}, self).__init__()
+
+    def get(self, params):
+        # handle business code
+        pass
+
+    def post(self, params):
+        # handle business code
+        pass
+
+    def put(self, params):
+        # handle business code
+        pass
+
+    def delete(self, params):
+        # handle business code
+        pass
\ No newline at end of file

tools/templates/model.tpl

+# -*- coding: utf-8 -*-
+
+from . import db
+from datetime import datetime
+from .base import BaseModel, PrimaryModel
+
+class {{ className }}(db.Model):
+    __tablename__ = '{{ tableName }}'
+{% for key, value in fields.items() %}
+    {% if value.get('primaryKey', None) %}
+        value.get('name') = db.Column(db.{{ value.get('dataType') }}, primary_key = True, {% if value.get('autoIncrement') %}autoincrement = True{% endif %})
+    {% else %}
+        value.get('name') = db.Column(db.{{ value.get('dataType') }}{% if value.get("length", None) %}(value.get("length")){% endif %}{% if value.get("index", False) %}, index = True{% endif %}{% if value.get("default", None) %}, default = value.get("default"){% endif %})
+    {% endif %}
+{% endfor %}
+
+    # __table_args__ = (
+    #     db.Index('idx_areaCode', 'areaCode', mysql_using='btree'),
+    # )
+
+    def __init__(self{% for for key, value in fields.items() %}, {{ value.get("name") }}{% endfor %}):
+{% for key, value in fields.items() %}
+        self.{{ value.get("name") }} = {{ value.get("name") }}
+{% endfor %}
+
+    def __repr__(self):
+        return '<{{ className }} %r>' % (self.{{ fields[0]["name"] }})

tools/templates/router.tpl

+# -*- coding: utf-8 -*-
+
+from flask import Blueprint
+from flask_restful import Api
+
+from . import {% for i, router in enumerate(routers) in  %}{% if i != 0 %},{% endif %}{{ router }}{% endfor %}
+
+api_v1 = Blueprint('api_v1', __name__)
+
+api = Api(api_v1)
+
+{% for v in views %}
+api.add_resource({{ name }}.{{ clasName }}, '{{ path }}'{% if endpoint %}, endpoint={{ endpoint }}{% endif %})
+{% endfor %}

tools/templates/view.tpl

+from flask import current_app
+from flask_restful import Resource
+from flask_restful.reqparse import RequestParser
+from flask_jwt_extended import ( jwt_required, get_jwt_identity )
+from common import code, pretty_result
+
+class {{ className }}(Resource):
+    def __init__(self):
+        self.parser = RequestParser()
+
+    @jwt_required
+    def get(self):
+        self.parser.add_argument("currentPage", type=int, location="args", default=1)
+        self.parser.add_argument("pageSize", type=int, location="args", default=15)
+        self.parser.add_argument("equipment", type=int, location="args", required=False)
+        args = self.parser.parse_args()
+
+        try:
+            
+        except Exception as e:
+            current_app.logger.error(e)
+            db.session.rollback()
+            return pretty_result(code.DB_ERROR)
\ No newline at end of file