cashdesk.py 6.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183
  1. # -*- coding: utf-8 -*-
  2. #!/usr/bin/env python
  3. """
  4. 手工对账
  5. """
  6. import os
  7. import sys
  8. import click
  9. from typing import Union, TYPE_CHECKING
  10. from apps.web.core.payment import PaymentGateway
  11. from apps.web.dealer.define import DEALER_INCOME_TYPE
  12. from apps.web.user.transaction import post_pay
  13. if TYPE_CHECKING:
  14. from apps.web.core.payment.wechat import WechatPaymentGateway
  15. from apps.web.core.payment.ali import AliPayGateway
  16. PROJECT_ROOT = os.path.join(os.path.abspath(os.path.split(os.path.realpath(__file__))[0] + "/.."), '..')
  17. sys.path.insert(0, PROJECT_ROOT)
  18. from script.base import init_env, get_logger
  19. logger = get_logger(__name__)
  20. init_env(interactive=True)
  21. from apilib.monetary import RMB
  22. from apilib.utils_datetime import to_datetime, yesterday_format_str
  23. from apps.web.common.models import WithdrawRecord
  24. from apps.web.dealer.models import Dealer
  25. from apps.web.dealer.proxy import DealerIncomeProxy
  26. from apps.web.user.models import RechargeRecord
  27. @click.group()
  28. def cli():
  29. """
  30. 同步mongoDB
  31. :return:
  32. """
  33. click.echo(u"欢迎进入自动对账脚本")
  34. def check_dealer_balance(dealer, income_type = DEALER_INCOME_TYPE.DEVICE_INCOME, source_key = None):
  35. proxies = DealerIncomeProxy.objects(dealerIds = dealer.id)
  36. acc_income = RMB(0)
  37. for proxy in proxies:
  38. acc_income += RMB(proxy.actualAmountMap[str(dealer.id)])
  39. dealer_actual_balance = dealer.sub_balance(income_type, source_key)
  40. for record in WithdrawRecord.get_succeeded_records(ownerId = str(dealer.id)):
  41. dealer_actual_balance += RMB(record.amount)
  42. return acc_income == dealer_actual_balance, acc_income, dealer_actual_balance, abs(
  43. dealer_actual_balance - acc_income)
  44. def check_dealer_balance_by_interval(dealer, income_type, source_key):
  45. """
  46. records = WithdrawRecord.get_succeeded_records(ownerId=str(dealer.id))
  47. Stage 0 interval <- [ records[-1], now] :
  48. balance == DealerIncomeProxy.sum_by_dealer(time >= last WithdrawRecord's date, dealerId)
  49. Stage 1 interval <- [ records[-2], records[-1])
  50. records[-2].dealerBalance == DealerIncomeProxy.sum_by_dealer( time in interval, dealerId)
  51. Stage 2 ...
  52. :param dealer:
  53. :return:
  54. """
  55. dealer_id = dealer.id
  56. records = list(WithdrawRecord.get_succeeded_records(ownerId=str(dealer_id)))
  57. def sum_by_dealer(**query):
  58. return DealerIncomeProxy.sum_by_dealer(dealerId=dealer_id, **query)
  59. if not records:
  60. assert dealer.sub_balance(income_type = income_type, source_key = source_key) == sum_by_dealer()
  61. return
  62. else:
  63. last_record = records.pop()
  64. endTime = last_record.postTime
  65. assert dealer.sub_balance(income_type = income_type, source_key = source_key) == sum_by_dealer(
  66. dateTimeAdded__gt = last_record.postTime)
  67. while len(records):
  68. record = records.pop() # type: WithdrawRecord
  69. assert record.balance == sum_by_dealer(dateTimeAdded__gt = record.postTime, dateTimeAdded__lt = endTime)
  70. endTime = record.postTime
  71. def generate_dealer_income_intervals(dealer):
  72. pass
  73. @cli.command()
  74. def check_dealer_balances():
  75. dealers = Dealer.objects()
  76. total = dealers.count()
  77. count = 0
  78. for dealer in dealers:
  79. match, balance, order_income, _ = check_dealer_balance(dealer)
  80. if not match:
  81. count += 1
  82. click.echo('mismatch!, dealer(phone=%s, id=%s), acc_recharge_record_income=%s, dealer_actual_balance=%s'
  83. % (dealer.username, str(dealer.id), order_income, balance))
  84. click.echo('there are %d mismatch records, total %d, normal %d' % (count, total, total - count ) )
  85. def fixed_upay_record(record, gateway_dict):
  86. click.echo('check unpay record. record = {}'.format(record))
  87. if record.payGatewayKey in gateway_dict:
  88. payment_gateway = gateway_dict[
  89. '{}-{}-{}'.format(record.gateway, record.payAppType, record.payGatewayKey)]
  90. else:
  91. payment_gateway = PaymentGateway.from_gateway_key(
  92. record.gateway,
  93. record.payGatewayKey,
  94. record.payAppType) # type: Union[WechatPaymentGateway, AliPayGateway, SaobeiPaymentGateway]
  95. gateway_dict[
  96. '{}-{}-{}'.format(record.gateway, record.payAppType, record.payGatewayKey)] = payment_gateway
  97. status, trade_no = payment_gateway.api_trade_query(out_trade_no = record.orderNo)
  98. if status in [PaymentGateway.TradeStatus.Finished, PaymentGateway.TradeStatus.Success]:
  99. click.echo('record really paid. something is error. record = {}'.format(str(record.id)))
  100. modified = record.succeed(wxOrderNo = trade_no)
  101. if not modified:
  102. click.echo('record has done. record = {}'.format(str(record.id)))
  103. post_pay(recharge_record = record, accounting = True)
  104. @cli.command()
  105. @click.option('-d','--dealer_id', prompt=u'please input dealer id')
  106. @click.option('-t','--check_time', prompt=u'please input check date. for example: 2019-09-04')
  107. def fixed_upay_record_by_dealer(dealer_id, check_time):
  108. if not check_time:
  109. check_time = yesterday_format_str()
  110. start_time = to_datetime('{} 00:00:00'.format(check_time))
  111. end_time = to_datetime('{} 23:59:59'.format(check_time))
  112. records = RechargeRecord.objects(ownerId = dealer_id,
  113. result = RechargeRecord.PayResult.UNPAY,
  114. dateTimeAdded__gte = start_time,
  115. dateTimeAdded__lte = end_time)
  116. gateway_dict = {}
  117. for record in records: # type: RechargeRecord
  118. try:
  119. fixed_upay_record(record, gateway_dict)
  120. except Exception as e:
  121. click.echo('exception = {}'.format(str(e)))
  122. @cli.command()
  123. @click.option('-r', '--order_no', prompt = u'please input order no')
  124. def fixed_upay_record_by_order(order_no):
  125. record = RechargeRecord.objects(orderNo = order_no).first()
  126. try:
  127. fixed_upay_record(record = record, gateway_dict = {})
  128. except Exception as e:
  129. click.echo('exception = {}'.format(str(e)))
  130. if __name__ == '__main__':
  131. # cli(sys.argv[1:])
  132. cli()