diff --git a/src/database/__init__.py b/src/database/__init__.py index 634c407..29160a7 100644 --- a/src/database/__init__.py +++ b/src/database/__init__.py @@ -1,6 +1,6 @@ from loguru import logger from peewee import ( - MySQLDatabase, + ReconnectMySQLDatabase, CompositeKey, Model, CharField, @@ -12,7 +12,7 @@ from src.config import settings -db = MySQLDatabase( +db = ReconnectMySQLDatabase( database=settings.database, host=settings.database_host, port=settings.database_port, diff --git a/src/order/afdian/factory.py b/src/order/afdian/factory.py index d7dfc80..6f92419 100644 --- a/src/order/afdian/factory.py +++ b/src/order/afdian/factory.py @@ -1,5 +1,5 @@ from dataclasses import dataclass -from typing import Tuple +from typing import Tuple, Any from loguru import logger from datetime import datetime, timedelta import json @@ -9,7 +9,7 @@ from .query_afdian import query_order_by_out_trade_no -async def process_order(out_trade_no: str) -> Tuple[bool, str]: +async def process_order(out_trade_no: str) -> Tuple[Any, str]: response = await query_order_by_out_trade_no(out_trade_no) if not response or response.get("ec") != 200: logger.error( @@ -42,6 +42,7 @@ async def process_order(out_trade_no: str) -> Tuple[bool, str]: ) except Exception as e: logger.error(f"Create bill failed, out_trade_no: {out_trade_no}, error: {e}") + return None, "Create bill failed" if not bill: logger.error(f"Create bill failed, out_trade_no: {out_trade_no}")