148 lines
5.2 KiB
Python
148 lines
5.2 KiB
Python
"""
|
||
支付模块配置（从 payment 迁入 app，从 app.core.config.settings 读取）
|
||
"""
|
||
from app.core.logging import get_logger
|
||
from dataclasses import dataclass, field
|
||
|
||
logger = get_logger(__name__)
|
||
|
||
|
||
@dataclass
|
||
class WeChatPayConfig:
|
||
app_id: str = ""
|
||
mch_id: str = ""
|
||
api_v3_key: str = ""
|
||
private_key_path: str = ""
|
||
private_key: str = ""
|
||
cert_serial_no: str = ""
|
||
notify_url: str = ""
|
||
platform_public_key: str = ""
|
||
platform_public_key_path: str = ""
|
||
platform_public_key_id: str = ""
|
||
|
||
@property
|
||
def is_configured(self) -> bool:
|
||
has_key = bool(self.private_key.strip()) or bool(self.private_key_path.strip())
|
||
return all(
|
||
[
|
||
self.app_id,
|
||
self.mch_id,
|
||
self.api_v3_key,
|
||
has_key,
|
||
self.cert_serial_no,
|
||
self.notify_url,
|
||
]
|
||
)
|
||
|
||
@property
|
||
def use_platform_public_key(self) -> bool:
|
||
has_pub = bool(self.platform_public_key.strip()) or bool(
|
||
self.platform_public_key_path.strip()
|
||
)
|
||
return has_pub and bool(self.platform_public_key_id.strip())
|
||
|
||
|
||
@dataclass
|
||
class AlipayConfig:
|
||
app_id: str = ""
|
||
private_key: str = ""
|
||
alipay_public_key: str = ""
|
||
notify_url: str = ""
|
||
sign_type: str = "RSA2"
|
||
|
||
@property
|
||
def is_configured(self) -> bool:
|
||
return all(
|
||
[
|
||
self.app_id,
|
||
self.private_key,
|
||
self.alipay_public_key,
|
||
self.notify_url,
|
||
]
|
||
)
|
||
|
||
|
||
@dataclass
|
||
class PaymentConfig:
|
||
wechat: WeChatPayConfig = field(default_factory=WeChatPayConfig)
|
||
alipay: AlipayConfig = field(default_factory=AlipayConfig)
|
||
alipay_under_development: bool = True
|
||
|
||
@classmethod
|
||
def from_settings(cls, settings) -> "PaymentConfig":
|
||
wechat_private_key = (
|
||
getattr(settings, "wechat_pay_private_key", "") or ""
|
||
).strip()
|
||
if wechat_private_key:
|
||
wechat_private_key = (
|
||
wechat_private_key.strip('"')
|
||
.strip("'")
|
||
.lstrip("\ufeff")
|
||
.replace("\\n", "\n")
|
||
)
|
||
wechat_platform_pub = (
|
||
getattr(settings, "wechat_pay_platform_public_key", "") or ""
|
||
).strip()
|
||
if wechat_platform_pub and "\\n" in wechat_platform_pub:
|
||
wechat_platform_pub = wechat_platform_pub.replace("\\n", "\n")
|
||
wechat_private_key_path = (
|
||
getattr(settings, "wechat_pay_private_key_path", "") or ""
|
||
).strip()
|
||
alipay_under = (
|
||
getattr(settings, "alipay_under_development", "true") or "true"
|
||
).lower()
|
||
config = cls(
|
||
wechat=WeChatPayConfig(
|
||
app_id=getattr(settings, "wechat_pay_app_id", "") or "",
|
||
mch_id=getattr(settings, "wechat_pay_mch_id", "") or "",
|
||
api_v3_key=getattr(settings, "wechat_pay_api_v3_key", "") or "",
|
||
private_key_path=wechat_private_key_path,
|
||
private_key=wechat_private_key if not wechat_private_key_path else "",
|
||
cert_serial_no=getattr(settings, "wechat_pay_cert_serial_no", "") or "",
|
||
notify_url=getattr(settings, "wechat_pay_notify_url", "") or "",
|
||
platform_public_key=wechat_platform_pub,
|
||
platform_public_key_path=(
|
||
getattr(settings, "wechat_pay_platform_public_key_path", "") or ""
|
||
).strip(),
|
||
platform_public_key_id=(
|
||
getattr(settings, "wechat_pay_platform_public_key_id", "") or ""
|
||
).strip(),
|
||
),
|
||
alipay=AlipayConfig(
|
||
app_id=getattr(settings, "alipay_app_id", "") or "",
|
||
private_key=getattr(settings, "alipay_private_key", "") or "",
|
||
alipay_public_key=getattr(settings, "alipay_public_key", "") or "",
|
||
notify_url=getattr(settings, "alipay_notify_url", "") or "",
|
||
sign_type=getattr(settings, "alipay_sign_type", "RSA2") or "RSA2",
|
||
),
|
||
alipay_under_development=alipay_under in ("true", "1", "yes"),
|
||
)
|
||
return config
|
||
|
||
@classmethod
|
||
def from_env(cls) -> "PaymentConfig":
|
||
from app.core.config import settings
|
||
|
||
config = cls.from_settings(settings)
|
||
if config.wechat.is_configured:
|
||
mode = (
|
||
"平台公钥模式"
|
||
if config.wechat.use_platform_public_key
|
||
else "平台证书模式"
|
||
)
|
||
key_src = "私钥内容" if config.wechat.private_key.strip() else "私钥路径"
|
||
logger.info(
|
||
"微信支付配置已加载: APP_ID=%s, MCH_ID=%s, 模式=%s, 商户私钥=%s",
|
||
config.wechat.app_id,
|
||
config.wechat.mch_id,
|
||
mode,
|
||
key_src,
|
||
)
|
||
else:
|
||
logger.warning("微信支付配置不完整，微信支付将不可用")
|
||
if config.alipay.is_configured:
|
||
logger.info("支付宝配置已加载: APP_ID=%s", config.alipay.app_id)
|
||
else:
|
||
logger.warning("支付宝配置不完整，支付宝支付将不可用")
|
||
return config
|