manyi/library/jdopen/client/api/complete.py

# -*- coding: utf-8 -*-
# !/usr/bin/env python

from django.conf import settings

from library.jdopen.client.api.base import BaseJdOpenAPI


class JdOpenComplete(BaseJdOpenAPI):

    def complete_customer(self, customerNum, licenseId=None, licenseStartTime=None, licenseEndTime=None, callbackUrl=None):
        url = "/v2/agent/declare/complete"
        defaultCallBackUrl = "{}/{}".format(settings.SERVER_END_BASE_URL, "merchant/auditNotify")
        data = {
            "customerNum": customerNum,
            "licenseId": licenseId,
            "licenseStartTime": licenseStartTime,
            "licenseEndTime": licenseEndTime,
            "callbackUrl": callbackUrl or defaultCallBackUrl
        }

        sendData = {_k: _v for _k, _v in data.items() if _v}
        return self._post(url, data=sendData)

    def confirm_customer(self, agentNum, customerNum):
        url = "/v1/agent/declare/sign/confirm"
        data = {
            "agentNum": agentNum,
            "customerNum": customerNum
        }

        return self._post(url, data=data)