|
@@ -0,0 +1,266 @@
|
|
|
+# coding=utf-8
|
|
|
+
|
|
|
+import time
|
|
|
+import os
|
|
|
+from django.conf import settings
|
|
|
+from django.db import models
|
|
|
+from django.utils import timezone
|
|
|
+from apps.WechatTp.models import WechatTp
|
|
|
+from utils.wx.wechat import WeChat
|
|
|
+from utils.exceptions import CustomError
|
|
|
+from utils.file_operation import CertPath
|
|
|
+from apps.tenant.models import Tenant
|
|
|
+
|
|
|
+
|
|
|
+class WechatApplet(models.Model):
|
|
|
+ AUDIT_SUCCESS = 0
|
|
|
+ AUDIT_REJECT = 1
|
|
|
+ AUDITING = 2
|
|
|
+ RECALL = 3
|
|
|
+ AUDIT_DELAY = 4
|
|
|
+
|
|
|
+ AUDIT_STATUS_CHOICE = (
|
|
|
+ (AUDIT_SUCCESS, u'审核通过'),
|
|
|
+ (AUDIT_REJECT, u'审核拒绝'),
|
|
|
+ (AUDITING, u'审核中'),
|
|
|
+ (RECALL, u'已撤回'),
|
|
|
+ (AUDIT_DELAY, u'审核延后'),
|
|
|
+ )
|
|
|
+
|
|
|
+ tenant = models.ForeignKey(Tenant, verbose_name=u'租户', on_delete=models.PROTECT, editable=False)
|
|
|
+ wechat_tp = models.ForeignKey(WechatTp, verbose_name=u'第三方平台', on_delete=models.PROTECT, editable=False)
|
|
|
+ authorizer_appid = models.CharField(max_length=512, verbose_name=u'授权方appid')
|
|
|
+ secret = models.CharField(max_length=512, verbose_name=u'小程序秘钥', null=True)
|
|
|
+ authorizer_refresh_token = models.CharField(max_length=512, verbose_name=u'刷新令牌')
|
|
|
+ authorizer_access_token = models.CharField(max_length=512, verbose_name=u'令牌')
|
|
|
+ access_token_gtime = models.DateTimeField(verbose_name=u"获取令牌时间")
|
|
|
+ expires_in = models.IntegerField(verbose_name=u'令牌有效期')
|
|
|
+ is_authorize = models.BooleanField(verbose_name=u'是否授权', default=False)
|
|
|
+ message_template_id = models.CharField(max_length=255, verbose_name=u'消息模版ID', null=True)
|
|
|
+
|
|
|
+ nick_name = models.CharField(max_length=255, verbose_name=u'昵称', default="")
|
|
|
+ head_img = models.CharField(max_length=255, verbose_name=u'头像', default="")
|
|
|
+ principal_name = models.CharField(max_length=255, verbose_name=u'主体名称', default="")
|
|
|
+ qrcode_url = models.CharField(max_length=512, verbose_name=u'二维码', default="")
|
|
|
+ user_version = models.CharField(max_length=512, verbose_name=u'当前程序版本', default="")
|
|
|
+ template_id = models.CharField(max_length=512, verbose_name=u'当前模板', default="")
|
|
|
+ tenant_num = models.CharField(max_length=512, verbose_name=u'商户号', default="")
|
|
|
+ tenant_key = models.CharField(max_length=512, verbose_name=u'商户密钥', default="")
|
|
|
+
|
|
|
+ apiclient_cert = models.CharField(max_length=255, verbose_name=u'API证书cert', default="")
|
|
|
+ apiclient_key = models.CharField(max_length=255, verbose_name=u'API证书key', default="")
|
|
|
+
|
|
|
+ auditid = models.CharField(max_length=512, verbose_name=u'待审核ID', default="")
|
|
|
+ wait_audit_version = models.CharField(max_length=512, verbose_name=u'待审核版本', null=True)
|
|
|
+ wait_audit_template = models.CharField(max_length=512, verbose_name=u'待审核模板', null=True)
|
|
|
+ audit_status = models.IntegerField(choices=AUDIT_STATUS_CHOICE, verbose_name=u'审核状态', null=True)
|
|
|
+ reject_reason = models.CharField(max_length=512, verbose_name=u'拒绝原因', default="")
|
|
|
+
|
|
|
+ class Meta:
|
|
|
+ db_table = "wechat_applet"
|
|
|
+ ordering = ['-id']
|
|
|
+ index_together = ()
|
|
|
+ verbose_name = u"小程序"
|
|
|
+ default_permissions = ()
|
|
|
+
|
|
|
+ @staticmethod
|
|
|
+ def getById(id):
|
|
|
+ try:
|
|
|
+ id = int(id)
|
|
|
+ except:
|
|
|
+ raise CustomError(u'无效的小程序ID!')
|
|
|
+
|
|
|
+ instance = WechatApplet.objects.filter(pk=id).first()
|
|
|
+ if not instance:
|
|
|
+ raise CustomError(u'未找到相应的小程序!')
|
|
|
+ return instance
|
|
|
+
|
|
|
+ @staticmethod
|
|
|
+ def getByAppid(appid):
|
|
|
+ instance = WechatApplet.objects.filter(authorizer_appid=appid).first()
|
|
|
+ if not instance:
|
|
|
+ raise CustomError(u'未找到相应的小程序!')
|
|
|
+ return instance
|
|
|
+
|
|
|
+ @staticmethod
|
|
|
+ def addAuthorizer(wechat_component, authorization_code, tenant):
|
|
|
+ gtime = timezone.now()
|
|
|
+ component_appid = wechat_component.getAppid()
|
|
|
+ component_access_token = wechat_component.getAccessToken()
|
|
|
+ res = WeChat.getAuthorizationInfo(component_appid, authorization_code, component_access_token)
|
|
|
+ ares = WeChat.getAuthorizerInfo(component_appid, res['authorization_info']['authorizer_appid'], component_access_token)
|
|
|
+ authorizer = WechatApplet.getByAppidAndComponentAppid(res['authorization_info']['authorizer_appid'], component_appid)
|
|
|
+ if authorizer:
|
|
|
+ if authorizer.is_authorize:
|
|
|
+ raise CustomError(u'该小程序已授权!')
|
|
|
+ authorizer.refresh(res['authorization_info'], ares['authorizer_info'], gtime, tenant)
|
|
|
+ else:
|
|
|
+ authorizer = WechatApplet.objects.create(
|
|
|
+ tenant=tenant,
|
|
|
+ wechat_tp=wechat_component,
|
|
|
+ authorizer_appid=res['authorization_info']['authorizer_appid'],
|
|
|
+ authorizer_refresh_token=res['authorization_info']['authorizer_refresh_token'],
|
|
|
+ authorizer_access_token=res['authorization_info']['authorizer_access_token'],
|
|
|
+ access_token_gtime=gtime,
|
|
|
+ expires_in=res['authorization_info']['expires_in'],
|
|
|
+ principal_name=ares['authorizer_info']['principal_name'],
|
|
|
+ nick_name=ares['authorizer_info']['nick_name'],
|
|
|
+ head_img=ares['authorizer_info']['head_img'],
|
|
|
+ qrcode_url=ares['authorizer_info']['qrcode_url'],
|
|
|
+ is_authorize=True
|
|
|
+ )
|
|
|
+ # 设置服务器域名
|
|
|
+ authorizer.setDomain()
|
|
|
+ return authorizer
|
|
|
+
|
|
|
+ @staticmethod
|
|
|
+ def getByAppidAndComponentAppid(appid, component_appid):
|
|
|
+ authorizer = WechatApplet.objects.filter(authorizer_appid=appid, wechat_tp_id__component_appid=component_appid).first()
|
|
|
+ return authorizer
|
|
|
+
|
|
|
+ def getComponent(self):
|
|
|
+ return self.wechat_tp
|
|
|
+
|
|
|
+ def getAccessToken(self):
|
|
|
+ if self.authorizer_access_token:
|
|
|
+ last_time = time.mktime(self.access_token_gtime.timetuple()) + self.expires_in
|
|
|
+ now = time.mktime(timezone.now().timetuple())
|
|
|
+ if last_time > now:
|
|
|
+ return self.authorizer_access_token
|
|
|
+ if not self.authorizer_refresh_token:
|
|
|
+ return ''
|
|
|
+ gtime = timezone.now()
|
|
|
+ component = self.getComponent()
|
|
|
+ res = WeChat.getAuthorizerAccessToken(component.getAppid(), self.authorizer_appid, self.authorizer_refresh_token, component.getAccessToken())
|
|
|
+ self.refreshAccessToken(gtime, res['authorizer_access_token'], res['expires_in'], res['authorizer_refresh_token'])
|
|
|
+ return self.authorizer_access_token
|
|
|
+
|
|
|
+ def revoke(self):
|
|
|
+ tenant = self.tenant
|
|
|
+ tenant.is_bind_app = False
|
|
|
+ tenant.save()
|
|
|
+ self.is_authorize = False
|
|
|
+ self.save()
|
|
|
+
|
|
|
+ def refreshAccessToken(self, gtime, access_token, expires_in, refresh_token):
|
|
|
+ self.authorizer_access_token = access_token
|
|
|
+ self.access_token_gtime = gtime
|
|
|
+ self.expires_in = expires_in
|
|
|
+ self.authorizer_refresh_token = refresh_token
|
|
|
+ self.save()
|
|
|
+
|
|
|
+ def refresh(self, authorization_info, authorizer_info, access_token_gtime, tenant):
|
|
|
+ self.authorizer_refresh_token = authorization_info['authorizer_refresh_token']
|
|
|
+ self.authorizer_access_token = authorization_info['authorizer_access_token']
|
|
|
+ self.access_token_gtime = access_token_gtime
|
|
|
+ self.expires_in = authorization_info['expires_in']
|
|
|
+ self.principal_name = authorizer_info['principal_name']
|
|
|
+ self.nick_name = authorizer_info['nick_name']
|
|
|
+ self.head_img = authorizer_info['head_img']
|
|
|
+ self.qrcode_url = authorizer_info['qrcode_url']
|
|
|
+ self.is_authorize = True
|
|
|
+ self.tenant = tenant
|
|
|
+ self.save()
|
|
|
+
|
|
|
+ def uploadCode(self, template_id, user_version, user_desc):
|
|
|
+ WeChat.commitCode(self.getAccessToken(), template_id, user_version, user_desc)
|
|
|
+ result = WeChat.submitAuditCode(self.getAccessToken())
|
|
|
+ self.auditid = result['auditid']
|
|
|
+ self.wait_audit_version = user_version
|
|
|
+ self.wait_audit_template = template_id
|
|
|
+ self.audit_status = WechatApplet.AUDITING
|
|
|
+ self.reject_reason = ''
|
|
|
+ self.save()
|
|
|
+
|
|
|
+ def refreshAuditStatus(self):
|
|
|
+ result = WeChat.getLastSubmitAuditCodeStatus(self.getAccessToken())
|
|
|
+ if self.auditid == str(result['auditid']):
|
|
|
+ if result['status'] == WechatApplet.AUDIT_SUCCESS:
|
|
|
+ self.weapp_audit_success()
|
|
|
+ elif result['status'] == WechatApplet.AUDIT_REJECT:
|
|
|
+ self.weapp_audit_fail(result['reason'])
|
|
|
+ elif result['status'] == WechatApplet.RECALL:
|
|
|
+ self.weapp_audit_recall()
|
|
|
+ elif result['status'] == WechatApplet.AUDIT_DELAY:
|
|
|
+ self.weapp_audit_delay(result['reason'])
|
|
|
+
|
|
|
+ def weapp_audit_recall(self):
|
|
|
+ self.auditid = ''
|
|
|
+ self.wait_audit_template = None
|
|
|
+ self.wait_audit_version = None
|
|
|
+ self.audit_status = None
|
|
|
+ self.reject_reason = ''
|
|
|
+ self.save()
|
|
|
+
|
|
|
+ def weapp_audit_success(self):
|
|
|
+ self.user_version = self.wait_audit_version
|
|
|
+ self.template_id = self.wait_audit_template
|
|
|
+ self.auditid = ''
|
|
|
+ self.wait_audit_template = None
|
|
|
+ self.wait_audit_version = None
|
|
|
+ self.audit_status = None
|
|
|
+ self.reject_reason = ''
|
|
|
+ self.save()
|
|
|
+
|
|
|
+ def weapp_audit_fail(self, reason):
|
|
|
+ self.audit_status = WechatApplet.AUDIT_REJECT
|
|
|
+ self.reject_reason = reason
|
|
|
+ self.save()
|
|
|
+
|
|
|
+ def weapp_audit_delay(self, reason):
|
|
|
+ self.audit_status = WechatApplet.AUDIT_DELAY
|
|
|
+ self.reject_reason = reason
|
|
|
+ self.save()
|
|
|
+
|
|
|
+ def setDomain(self):
|
|
|
+ requestdomain = uploaddomain = downloaddomain = ['https://baoxiu360.top', ]
|
|
|
+ wsrequestdomain = []
|
|
|
+ WeChat.modify_domain(self.getAccessToken(), 'set', requestdomain, wsrequestdomain, uploaddomain, downloaddomain)
|
|
|
+
|
|
|
+ def addPlugin(self):
|
|
|
+ result = WeChat.addPlugin(self.getAccessToken())
|
|
|
+ return result
|
|
|
+
|
|
|
+ def releaseApplet(self):
|
|
|
+ result = WeChat.releaseCode(self.getAccessToken())
|
|
|
+ return result
|
|
|
+
|
|
|
+ def getMsgTemplateId(self):
|
|
|
+ if self.message_template_id:
|
|
|
+ return self.message_template_id
|
|
|
+ templates = WeChat.getTemplateList(self.getAccessToken())
|
|
|
+ for template in templates:
|
|
|
+ if template['title'] == u'维修完成通知':
|
|
|
+ self.message_template_id = template['priTmplId']
|
|
|
+ self.save()
|
|
|
+ return template['priTmplId']
|
|
|
+ return ''
|
|
|
+
|
|
|
+ def sendMsg(self, openid, name, address, fault_des, no):
|
|
|
+ template_id = self.getMsgTemplateId()
|
|
|
+ if not template_id:
|
|
|
+ return
|
|
|
+ time = timezone.now().strftime('%Y-%m-%d %H:%M:%S')
|
|
|
+ data = {
|
|
|
+ 'time5': {'value': time},
|
|
|
+ 'name4':{'value':name},
|
|
|
+ 'thing3':{'value':address},
|
|
|
+ 'thing2':{'value': fault_des},
|
|
|
+ 'character_string7':{'value': no},
|
|
|
+ }
|
|
|
+ page = 'pages/repairList/repairList?sort=baoxiu&name=我的报修'
|
|
|
+ WeChat.sendSubscribeMessage(self.getAccessToken(), openid, template_id, page, data)
|
|
|
+
|
|
|
+ def upload_cert_file(self, file):
|
|
|
+ path = 'zzly_xcx_cert/%d/' % self.id
|
|
|
+ upload_path = CertPath(path)
|
|
|
+ filename = '%s%s' % (upload_path.path, file.name)
|
|
|
+ full_filename = "%s/%s" % (settings.MEDIA_ROOT, filename)
|
|
|
+ with open(full_filename, 'wb+') as destination:
|
|
|
+ for chunk in file.chunks():
|
|
|
+ destination.write(chunk)
|
|
|
+ if file.name == 'apiclient_cert.pem':
|
|
|
+ self.apiclient_cert = full_filename
|
|
|
+ elif file.name == 'apiclient_key.pem':
|
|
|
+ self.apiclient_key = full_filename
|
|
|
+ self.save()
|