添加微信支付

Author: cloverstd

.gitignore

@@ -55,3 +55,7 @@ docs/_build/
 
 # PyBuilder
 target/
+
+# wechat pay
+apiclient_cert.pem
+apiclient_key.pem

README.md

@@ -19,6 +19,11 @@ WECHAT_SESSTION_MEMCACHED | None | eg. 127.0.0.1:11211
 WECHAT_AES_KEY | None | 微信消息加密的 key，如果是企业号则必填
 WECHAT_TOKEN | None | 微信接收消息时的 token
 WECHAT_OAUTH_URI | None | oauth 时的回调地址，默认为当前 url
+WECHAT_PAY_API_KEY | None | 微信支付 api key
+WECHAT_PAY_MCH_CERT | None | 微信支付 商户证书路径 eg. apiclient_cert.pem
+WECHAT_PAY_MCH_KEY | None | 微信支付 商户密钥路径 eg. apiclient_key.pem
+WECHAT_PAY_MCH_ID | None | 微信支付 商户号
+WECHAT_PAY_SUB_MCH_ID | None | 微信支付 子商户号，非必填
 
 #Usage
 

flask_wechatpy/__init__.py

@@ -4,6 +4,7 @@
 import functools
 from flask import request, current_app, abort, redirect
 from wechatpy.replies import BaseReply
+from wechatpy.pay import WeChatPay as ori_WeChatPay
 from wechatpy.utils import check_signature
 from wechatpy.exceptions import (
     InvalidSignatureException,
@@ -269,3 +270,43 @@ def wrapper(*args, **kwargs):
             return method(*args, **kwargs)
         return wrapper
     return decorater
+
+
+class WechatPay(object):
+
+    def __init__(self, app=None):
+
+        self._wechat_client = None
+
+        if app is not None:
+            self.init_app(app)
+
+    def init_app(self, app):
+        config = app.config
+        config.setdefault('WECHAT_APPID', None)
+        config.setdefault('WECHAT_PAY_API_KEY', None)
+        config.setdefault('WECHAT_PAY_MCH_CERT', None)
+        config.setdefault('WECHAT_PAY_MCH_KEY', None)
+        config.setdefault('WECHAT_PAY_MCH_ID', None)
+        config.setdefault('WECHAT_PAY_SUB_MCH_ID', None)
+
+        assert config['WECHAT_APPID'] is not None
+        assert config['WECHAT_PAY_API_KEY'] is not None
+        assert config['WECHAT_PAY_MCH_CERT'] is not None
+        assert config['WECHAT_PAY_MCH_KEY'] is not None
+        assert config['WECHAT_PAY_MCH_ID'] is not None
+
+        self._wechat_pay = ori_WeChatPay(
+            appid=config['WECHAT_APPID'],
+            api_key=config['WECHAT_PAY_API_KEY'],
+            mch_id=config['WECHAT_PAY_MCH_ID'],
+            sub_mch_id=config.get('WECHAT_PAY_SUB_MCH_ID', None),
+            mch_cert=config['WECHAT_PAY_MCH_CERT'],
+            mch_key=config['WECHAT_PAY_MCH_KEY'],
+        )
+        if not hasattr(app, 'extensions'):
+            app.extensions = {}
+        app.extensions['wechat_pay'] = self
+
+    def __getattr__(self, name):
+        return getattr(self._wechat_pay, name)