views.py 36 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998
  1. # -*- coding: utf-8 -*-
  2. # !/usr/bin/env python
  3. import datetime
  4. import hashlib
  5. import logging
  6. import re
  7. import ssl
  8. import subprocess
  9. import time
  10. import urllib2
  11. import simplejson as json
  12. import xmltodict
  13. from django.http import HttpResponse, JsonResponse, HttpResponseBadRequest
  14. from django.views.decorators.gzip import gzip_page
  15. from django.views.generic.base import View
  16. from pymongo.results import UpdateResult
  17. from typing import TYPE_CHECKING, Iterable, Dict
  18. from voluptuous import MultipleInvalid
  19. from apilib.monetary import RMB
  20. from apilib.utils import generate_RSA_key_pairs
  21. from apilib.utils_datetime import to_datetime
  22. from apilib.utils_string import cn
  23. from apps import serviceCache
  24. from apps.web import district
  25. from apps.web.ad.models import Advertisement, AdStatistics
  26. from apps.web.agent.models import MoniApp
  27. from apps.web.common.models import AddressType, WithdrawRecord, WithdrawBanks, WithdrawDistrict, \
  28. WithdrawBranchBanks, WithdrawBankCard
  29. from apps.web.common.models import FrontendLog, FAQ
  30. from apps.web.common.transaction.pay.alipay import AliPayWithdrawNotifier
  31. from apps.web.common.utils import WechatMessage, WechatText, WechatMenu, WechatSubscribe, WechatUnSubscribe, \
  32. WechatScanEvent, MessageHandler
  33. from apps.web.constant import Const, AppPlatformType, AdSpace, USER_RECHARGE_TYPE
  34. from apps.web.core import ROLE
  35. from apps.web.core.exceptions import InvalidFileSize, InvalidFileName
  36. from apps.web.core.file import AliOssFileUploader, WechatSubscriptionAccountVerifyFileUploader
  37. from apps.web.core.models import WechatPayApp, AliApp
  38. from apps.web.core.payment.ali import AliPayGateway
  39. from apps.web.core.utils import DefaultJsonErrorResponse, JsonOkResponse, JsonErrorResponse
  40. from apps.web.dealer.models import Dealer
  41. from apps.web.device.models import Device
  42. from apps.web.device.timescale import PowerManager
  43. from apps.web.user.models import MoniUser, AskRobot
  44. from apps.web.user.models import MyUser
  45. from apps.web.user.utils import get_consume_order
  46. from apps.web.utils import error_tolerate, permission_required, concat_dealer_main_page_url, \
  47. concat_front_end_url, concat_user_center_url, concat_user_login_entry_url
  48. from apps.web.validation import SaveWithdrawBankCardSchema, SaveWithdrawAlipaySchema, SaveWithdrawWechatSchema
  49. from library.misc import BankAPI
  50. if TYPE_CHECKING:
  51. from apps.web.core.db import RoleBaseDocument
  52. from django.core.handlers.wsgi import WSGIRequest
  53. from apps.web.device.models import DeviceDict
  54. from apps.web.core.adapter.bolai_gateway import ChargingGatewayBox
  55. logger = logging.getLogger(__name__)
  56. @permission_required(ROLE.dealer, ROLE.agent, ROLE.supermanager)
  57. def loadDistrictData(request):
  58. return JsonResponse({'result': 1, 'description': None, 'payload': district.DISTRICT})
  59. @error_tolerate(nil = lambda: JsonResponse({'result': 0, 'description': u'系统错误'}), logger = logger)
  60. def logFrontend(request):
  61. """
  62. 收集来自前端的日志
  63. :param request:
  64. :return:
  65. """
  66. if not request.body:
  67. return JsonResponse({'result': 1, 'description': u'BODY为空'})
  68. payload = json.loads(request.body)
  69. if 'onlyLog' in payload and payload['onlyLog']:
  70. logger.debug('--[BLUETOOTH]onlyLog--' + str(payload) + '--eof onlyLog--')
  71. else:
  72. FrontendLog(**payload).save()
  73. return JsonResponse({'result': 1, 'description': u'记录成功'})
  74. def getFAQ(request):
  75. """
  76. 使用的常见说明
  77. :param request:
  78. :return:
  79. """
  80. current_user = request.user # type: RoleBaseDocument
  81. faqs = FAQ.get_by_role(current_user.role)
  82. return JsonResponse({
  83. "result": 1,
  84. "description": None,
  85. "payload": {'dataList': [faq.to_dict() for faq in faqs], 'total': len(faqs)}
  86. })
  87. def getNotifications(request):
  88. """
  89. 做一个临时的用户通知 如果可以使用京东引流 就通知经销商
  90. :param request:
  91. :return:
  92. """
  93. return JsonResponse(
  94. {
  95. "result": 1,
  96. "description": None,
  97. "payload": {
  98. "total": 0,
  99. "dataList": []
  100. }
  101. }
  102. )
  103. # @trace_call()
  104. # def wx_biz_dispatch(request):
  105. # def response_xml(payload):
  106. # xmlData = dicttoxml.dicttoxml(payload, attr_type = False, custom_root = 'xml', cdata = True)
  107. # return HttpResponse(xmlData)
  108. #
  109. # logger.info('has receive wechat dispatch message')
  110. #
  111. # def check_signature(signature, timestamp, nonce, token):
  112. # args = [token, timestamp, nonce]
  113. # args.sort()
  114. # return hashlib.sha1(''.join(args).encode('utf-8')).hexdigest() == signature
  115. #
  116. # if request.method == 'GET':
  117. # signature = request.GET['signature']
  118. # timestamp = request.GET['timestamp']
  119. # nonce = request.GET['nonce']
  120. # echostr = request.GET['echostr']
  121. # if check_signature(signature, timestamp, nonce, settings.MY_WECHAT_USER_TOKEN):
  122. # return HttpResponse(str(echostr))
  123. # else:
  124. # return HttpResponse('')
  125. # else:
  126. # replyData = event_reply(request)
  127. # logger.info('calc replaydata = %s' % replyData)
  128. # if replyData:
  129. # return response_xml(replyData)
  130. #
  131. # return HttpResponse('success')
  132. @error_tolerate(nil = lambda: JsonResponse({'result': 0, 'description': u'文件操作有误'}), logger = logger)
  133. @permission_required(ROLE.dealer, ROLE.manager, ROLE.myuser, ROLE.advertiser, ROLE.supermanager, ROLE.subaccount)
  134. def uploadFile(request):
  135. """
  136. 公共的文件上传接口
  137. TODO 需要注意安全选项!!
  138. ..added 2018/03/21 添加参数判断安全路径
  139. :param request:
  140. :return:
  141. """
  142. UPLOADER_MAP = {
  143. 'wechatValidate': WechatSubscriptionAccountVerifyFileUploader,
  144. 'public': AliOssFileUploader
  145. }
  146. Uploader = UPLOADER_MAP.get(request.GET.get('category', 'public'))
  147. uploader = Uploader(inputFile = request.FILES.getlist('file')[0], uploadType = request.GET.get('type', 'others'))
  148. logger.info('[upload] %s is being used' % (repr(uploader),))
  149. try:
  150. outputUrl = uploader.upload()
  151. return JsonResponse({'result': 1, 'description': '', 'payload': outputUrl})
  152. except InvalidFileSize, e:
  153. logger.info(
  154. '%s(id=%s)\'s uploaded file reached limit' % (request.user.__class__.__name__, str(request.user.id),))
  155. return JsonResponse({'result': 0, 'description': e.message, 'payload': {}})
  156. except InvalidFileName, e:
  157. logger.info(
  158. '%s(id=%s)\'s uploaded file name is not legal' % (request.user.__class__.__name__, str(request.user.id),))
  159. return JsonResponse({'result': 0, 'description': e.message, 'payload': {}})
  160. @error_tolerate(nil = DefaultJsonErrorResponse, logger = logger)
  161. @permission_required(ROLE.manager, ROLE.advertisement, ROLE.advertiser, ROLE.dealer, ROLE.agent, ROLE.subaccount)
  162. def getAddressType(request):
  163. # type: (WSGIRequest)->JsonResponse
  164. """
  165. 获取地址类型
  166. :param request:
  167. :return:
  168. """
  169. addressTypes = AddressType.objects.all().order_by('-showWeight') # type: Iterable[AddressType]
  170. dataList = [{'value': addressType.value, 'label': addressType.label} for addressType in addressTypes]
  171. return JsonResponse(
  172. {
  173. 'result': 1,
  174. 'description': '',
  175. 'payload':
  176. {
  177. 'dataList': dataList,
  178. 'total': len(dataList)
  179. }
  180. }
  181. )
  182. @error_tolerate(nil = DefaultJsonErrorResponse, logger = logger)
  183. def getVersion(request):
  184. version = subprocess.check_output(["git", "log", "-1", "--format=%cd"]).strip()
  185. return JsonOkResponse(payload = {'version': version})
  186. @error_tolerate(nil = DefaultJsonErrorResponse, logger = logger)
  187. @permission_required(ROLE.supermanager, ROLE.manager)
  188. def generateNewAppKeyPair(request):
  189. # type: (WSGIRequest)->JsonResponse
  190. """
  191. 生成新的应用私钥公匙对, 返回给前台私钥的地址和公钥的内容 (方便前台去粘贴给支付宝)
  192. :param request:
  193. :return:
  194. """
  195. payload = json.loads(request.body)
  196. appid = payload.get('appid')
  197. if not appid:
  198. return JsonErrorResponse(description = u'appid未传入')
  199. key = 'alipay_app_private_{appid}_{timestamp}'.format(appid = appid, timestamp = int(time.time() * 1000))
  200. pair = generate_RSA_key_pairs()
  201. serviceCache.set(key, pair.private, 3600)
  202. appPublicKeyContent = re.sub(r'-.*', '', pair.public)
  203. return JsonOkResponse(description = u'生成成功', payload = {'path': key, 'appPublicKeyContent': appPublicKeyContent})
  204. @error_tolerate(nil = DefaultJsonErrorResponse, logger = logger)
  205. @permission_required(ROLE.supermanager, ROLE.manager)
  206. def verifyAppKeyPairs(request):
  207. # type: (WSGIRequest)->JsonResponse
  208. """
  209. :param request:
  210. :return:
  211. """
  212. return JsonErrorResponse(description = u'暂不开放')
  213. @error_tolerate(nil = DefaultJsonErrorResponse)
  214. @permission_required(ROLE.supermanager)
  215. def getAliAppList(request):
  216. # type: (WSGIRequest)->JsonResponse
  217. """
  218. 获取阿里APP列表
  219. :param request:
  220. :return:
  221. """
  222. page_index = int(request.GET.get('pageIndex', 1))
  223. page_size = int(request.GET.get('pageSize', 10))
  224. search_key = request.GET.get('searchKey', None)
  225. total, items = AliApp.list(page_index = page_index, page_size = page_size, search_key = search_key)
  226. return JsonResponse({'result': 1,
  227. 'description': None,
  228. 'payload': {
  229. 'total': total,
  230. 'dataList': items
  231. }
  232. })
  233. @error_tolerate(nil = DefaultJsonErrorResponse)
  234. @permission_required(ROLE.supermanager)
  235. def editAliApp(request):
  236. # type: (WSGIRequest)->JsonResponse
  237. """
  238. 编辑阿里APP
  239. :param request:
  240. :return:
  241. """
  242. payload = json.loads(request.body) # type: dict
  243. id = payload.pop('id', None)
  244. update = payload
  245. if id:
  246. query = {'_id': id}
  247. else:
  248. query = {'appid': payload.get('appid')}
  249. result = AliApp.get_collection().update_one(query, update, upsert = True) # type: UpdateResult
  250. return JsonResponse({
  251. 'result': 1 if (result.modified_count == 1 or result.upserted_id) else 0,
  252. 'description': u'修改成功' if (result.modified_count == 1 or result.upserted_id) else u'修改失败',
  253. 'payload': {
  254. 'id': str(result.upserted_id) if result.upserted_id else id
  255. }})
  256. @error_tolerate(nil = DefaultJsonErrorResponse, logger = logger)
  257. @permission_required(ROLE.supermanager, ROLE.manager)
  258. def getWechatFundApp(request):
  259. # type: (WSGIRequest)->JsonResponse
  260. """
  261. 更改微信资金平台APP信息
  262. :param request:
  263. :return:
  264. """
  265. appid = request.GET.get('appid')
  266. mchid = request.GET.get('mchid')
  267. app = WechatPayApp.objects(appid = appid, mchid = mchid).first() # type: WechatPayApp
  268. if not app:
  269. return JsonOkResponse(payload = {})
  270. else:
  271. return JsonOkResponse(payload = app.to_dict())
  272. @error_tolerate(nil = DefaultJsonErrorResponse)
  273. @permission_required(ROLE.supermanager, ROLE.manager)
  274. def editWechatFundApp(request):
  275. # type: (WSGIRequest)->JsonResponse
  276. """
  277. 编辑阿里APP
  278. :param request:
  279. :return:
  280. """
  281. payload = json.loads(request.body) # type: dict
  282. id = payload.pop('id', None)
  283. update = payload
  284. if id:
  285. query = {'_id': id}
  286. else:
  287. query = {'appid': payload.get('appid')}
  288. result = WechatPayApp.get_collection().update_one(query, update, upsert = True) # type: UpdateResult
  289. return JsonResponse({
  290. 'result': 1 if (result.modified_count == 1 or result.upserted_id) else 0,
  291. 'description': u'修改成功' if (result.modified_count == 1 or result.upserted_id) else u'修改失败',
  292. 'payload': {
  293. 'id': str(result.upserted_id) if result.upserted_id else id
  294. }})
  295. @error_tolerate(nil = DefaultJsonErrorResponse)
  296. @permission_required(ROLE.supermanager, ROLE.manager)
  297. def getWechatAppList(request):
  298. # type: (WSGIRequest)->JsonResponse
  299. """
  300. 获取微信支付APP列表
  301. :param request:
  302. :return:
  303. """
  304. page_index = int(request.GET.get('pageIndex', 1))
  305. page_size = int(request.GET.get('pageSize', 10))
  306. search_key = request.GET.get('searchKey', None)
  307. total, items = WechatPayApp.list(page_index = page_index, page_size = page_size, search_key = search_key)
  308. return JsonResponse({'result': 1,
  309. 'description': None,
  310. 'payload': {
  311. 'total': total,
  312. 'dataList': items
  313. }
  314. })
  315. def event_reply(request):
  316. logger.info('has receive event_reply dispatch message = %s' % request.body)
  317. raw = xmltodict.parse(request.body)['xml']
  318. msgType = raw.get('MsgType', '')
  319. event = raw.get('Event', '')
  320. if msgType == 'event' and event in ['subscribe', 'unsubscribe']:
  321. moniOpenId = raw['FromUserName']
  322. rawAppId = raw['ToUserName']
  323. try:
  324. moniApp = MoniApp.objects.get(rawAppId = rawAppId)
  325. except Exception, e:
  326. return ''
  327. try:
  328. moniUser = MoniUser.objects.filter(moniAppId = moniApp.appId, moniOpenId = moniOpenId).first()
  329. if moniUser:
  330. if event == 'subscribe':
  331. moniUser.subTime = datetime.datetime.now()
  332. moniUser.isSubscribe = True
  333. AskRobot.reply(event, moniOpenId, '', moniApp.appId, moniApp.secret, openId = moniUser.openId)
  334. else:
  335. moniUser.unsubTime = datetime.datetime.now()
  336. moniUser.isSubscribe = False
  337. moniUser.save()
  338. else:
  339. user = MyUser.objects.filter(managerialOpenId = raw['FromUserName']).first()
  340. if user is None:
  341. newUser = MoniUser(moniAppId = moniApp.appId, moniOpenId = moniOpenId,
  342. subTime = datetime.datetime.now(), isSubscribe = True)
  343. newUser.save()
  344. else:
  345. AskRobot.reply(event, moniOpenId, '', moniApp.appId, moniApp.secret, openId = user.openId,
  346. agentId = user.agentId)
  347. except Exception, e:
  348. logger.exception('event_reply error=%s' % e)
  349. return ''
  350. elif msgType == 'text':
  351. try:
  352. # 如果是监督号,就直接回复。如果不是监督号,可能是用户的管理app,这个时候,需要走另外的流程。用户付费后关注
  353. moniApp = MoniApp.objects.filter(rawAppId = raw['ToUserName']).first()
  354. if moniApp:
  355. try:
  356. logger.info('111')
  357. moniUser = MoniUser.objects.filter(moniAppId = moniApp.appId,
  358. moniOpenId = raw['FromUserName']).first()
  359. if moniUser and moniUser.openId and moniUser.subAgentId:
  360. logger.info('222')
  361. logger.info('content = %s,openId=%s' % (raw['Content'], moniUser.openId))
  362. AskRobot.reply(event, raw['FromUserName'], raw['Content'], moniApp.appId, moniApp.secret,
  363. openId = moniUser.openId, agentId = moniUser.subAgentId)
  364. else:
  365. logger.info('333')
  366. user = MyUser.objects.filter(managerialOpenId = raw['FromUserName']).first()
  367. if user is None:
  368. logger.info('444')
  369. return {'MsgType': 'text', 'FromUserName': raw['ToUserName'],
  370. 'ToUserName': raw['FromUserName'], 'Content': u'没有找到您在平台扫码使用设备的任何信息',
  371. 'CreateTime': int(time.time())}
  372. AskRobot.reply(event, raw['FromUserName'], raw['Content'], moniApp.appId, moniApp.secret,
  373. openId = user.openId, agentId = user.agentId)
  374. except Exception, e:
  375. logger.exception('event_reply error=%s' % e)
  376. return ''
  377. logger.info('555')
  378. return ''
  379. except Exception, e:
  380. return ''
  381. return ''
  382. @error_tolerate(logger = logger, nil = JsonErrorResponse(u'故障部件图片存储失败,请重新上传'))
  383. @permission_required(ROLE.dealer, ROLE.subaccount)
  384. def uploadFaultPartPic(request):
  385. # type: (WSGIRequest)->JsonResponse
  386. files = request.FILES.getlist('file')
  387. if not len(files):
  388. return JsonResponse({'result': 0, 'description': u'故障部件图片存储失败,请重新上传', 'payload': ''})
  389. uploader = AliOssFileUploader(inputFile = request.FILES.getlist('file')[0], uploadType = 'fault')
  390. logger.info('[uploadFaultPartPic] %s is being used' % (repr(uploader),))
  391. try:
  392. outputUrl = uploader.upload()
  393. return JsonResponse({'result': 1, 'description': '', 'payload': outputUrl})
  394. except InvalidFileSize, e:
  395. logger.info(
  396. '%s(id=%s)\'s uploaded file reached limit' % (request.user.__class__.__name__, str(request.user.id),))
  397. return JsonResponse({'result': 0, 'description': e.message, 'payload': {}})
  398. except InvalidFileName, e:
  399. logger.info(
  400. '%s(id=%s)\'s uploaded file name is not legal' % (request.user.__class__.__name__, str(request.user.id),))
  401. return JsonResponse({'result': 0, 'description': e.message, 'payload': {}})
  402. def queryExpressInfo(request):
  403. orderNo = request.GET.get('orderNo')
  404. bodys = {}
  405. url = 'https://kuaidid.market.alicloudapi.com/danhao'
  406. bodys['com'] = ''
  407. bodys['src'] = orderNo
  408. from six.moves.urllib import parse
  409. post_data = parse.urlencode(bodys)
  410. request = urllib2.Request(url, post_data)
  411. request.add_header('Authorization', 'APPCODE ' + Const.KUAIDI_API_APPCODE)
  412. request.add_header('Content-Type', 'application/x-www-form-urlencoded; charset=UTF-8')
  413. ctx = ssl.create_default_context()
  414. ctx.check_hostname = False
  415. ctx.verify_mode = ssl.CERT_NONE
  416. response = urllib2.urlopen(request, timeout = 15, context = ctx)
  417. content = response.read()
  418. if (content):
  419. result = json.loads(content)
  420. if result['status'] == 200:
  421. resultList = []
  422. for info in result['msg']['context']:
  423. resultList.append({'time': datetime.datetime.fromtimestamp(info['time']).strftime(Const.DATETIME_FMT),
  424. 'desc': info['desc']})
  425. return JsonOkResponse(payload = {'dataList': resultList})
  426. else:
  427. return JsonErrorResponse(description = u'无法获取到订单详细信息')
  428. else:
  429. return JsonErrorResponse(description = u'无法获取到订单详细信息')
  430. @error_tolerate(logger = logger)
  431. def aliNotify(request):
  432. payload = request.POST.dict() # type: Dict
  433. logger.info('ali common notify: %s' % str(payload))
  434. if 'service' in payload and payload['service'] == 'alipay.service.check':
  435. biz_content = xmltodict.parse(payload['biz_content'])['XML']
  436. app = AliApp.objects(appid = biz_content['AppId']).first() # type: AliApp
  437. gateway = app.new_gateway(app_platform_type = AppPlatformType.ALIPAY) # type: AliPayGateway
  438. if gateway.check(data = payload, pop_sign_type = False):
  439. sign = gateway.client._sign('<success>true</success>')
  440. return HttpResponse(
  441. '<?xml version="1.0" encoding="GBK"?><alipay><response><success>true</success></response><app_cert_sn>{}</app_cert_sn><sign>{}</sign><sign_type>RSA2</sign_type></alipay>'.format(
  442. gateway.client.app_cert_sn, sign))
  443. if payload['msg_method'] == 'alipay.fund.trans.order.changed':
  444. payload['biz_content'] = json.loads(payload['biz_content'])
  445. order_no = payload['biz_content']['out_biz_no']
  446. if WithdrawRecord.is_my(order_no):
  447. return AliPayWithdrawNotifier(record_cls = WithdrawRecord).do(payload)
  448. else:
  449. logger.warn('not support this order. orderNo = {}'.format(order_no))
  450. else:
  451. logger.warn('not support this method = {}'.format(payload['msg_method']))
  452. return HttpResponse('success')
  453. @error_tolerate(logger=logger, nil=JsonErrorResponse(description=u"获取订单失败,请刷新"))
  454. def getReceipt(request): # type: (WSGIRequest)->JsonResponse
  455. """
  456. 获取支付结果
  457. 消费单的结果直接跳转回去
  458. """
  459. user = request.user # type: RoleBaseDocument
  460. order_type = request.GET.get('order_type', 'recharge')
  461. out_trade_no = request.GET.get('out_trade_no')
  462. check_code = request.GET.get('check_code', None)
  463. from apps.web.user.models import RechargeRecord, ConsumeRecord
  464. if order_type == 'consume':
  465. record = get_consume_order(out_trade_no) # type: ConsumeRecord
  466. payload = record.receiptDesc
  467. else:
  468. record = RechargeRecord.objects(__raw__={
  469. '$or': [
  470. {
  471. 'wxOrderNo': out_trade_no
  472. },
  473. {
  474. 'orderNo': out_trade_no
  475. }
  476. ]}).first() # type: RechargeRecord
  477. if not record:
  478. from apps.web.dealer.models import DealerRechargeRecord
  479. record = DealerRechargeRecord.objects(__raw__={
  480. '$or': [
  481. {
  482. 'wxOrderNo': out_trade_no
  483. },
  484. {
  485. 'orderNo': out_trade_no
  486. }
  487. ]}).first() # type: DealerRechargeRecord
  488. if not record:
  489. return JsonErrorResponse(description = u'订单获取失败,请重新获取')
  490. else:
  491. payload = {
  492. 'itemName': record.subject,
  493. 'orderNo': record.orderNo,
  494. 'payment': RMB.fen_to_yuan(record.totalFee),
  495. 'homeLink': concat_dealer_main_page_url(),
  496. 'finishedTime': record.finishedTime,
  497. 'adInfo': {
  498. 'adShow': 'noshow'
  499. }
  500. }
  501. if record.product == DealerRechargeRecord.ProductType.SimCard:
  502. payload.update(
  503. {'detailLink': concat_front_end_url(
  504. uri = '/app/payOrderDetail.html?orderNo={}'.format(record.orderNo))})
  505. return JsonOkResponse(payload = payload)
  506. else:
  507. if record.result != RechargeRecord.PayResult.SUCCESS:
  508. return JsonErrorResponse(description = u'订单回调还未调用')
  509. payload = {
  510. 'orderNo': record.orderNo,
  511. 'payment': record.money,
  512. 'finishedTime': record.finished_time
  513. }
  514. from apps.web.device.models import Device
  515. # 是否是启动设备的
  516. start_key = record.startKey
  517. if start_key:
  518. payload.update({
  519. 'startKey': start_key,
  520. 'devNo': record.devNo,
  521. 'logicalCode': record.logicalCode,
  522. 'homeLink': concat_user_login_entry_url(l = record.logicalCode),
  523. 'serviceLink': concat_user_center_url(l = record.logicalCode)
  524. })
  525. else:
  526. payload.update({
  527. 'serviceLink': concat_user_center_url(),
  528. 'homeLink': concat_user_center_url(),
  529. 'detailLink': concat_front_end_url(
  530. uri = '/user/index.html#/user/chargeRecordDetail?ownerId={}&id={}'.format(record.ownerId,
  531. str(record.id)))
  532. })
  533. if isinstance(record, RechargeRecord):
  534. payload['itemName'] = record.subject
  535. if record.devNo:
  536. device = Device.get_dev(record.devNo)
  537. else:
  538. device = None
  539. else:
  540. device = Device.get_dev(record.devNo)
  541. if record.port:
  542. payload['itemName'] = cn(u'{major_type}/{logicalCode}-端口{port}'.format(
  543. major_type = device.majorDeviceType[0:8], logicalCode = record.logicalCode,
  544. port = record.port))
  545. else:
  546. payload['itemName'] = cn(u'{major_type}/{logicalCode}'.format(
  547. major_type = device.majorDeviceType[0:8], logicalCode = record.logicalCode))
  548. if not check_code:
  549. owner = Dealer.objects(id=record.ownerId).first() # type: Dealer
  550. ad_dict = Advertisement.fetch_payafter_ad(
  551. ua=request.META.get('HTTP_USER_AGENT', ''), user=request.user, dealer=owner, device=device)
  552. logger.debug('ad dict is {}'.format(ad_dict))
  553. payload.update({'adInfo': ad_dict})
  554. else:
  555. AdStatistics.inc_make_count(adSpace=AdSpace.PAYAFTER, showType='payAfter_dianjin', gateway='wechat')
  556. return JsonOkResponse(payload=payload)
  557. def bolaiEvent(request):
  558. payload = json.loads(request.body)
  559. logger.debug('received bolai event: {}'.format(payload))
  560. gateId = payload.get('gateway_id')
  561. gateway_dev = Device.get_dev(gateId) # type: DeviceDict
  562. if gateway_dev is None: # 可能是一体化设备
  563. rcds = Device.get_collection().find({'gateImei': gateId})
  564. if rcds.count() == 0:
  565. return JsonOkResponse()
  566. gateway_dev = Device.get_dev(rcds[0]['devNo'])
  567. logger.info('this is gatewayplug device')
  568. if gateway_dev is None:
  569. return JsonOkResponse()
  570. if 'data' in payload and 'node_index' in payload['data']:
  571. if payload['data']['node_index'] != 0:
  572. logger.info('this is nodeplug device')
  573. gateway_adapter = gateway_dev.deviceAdapter # type: ChargingGatewayBox
  574. node_devno = gateway_adapter.get_node_devNo(str(payload['data']['node_index']))
  575. node_dev = Device.get_dev(node_devno) # type: DeviceDict
  576. if not node_dev:
  577. return JsonOkResponse()
  578. payload.update({
  579. 'nodeType': 'node'
  580. })
  581. event = node_dev.eventer.getEvent(payload)
  582. event.do()
  583. else:
  584. logger.info('this is gateway device')
  585. payload.update({
  586. 'nodeType': 'node'
  587. })
  588. event = gateway_dev.eventer.getEvent(payload)
  589. event.do()
  590. else:
  591. payload.update({
  592. 'nodeType': 'gateway'
  593. })
  594. event = gateway_dev.eventer.getEvent(payload)
  595. event.do()
  596. return JsonOkResponse()
  597. def bolaitenEvent(request):
  598. payload = json.loads(request.body)
  599. logger.debug('received bolai event: {}'.format(payload))
  600. devNo = payload.get('gateway_id')
  601. dev = Device.get_dev(devNo)
  602. if dev is None:
  603. return JsonOkResponse()
  604. if 'data' in payload:
  605. event = dev.eventer.getEvent(payload)
  606. event.do()
  607. return JsonOkResponse()
  608. @error_tolerate(nil = DefaultJsonErrorResponse)
  609. @permission_required(ROLE.dealer, ROLE.supermanager, ROLE.myuser)
  610. def getPowerGraphByDevice(request):
  611. startTime = to_datetime(request.GET.get('startTime'))
  612. endTime = to_datetime(request.GET.get('endTime'))
  613. port = str(request.GET.get('port', '1'))
  614. device = Device.get_dev_by_logicalCode(request.GET.get('logicalCode')) # type: DeviceDict
  615. if device.devTypeCode in [Const.DEVICE_TYPE_CODE_CHANGING_YUEWANGTONG]:
  616. interval = 600
  617. else:
  618. interval = device.cycle
  619. items = PowerManager.instence().get(devNo = device.devNo, port = port, sTime = startTime, eTime = endTime,
  620. interval = interval)
  621. return JsonOkResponse(payload = {'dataList': items})
  622. class WechatFollowView(View):
  623. def __init__(self, **kwargs):
  624. super(WechatFollowView, self).__init__(**kwargs)
  625. self.app = None
  626. @staticmethod
  627. def auth_request(request, app): # type:(WSGIRequest, MoniApp) -> bool
  628. """
  629. 对微信的请求进行验签
  630. """
  631. token = app.appToken
  632. signature = request.GET.get('signature')
  633. timestamp = request.GET.get('timestamp')
  634. nonce = request.GET.get('nonce')
  635. params = [token, timestamp, nonce]
  636. params.sort()
  637. return hashlib.sha1("".join(params)).hexdigest() == signature
  638. @staticmethod
  639. def get_followed_app(rawAppId):
  640. return MoniApp.get_app_by_raw(rawAppId)
  641. def dispatch(self, request, *args, **kwargs):
  642. rawAppId = kwargs.pop("rawAppId", "")
  643. app = self.get_followed_app(rawAppId)
  644. # 查找是否是入库的app
  645. if not app:
  646. logger.warning("[WechatFollowView dispatch], not find app! rawAppId = {}".format(rawAppId))
  647. return HttpResponseBadRequest()
  648. # 查找是否通过校验
  649. if not self.auth_request(request, app):
  650. logger.warning("[WechatFollowView dispatch], verify server error! params = {}".format(request.GET))
  651. return HttpResponseBadRequest()
  652. self.app = app
  653. return super(WechatFollowView, self).dispatch(request)
  654. def get(self, request):
  655. """
  656. 微信公众号服务器验证token使用 配置的时候只会访问一次
  657. """
  658. logger.info("[WechatFollowView GET], verify server success! params = {}, app = {}".format(request.GET, self.app))
  659. echoStr = request.GET["echostr"]
  660. return HttpResponse(echoStr)
  661. def post(self, request):
  662. """
  663. 业务请求都在这里
  664. 大致分为两类
  665. 1.业务事件(关注、订阅、点击菜单)
  666. 2.用户互动文字
  667. """
  668. logger.info("[WechatFollowView POST], payload = {}, app = {}".format(request.body, self.app))
  669. message = WechatMessage(request.body)
  670. if message.isText:
  671. response = WechatText(message, self.app).handle()
  672. elif message.isEvent:
  673. if message.isScanEvent:
  674. response = WechatScanEvent(message, self.app).handle()
  675. elif message.isSubscribeEvent:
  676. response = WechatSubscribe(message, self.app).handle()
  677. elif message.isUnSubscribeEvent:
  678. response = WechatUnSubscribe(message, self.app).handle()
  679. elif message.isMenuEvent:
  680. response = WechatMenu(message, self.app).handle()
  681. else:
  682. response = MessageHandler.defaultResponse()
  683. else:
  684. response = MessageHandler.defaultResponse()
  685. return response
  686. @permission_required(ROLE.dealer, ROLE.agent, ROLE.supermanager)
  687. def loadWithdrawArea(request):
  688. province = request.GET.get("province", None)
  689. return JsonOkResponse(payload = {'dataList': WithdrawDistrict.get_area(province)})
  690. @error_tolerate(logger = logger, nil = JsonErrorResponse(description = u"获取数据失败,请刷新"))
  691. @permission_required(ROLE.dealer, ROLE.agent, ROLE.supermanager)
  692. def supportedWithdrawBanks(request):
  693. # type: (WSGIRequest)->JsonResponse
  694. keyWord = request.GET.get('keyWord', None)
  695. chooseRange = request.GET.get('range', 'wechat')
  696. if chooseRange == 'all':
  697. return JsonOkResponse(payload = {'dataList': WithdrawBanks.get_all_banks(keyWord)})
  698. elif chooseRange == 'wechat':
  699. return JsonOkResponse(payload = {'dataList': WithdrawBanks.get_wechat_support_banks(keyWord)})
  700. else:
  701. return JsonErrorResponse(description = u'参数错误')
  702. @error_tolerate(logger = logger, nil = JsonErrorResponse(description = u"获取数据失败,请刷新"))
  703. @permission_required(ROLE.dealer, ROLE.agent, ROLE.supermanager)
  704. def getWithdrawBankByAccount(request):
  705. # type: (WSGIRequest)->JsonResponse
  706. accountCode = request.GET.get('accountCode', None)
  707. if not accountCode:
  708. return JsonResponse({'result': 2, 'description': u'银行账号不能为空', 'payload': {}})
  709. bank_info = BankAPI().get_bank_card_info(card_no = accountCode)
  710. if bank_info:
  711. rv = WithdrawBanks.get_by_bank_abbrev_code(bank_info['bankAbbrevCode'])
  712. if rv:
  713. return JsonOkResponse(payload = rv)
  714. else:
  715. logger.warning('not find this bank: {}'.format(bank_info))
  716. return JsonResponse({'result': 2, 'description': u'该银行卡不支持', 'payload': {}})
  717. else:
  718. return JsonResponse({'result': 2, 'description': u'没有查到该银行账号信息', 'payload': {}})
  719. @error_tolerate(logger = logger, nil = JsonErrorResponse(description = u"获取数据失败,请刷新"))
  720. @permission_required(ROLE.dealer, ROLE.agent, ROLE.supermanager)
  721. def withdrawBranchBankList(request):
  722. # type: (WSGIRequest)->JsonResponse
  723. provinceCode = request.GET.get('provinceCode')
  724. cityCode = request.GET.get('cityCode')
  725. bankCode = request.GET.get('bankCode')
  726. keyWord = request.GET.get('keyWord')
  727. queryset = WithdrawBranchBanks.objects(**{
  728. 'provinceCode': provinceCode,
  729. 'cityCode': cityCode,
  730. 'bankCode': bankCode
  731. }).search(keyWord)
  732. rv = [
  733. {
  734. 'code': item.code,
  735. 'name': item.name
  736. } for item in queryset
  737. ]
  738. return JsonOkResponse(payload = {'dataList': rv})
  739. @error_tolerate(nil = DefaultJsonErrorResponse)
  740. @permission_required(ROLE.dealer, ROLE.agent)
  741. def saveWithdrawBankCard(request):
  742. # type: (WSGIRequest)->JsonResponse
  743. """
  744. 添加银行卡
  745. """
  746. ownerId = str(request.user.id)
  747. if WithdrawBankCard.objects.filter(ownerId = ownerId, role = request.user.role).count() > 5:
  748. return JsonResponse({"result": 2, "description": u"银行卡数量已添加至上限(5),不允许继续添加"})
  749. try:
  750. payload = SaveWithdrawBankCardSchema(json.loads(request.body))
  751. except MultipleInvalid as e:
  752. logger.exception(e)
  753. return JsonErrorResponse(description = u"信息不完整")
  754. if WithdrawBankCard.objects(ownerId = ownerId, role = request.user.role,
  755. accountCode = payload['accountCode']).first():
  756. return JsonResponse({"result": 2, "description": u"请勿重复添加"})
  757. accountType = payload.pop('accountType')
  758. if accountType == WithdrawBankCard.AccountType.PERSONAL:
  759. item = WithdrawBankCard.new_personal_withdraw_bank_card(ownerId = ownerId, role = request.user.role, **payload)
  760. return JsonOkResponse(payload = {"id": str(item.id)})
  761. else:
  762. return JsonErrorResponse(description = u"不支持添加对公提现银行卡,请您开通商户。")
  763. @error_tolerate(nil = DefaultJsonErrorResponse)
  764. @permission_required(ROLE.dealer, ROLE.agent)
  765. def unbindWithdrawBankCard(request):
  766. # type: (WSGIRequest)->JsonResponse
  767. ownerId = str(request.user.id)
  768. payload = json.loads(request.body)
  769. accountCode = payload.get('accountCode')
  770. if not accountCode:
  771. return JsonResponse({"result": 2, "description": u"参数错误,请刷新后重试"})
  772. try:
  773. card = WithdrawBankCard.objects.get(ownerId = ownerId, role = request.user.role, accountCode = accountCode)
  774. except Exception as e:
  775. logger.exception(e)
  776. return JsonResponse({"result": 2, "description": u"解绑定失败,请联系客服解决。"})
  777. else:
  778. card.delete()
  779. return JsonOkResponse()
  780. @error_tolerate(nil = DefaultJsonErrorResponse)
  781. @permission_required(ROLE.dealer, ROLE.agent)
  782. def getWithdrawBankCards(request):
  783. # type: (WSGIRequest)->JsonResponse
  784. ownerId = str(request.user.id)
  785. bankCards = WithdrawBankCard.objects.filter(ownerId = ownerId, role = request.user.role)
  786. dataList = list()
  787. for bankCard in bankCards:
  788. dataList.append(bankCard.to_dict())
  789. return JsonOkResponse(payload = {'dataList': dataList})
  790. @error_tolerate(nil = DefaultJsonErrorResponse)
  791. @permission_required(ROLE.dealer, ROLE.agent)
  792. def saveAlipayWithdrawInfo(request):
  793. # type: (WSGIRequest)->JsonResponse
  794. try:
  795. payload = SaveWithdrawAlipaySchema(json.loads(request.body))
  796. except MultipleInvalid as e:
  797. logger.exception(e)
  798. return JsonErrorResponse(description = u"参数错误")
  799. else:
  800. request.user.set_withdraw_alipay(**payload)
  801. return JsonOkResponse()
  802. @error_tolerate(nil = DefaultJsonErrorResponse)
  803. @permission_required(ROLE.dealer, ROLE.agent)
  804. def saveWechatWithdrawInfo(request):
  805. # type: (WSGIRequest)->JsonResponse
  806. try:
  807. payload = SaveWithdrawWechatSchema(json.loads(request.body))
  808. except MultipleInvalid as e:
  809. logger.exception(e)
  810. return JsonErrorResponse(description = u"参数错误")
  811. else:
  812. request.user.set_withdraw_wechat(**payload)
  813. return JsonOkResponse()