vanna_flask/service/result.py

import logging
import numpy as np
import json
import datetime
import decimal


class MyEncoder(json.JSONEncoder):
    def default(self, obj):
        if isinstance(obj, np.integer):
            return int(obj)
        elif isinstance(obj, np.floating):
            return float(obj)
        elif isinstance(obj, np.ndarray):
            return obj.tolist()
        elif isinstance(obj,decimal.Decimal):
            return float(obj)
        elif isinstance(obj, datetime.datetime):
            return obj.strftime("%Y-%m-%d %H:%M:%S")
        else:
            return super(MyEncoder, self).default(obj)



SUCCESS_CODE=20000
FAILED_CODE=50000
logger = logging.getLogger('app')

def success(data, msg):
    return {"code": SUCCESS_CODE, "data": data, "level": 0, "msg": msg}


def failed(data, msg):
    return {"code": FAILED_CODE, "data": data, "level": 1, "msg": msg}