async def request_validation_exception_handler(request: Request, exc: RequestValidationError):
logger.error(f"参数错误{request.method} {request.url}")
return JSONResponse(
{
"code": status.HTTP_400_BAD_REQUEST,
"message": "请求参数校验异常",
"data": exc.errors()
}
)
def fetch_all(self, sql, args=None):
'''
批量查询记录
:param sql:
:param args:条件,可以为空
:return:list
'''
cursor, conn, count = self.execute(sql, param=args, is_dict=True)
try:
result = cursor.fetchall()
self.close(cursor, conn)
return result
except Exception as e:
logger.error(e)
self.close(cursor, conn)
raise UnicornException(StatusCode.C10005['code'],
StatusCode.C10005['msg'])
def insert(self, sql, *args):
'''
插入数据
:param sql:
:param args:字段
:return:
'''
cursor, conn, count = self.execute(sql, args)
try:
conn.commit()
self.close(cursor, conn)
except Exception as e:
conn.rollback()
self.close(cursor, conn)
logger.error(e)
raise UnicornException(StatusCode.C10001['code'],
StatusCode.C10001['msg'])
def fetch_one(self, sql, *args):
'''
查询单条记录
:param sql:
:param args:条件,可以多个条件
:return:dict
'''
cursor, conn, count = self.execute(sql, param=args, is_dict=True)
try:
result = cursor.fetchone()
self.close(cursor, conn)
return result if result else {}
except Exception as e:
self.close(cursor, conn)
logger.error(e)
raise UnicornException(StatusCode.C10004['code'],
StatusCode.C10004['msg'])
def update(self, sql, *args):
"""
更新记录
:param sql:
:param args: 条件,可以有多个条件
:return:
"""
cursor, conn, count = self.execute(sql, args)
try:
conn.commit()
self.close(cursor, conn)
return count
except Exception as e:
conn.rollback()
self.close(cursor, conn)
logger.error(e)
raise UnicornException(StatusCode.C10002['code'],
StatusCode.C10002['msg'])