Exemplo n.º 1
0
    def charge(json_data):
        if 'id' in json_data.keys():
            mer_id = json_data['id']
        else:
            raise InvalidMerchantException

        amount = json_data['amount'] if 'amount' in json_data.keys() else 123

        # 根据mer_id获取商户数据
        db = Sqlite()
        merchant = db.get_upmp_data_by_merid(mer_id)

        if not merchant:
            # 更新商户数据库
            repo = RepoGit("../data/")
            repo.pull()

            # 商户数据不存在,则抛出异常
            if not os.path.isfile(os.path.join("data/2014", mer_id + ".txt")):
                raise InvalidMerchantException

            mer_name, mer_key = UpmpHandler.get_merchant_info_by_merid(
                "data/2014/", mer_id)
            db.set_upmp_basic_info(mer_name, mer_id, mer_key)
            merchant = db.get_upmp_data_by_merid(mer_id)

        if not merchant or not merchant['sk']:
            raise InvalidMerchantException

        # 交易请求
        mer_key = merchant['sk']
        uc = UpmpChannel(mer_id, mer_key)
        ret = uc.charge(amount)

        if not ret:
            raise ChargeFailException

        print(ret)
        post_data, res_data, req_dict, res_dict = ret

        res = dict()
        res['upmp'] = dict()
        res['upmp']['tn'] = res_dict['tn']
        res['upmp']['mode'] = '01'

        if not merchant['charge_notify']:
            print('=========charge========')
            db.set_upmp_charge_data(mer_id, post_data, res_data)

        return res
Exemplo n.º 2
0
    def charge(json_data):
        if 'id' in json_data.keys():
            mer_id = json_data['id']
        else:
            raise InvalidMerchantException

        amount = json_data['amount'] if 'amount' in json_data.keys() else 123

        # 根据mer_id获取商户数据
        db = Sqlite()
        merchant = db.get_upmp_data_by_merid(mer_id)

        if not merchant:
            # 更新商户数据库
            repo = RepoGit("../data/")
            repo.pull()

            # 商户数据不存在,则抛出异常
            if not os.path.isfile(os.path.join("data/2014", mer_id + ".txt")):
                raise InvalidMerchantException

            mer_name, mer_key = UpmpHandler.get_merchant_info_by_merid("data/2014/", mer_id)
            db.set_upmp_basic_info(mer_name, mer_id, mer_key)
            merchant = db.get_upmp_data_by_merid(mer_id)

        if not merchant or not merchant['sk']:
            raise InvalidMerchantException

        # 交易请求
        mer_key = merchant['sk']
        uc = UpmpChannel(mer_id, mer_key)
        ret = uc.charge(amount)

        if not ret:
            raise ChargeFailException

        print(ret)
        post_data, res_data, req_dict, res_dict = ret

        res = dict()
        res['upmp'] = dict()
        res['upmp']['tn'] = res_dict['tn']
        res['upmp']['mode'] = '01'

        if not merchant['charge_notify']:
            print('=========charge========')
            db.set_upmp_charge_data(mer_id, post_data, res_data)

        return res
Exemplo n.º 3
0
    def to_log(notify_data, notify_dict, mer_id, uc):
        print notify_dict
        order_no = notify_dict['orderNumber']
        order_time = notify_dict['orderTime']
        settle_amount = notify_dict['settleAmount']
        order_amount = settle_amount
        trans_type = notify_dict['transType']
        qn = notify_dict['qn']
        # print(order_no, order_time, settle_amount, trans_type, mer_id, qn)

        # 如果商户号不存,则立即返回
        db = Sqlite()
        merchant = db.get_upmp_data_by_merid(mer_id)
        if not merchant:
            return

        if trans_type == UpmpConfig.TRANS_TYPE_TRADE:
            if not merchant['charge_notify']:
                print('=========charge_notify========')
                db.set_upmp_charge_notify_data(mer_id, notify_data)

            if not merchant['charge_query_res']:
                print('=========charge_query========')
                # charge retrieve
                post_data, res_data = uc.charge_retrieve(order_no, order_time)
                db.set_upmp_charge_query_data(mer_id, post_data, res_data)

            if not merchant['void_notify']:
                print('=========void========')
                # void
                post_data, res_data, req_dict, res_dict = uc.void(
                    order_time, order_amount, qn)
                db.set_upmp_void_data(mer_id, post_data, res_data)
            elif not merchant['refund_notify']:
                print('=========refund========')
                # refund
                post_data, res_data, req_dict, res_dict = uc.refund(
                    order_time, qn)
                db.set_upmp_refund_data(mer_id, post_data, res_data)
        elif trans_type == UpmpConfig.TRANS_TYPE_VOID:
            print('=========void_notify========')
            db.set_upmp_void_notify_data(mer_id, notify_data)

            if not merchant['void_query_res']:
                print('=========void_query========')
                # void retrieve
                post_data, res_data = uc.void_retrieve(order_no, order_time)
                db.set_upmp_void_query_data(mer_id, post_data, res_data)
        elif trans_type == UpmpConfig.TRANS_TYPE_REFUND:
            print('=========refund_notify========')
            db.set_upmp_refund_notify_data(mer_id, notify_data)

            if not merchant['refund_query_res']:
                print('=========refund_query========')
                # refund retrieve
                post_data, res_data = uc.refund_retrieve(order_no, order_time)
                db.set_upmp_refund_query_data(mer_id, post_data, res_data)

        if db.is_upmp_data_complete(mer_id):
            print('=========TO EXCEL========')
            from util.excel_handler import ExcelHandler
            eh = ExcelHandler()
            report_file = os.path.join('data/2014', merchant['id'] + '.xlsx')
            merchant = db.get_upmp_data_by_merid(mer_id)
            eh.save('./data/template.xlsx', report_file, merchant)

            print('========SEND MAIL========')
            from util import mail
            attchment_path = os.path.join('data/2014',
                                          merchant['id'] + '.xlsx')
            if os.path.isfile(attchment_path):
                mailto_list = list()
                with open('data/maillist') as f:
                    for line in f:
                        mailto_list.append(line.strip())

                if mail.send_mail(mailto_list, merchant, attchment_path):
                    print "发送成功"
                else:
                    print "发送失败"

            print('=========GIT PUSH========')
            repo = RepoGit(root_path)
            repo.add(os.path.join('2014', '*'))
            repo.commit("Merchant " + merchant['id'] + ' test finished')
            repo.push()

            print('========TEST DONE========')
        else:
            print('========CONTINUE=========')
