controller.tpl 2.82 KB
Newer Older
1 2
from datetime import datetime
from application.app import db
wanli's avatar
wanli committed
3
from models.{{ config['name'] }} import {{ config['name'] | letterUpper }}Model
wanli's avatar
wanli committed
4 5
from webcreator.log import logger
from webcreator.response import ResponseCode, response_result
wanli's avatar
wanli committed
6

wanli's avatar
wanli committed
7
class {{ config['name'] | letterUpper }}Resource(object):
wanli's avatar
wanli committed
8
    def __init__(self):
9
        super().__init__()
wanli's avatar
wanli committed
10 11

    def get(self, params):
wanli's avatar
wanli committed
12
        # handle business
wanli's avatar
wanli committed
13 14
        logger.warn(params)
        filters = []
wanli's avatar
wanli committed
15
        result = {{ config['name'] | letterUpper }}Model.query.filter(*filters).order_by({{ config['name'] | letterUpper }}Model.{{ config["model"]["fields"][0]["name"] }}).paginate(params.get('page', 1), params.get('pageSize', 10), error_out=False)
wanli's avatar
wanli committed
16 17 18 19 20 21

        return result

    def getList(self, params):
        # handle business
        logger.warn(params)
22
        filters = []
wanli's avatar
wanli committed
23
        result = {{ config['name'] | letterUpper }}Model.query.filter(*filters).order_by({{ config['name'] | letterUpper }}Model.{{ config["model"]["fields"][0]["name"] }}).paginate(params.get('page', 1), params.get('pageSize', 10), error_out=False)
wanli's avatar
wanli committed
24 25

        return result
wanli's avatar
wanli committed
26

27
    def post(self, params, jwt=None):
wanli's avatar
wanli committed
28
        # handle business
wanli's avatar
wanli committed
29
        result = {{ config['name'] | letterUpper }}Model.query.filter({{ config['name'] | letterUpper }}Model.{{ config["model"]["fields"][0]["name"] }} == params.get('{{ config["model"]["fields"][0]["name"] }}')).first()
30 31
        if result and result.is_delete:
            result.is_delete = False
wanli's avatar
wanli committed
32 33 34
            result.update_by = jwt['id']
            result.update_date = datetime.now()
            db.session.commit()
wanli's avatar
wanli committed
35
            return (True, None)
36
        elif result and result.is_delete == False:
wanli's avatar
wanli committed
37
            return (False, "exists")
wanli's avatar
wanli committed
38

wanli's avatar
wanli committed
39
        result = {{ config['name'] | letterUpper }}Model(**params)
wanli's avatar
wanli committed
40 41
        db.session.add(result)
        db.session.commit()
wanli's avatar
wanli committed
42
        return (True, None)
wanli's avatar
wanli committed
43

44
    def put(self, id, params, jwt=None):
wanli's avatar
wanli committed
45
        # handle business
wanli's avatar
wanli committed
46
        result = {{ config['name'] | letterUpper }}Model.query.get(id)
47
        if not result: return response_result(ResponseCode.NO_DATA)
wanli's avatar
wanli committed
48 49
        if params:
            for key, value in params.items():
wanli's avatar
wanli committed
50 51 52 53 54
                if value != None: setattr(result, key, value)
            result.update_by = jwt['id']
            result.update_date = datetime.now()
            db.session.commit()
        else:
55
            return response_result(ResponseCode.PARAM_NULL)
wanli's avatar
wanli committed
56

57
    def delete(self, id, jwt=None):
wanli's avatar
wanli committed
58
        # handle business
wanli's avatar
wanli committed
59
        result = {{ config['name'] | letterUpper }}Model.query.get(id)
60
        if not result: return response_result(ResponseCode.NO_DATA_FOUND)
wanli's avatar
wanli committed
61 62 63
        else:
            result.update_by = jwt['id']
            result.update_date = datetime.now()
64
            result.is_delete = True
wanli's avatar
wanli committed
65
            db.session.delete(result)
66 67
            db.session.commit()

wanli's avatar
wanli committed
68
{{ config["name"] }}Manager = {{ config['name'] | letterUpper }}Resource()