user.py 4.14 KB
Newer Older
wanli's avatar
wanli committed
1 2 3
#!/usr/bin/env python
# -*- coding: utf_8 -*-

wanli's avatar
wanli committed
4 5 6 7 8
from flask import current_app, jsonify, request
from flask_restful import Resource
from flask_restful.reqparse import RequestParser
from flask_jwt_extended import ( jwt_required, get_jwt_identity )
from application.signal_manager import signalManager
wanli's avatar
wanli committed
9
from models.user import  postUserSchema, deleteUserSchema, getListUserSchema, getListUsersSchema, getUserSchema, putUserSchema
wanli's avatar
wanli committed
10
from webcreator.log import logger
wanli's avatar
wanli committed
11 12
from webcreator.response import ResponseCode, response_result

wanli's avatar
wanli committed
13
class UserResourceList(Resource):
wanli's avatar
wanli committed
14 15 16 17 18 19 20 21 22 23 24 25 26
    def __init__(self):
        pass
        # 特殊参数,即不是从json获取参数的接口,可以将这个注释打开
        # self.parser = RequestParser()

    def get(self):
        # 特殊参数,即不是从json获取参数的接口,可以将这个注释打开
        # self.parser.add_argument("page", type=int, location="args", default=1)
        # self.parser.add_argument("pageSize", type=int, location="args", default=15)
        # args = self.parser.parse_args()

        try:
            json_payload = request.json
wanli's avatar
wanli committed
27
            logger.warn(json_payload)
wanli's avatar
wanli committed
28
            data = getListUserSchema.load(json_payload)
29
            result, message = signalManager.actionGetlistUser.emit(data)
wanli's avatar
wanli committed
30
            json_dumps = getListUserSchema.dump(result)
wanli's avatar
wanli committed
31 32
            if result:
                json_dumps = getListUsersSchema.dump(result.items)
33
                logger.warn(json_dumps)
wanli's avatar
wanli committed
34 35
                return response_result(message, data=json_dumps, count=result.total)
            return response_result(message)
wanli's avatar
wanli committed
36 37
        except Exception as e:
            current_app.logger.error(e)
wanli's avatar
wanli committed
38
            return response_result(ResponseCode.HTTP_SERVER_ERROR)
wanli's avatar
wanli committed
39 40 41 42 43

    def post(self):
        try:
            json_payload = request.json
            data = postUserSchema.load(json_payload)
wanli's avatar
wanli committed
44 45 46 47
            result, message = signalManager.actionPostUser.emit(data)
            logger.info(result)
            logger.warn(message)
            return response_result(message)
wanli's avatar
wanli committed
48 49
        except Exception as e:
            current_app.logger.error(e)
wanli's avatar
wanli committed
50
            return response_result(ResponseCode.HTTP_SERVER_ERROR)
wanli's avatar
wanli committed
51 52


wanli's avatar
wanli committed
53
class UserResource(Resource):
wanli's avatar
wanli committed
54 55 56 57 58
    def __init__(self):
        pass
        # 特殊参数,即不是从json获取参数的接口,可以将这个注释打开
        # self.parser = RequestParser()

59
    @jwt_required(locations=["headers"])
wanli's avatar
wanli committed
60
    def get(self, uuid):
wanli's avatar
wanli committed
61 62 63 64 65 66 67
        # 特殊参数,即不是从json获取参数的接口,可以将这个注释打开
        # self.parser.add_argument("page", type=int, location="args", default=1)
        # self.parser.add_argument("pageSize", type=int, location="args", default=15)
        # args = self.parser.parse_args()

        try:
            json_payload = request.json
wanli's avatar
wanli committed
68
            print("========>", uuid, json_payload)
wanli's avatar
wanli committed
69
            data = getUserSchema.load(json_payload)
wanli's avatar
wanli committed
70 71 72 73 74
            result, message = signalManager.actionGetUser.emit(uuid, data)
            if result:
                json_dumps = getUserSchema.dump(result)
                return response_result(message, data=json_dumps)
            return response_result(message)
wanli's avatar
wanli committed
75 76
        except Exception as e:
            current_app.logger.error(e)
wanli's avatar
wanli committed
77
            return response_result(ResponseCode.HTTP_SERVER_ERROR)
wanli's avatar
wanli committed
78 79


80
    @jwt_required(locations=["headers"])
wanli's avatar
wanli committed
81
    def put(self, uuid):
wanli's avatar
wanli committed
82 83 84
        try:
            json_payload = request.json
            data = putUserSchema.load(json_payload)
wanli's avatar
wanli committed
85 86
            result, message = signalManager.actionPutUser.emit(uuid, data)
            return response_result(message, data=result)
wanli's avatar
wanli committed
87 88
        except Exception as e:
            current_app.logger.error(e)
wanli's avatar
wanli committed
89
            return response_result(ResponseCode.HTTP_SERVER_ERROR)
wanli's avatar
wanli committed
90 91


92
    @jwt_required(locations=["headers"])
wanli's avatar
wanli committed
93
    def delete(self, uuid):
wanli's avatar
wanli committed
94 95
        try:
            json_payload = request.json
wanli's avatar
wanli committed
96
            print("========>", uuid, json_payload)
97
            # data = deleteUserSchema.load(json_payload)
wanli's avatar
wanli committed
98 99
            result, message = signalManager.actionDeleteUser.emit(uuid)
            return response_result(message, data=result)
wanli's avatar
wanli committed
100 101
        except Exception as e:
            current_app.logger.error(e)
wanli's avatar
wanli committed
102
            return response_result(ResponseCode.HTTP_SERVER_ERROR)