107 lines
4.3 KiB
Python
107 lines
4.3 KiB
Python
from typing import Dict, Optional
|
|
from backend.services.project.models import ProjectDoc
|
|
from backend.services.payment.models import IncomeProfileDoc
|
|
from datetime import datetime
|
|
from common.log.module_logger import ModuleLogger
|
|
|
|
|
|
class PaymentManager:
|
|
def __init__(self) -> None:
|
|
self.module_logger = ModuleLogger(sender_id=PaymentManager)
|
|
|
|
async def fetch_wechat_qr_code(self, project_id: str) -> Optional[Dict[str, any]]:
|
|
project = await ProjectDoc.get(project_id)
|
|
proposer = project.proposer_id
|
|
income_profile = await IncomeProfileDoc.find_one(
|
|
IncomeProfileDoc.user_id == proposer
|
|
)
|
|
if income_profile:
|
|
return income_profile.bank_account.money_collecting_methods[
|
|
0
|
|
].wechat_qr_code
|
|
return None
|
|
|
|
async def fetch_stripe_account_id(self, user_id: str) -> Optional[str]:
|
|
income_profile = await IncomeProfileDoc.find_one(IncomeProfileDoc.user_id == user_id)
|
|
if income_profile:
|
|
return income_profile.bank_account.money_collecting_methods[
|
|
0
|
|
].stripe_account_id
|
|
return None
|
|
|
|
async def update_stripe_account_status(
|
|
self,
|
|
stripe_account_id: str,
|
|
setup: bool,
|
|
verified: bool
|
|
) -> bool:
|
|
try:
|
|
# Should use IncomeProfileDoc to be consistent with other methods
|
|
payment_profile = await IncomeProfileDoc.find_one(
|
|
{"bank_account.money_collecting_methods": {
|
|
"$elemMatch": {
|
|
"stripe_account_id": stripe_account_id
|
|
}
|
|
}}
|
|
)
|
|
|
|
if not payment_profile:
|
|
await self.module_logger.log_warning(
|
|
warning="No payment profile found for Stripe account",
|
|
properties={
|
|
"stripe_account_id": stripe_account_id,
|
|
"action": "update_stripe_account_status"
|
|
}
|
|
)
|
|
return False
|
|
|
|
# Update the stripe method status
|
|
updated = False
|
|
# Need to check if money_collecting_methods exists and is not empty
|
|
if payment_profile.bank_account and payment_profile.bank_account.money_collecting_methods:
|
|
for method in payment_profile.bank_account.money_collecting_methods:
|
|
if method.stripe_account_id == stripe_account_id:
|
|
method.setup = setup
|
|
method.verified = verified
|
|
method.last_update_time = int(datetime.now().timestamp())
|
|
updated = True
|
|
break # Exit loop once found and updated
|
|
|
|
if updated:
|
|
await payment_profile.save()
|
|
await self.module_logger.log_info(
|
|
info="Successfully updated Stripe account status",
|
|
properties={
|
|
"stripe_account_id": stripe_account_id,
|
|
"user_id": payment_profile.user_id,
|
|
"setup": setup,
|
|
"verified": verified
|
|
}
|
|
)
|
|
return True
|
|
|
|
# Log warning with more context
|
|
await self.module_logger.log_warning(
|
|
warning="Stripe account not found in payment methods",
|
|
properties={
|
|
"stripe_account_id": stripe_account_id,
|
|
"user_id": payment_profile.user_id if payment_profile else None,
|
|
"has_bank_account": bool(payment_profile and payment_profile.bank_account),
|
|
"has_methods": bool(payment_profile and payment_profile.bank_account and payment_profile.bank_account.money_collecting_methods)
|
|
}
|
|
)
|
|
return False
|
|
|
|
except Exception as e:
|
|
await self.module_logger.log_exception(
|
|
exception=e,
|
|
info="Failed to update Stripe account status",
|
|
properties={
|
|
"stripe_account_id": stripe_account_id,
|
|
"setup": setup,
|
|
"verified": verified,
|
|
"error": str(e)
|
|
}
|
|
)
|
|
return False
|