Exemplo n.º 4
0
    def to_log(notify_data, notify_dict, mer_id, uc):
        print notify_dict
        order_no = notify_dict['orderNumber']
        order_time = notify_dict['orderTime']
        settle_amount = notify_dict['settleAmount']
        order_amount = settle_amount
        trans_type = notify_dict['transType']
        qn = notify_dict['qn']
        # print(order_no, order_time, settle_amount, trans_type, mer_id, qn)

        # 如果商户号不存,则立即返回
        db = Sqlite()
        merchant = db.get_upmp_data_by_merid(mer_id)
        if not merchant:
            return

        if trans_type == UpmpConfig.TRANS_TYPE_TRADE:
            if not merchant['charge_notify']:
                print('=========charge_notify========')
                db.set_upmp_charge_notify_data(mer_id, notify_data)

            if not merchant['charge_query_res']:
                print('=========charge_query========')
                # charge retrieve
                post_data, res_data = uc.charge_retrieve(order_no, order_time)
                db.set_upmp_charge_query_data(mer_id, post_data, res_data)

            if not merchant['void_notify']:
                print('=========void========')
                # void
                post_data, res_data, req_dict, res_dict = uc.void(order_time, order_amount, qn)
                db.set_upmp_void_data(mer_id, post_data, res_data)
            elif not merchant['refund_notify']:
                print('=========refund========')
                # refund
                post_data, res_data, req_dict, res_dict = uc.refund(order_time, qn)
                db.set_upmp_refund_data(mer_id, post_data, res_data)
        elif trans_type == UpmpConfig.TRANS_TYPE_VOID:
            print('=========void_notify========')
            db.set_upmp_void_notify_data(mer_id, notify_data)

            if not merchant['void_query_res']:
                print('=========void_query========')
                # void retrieve
                post_data, res_data = uc.void_retrieve(order_no, order_time)
                db.set_upmp_void_query_data(mer_id, post_data, res_data)
        elif trans_type == UpmpConfig.TRANS_TYPE_REFUND:
            print('=========refund_notify========')
            db.set_upmp_refund_notify_data(mer_id, notify_data)

            if not merchant['refund_query_res']:
                print('=========refund_query========')
                # refund retrieve
                post_data, res_data = uc.refund_retrieve(order_no, order_time)
                db.set_upmp_refund_query_data(mer_id, post_data, res_data)

        if db.is_upmp_data_complete(mer_id):
            print('=========TO EXCEL========')
            from util.excel_handler import ExcelHandler
            eh = ExcelHandler()
            report_file = os.path.join('data/2014', merchant['id'] + '.xlsx')
            merchant = db.get_upmp_data_by_merid(mer_id)
            eh.save('./data/template.xlsx', report_file, merchant)

            print('========SEND MAIL========')
            from util import mail
            attchment_path = os.path.join('data/2014', merchant['id'] + '.xlsx')
            if os.path.isfile(attchment_path):
                mailto_list = list()
                with open('data/maillist') as f:
                    for line in f:
                        mailto_list.append(line.strip())

                if mail.send_mail(mailto_list, merchant, attchment_path):
                    print "发送成功"
                else:
                    print "发送失败"

            print('=========GIT PUSH========')
            repo = RepoGit(root_path)
            repo.add(os.path.join('2014', '*'))
            repo.commit("Merchant " + merchant['id'] + ' test finished')
            repo.push()

            print('========TEST DONE========')
        else:
            print('========CONTINUE=========')