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

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


class JdOpenSettleAccount(BaseJdOpenAPI):

    def create_account(
            self, customerNum, bankAccountName, bankAccountNum, province, city, bankName, bankBranchName,
            settleAmount, payBankList, accountType, phone,
            privateType=None, settlerCertificateCode=None, settlerCertificateStartDate=None, settlerCertificateEndDate=None
    ):
        """
        创建结算账户
        """
        url = "/v2/agent/declare/settleinfo/create"
        data = {
            "customerNum": customerNum,
            "bankAccountName": bankAccountName,
            "bankAccountNum": bankAccountNum,
            "province": province,
            "city": city,
            "bankName": bankName,
            "bankBranchName": bankBranchName,
            "settleAmount": settleAmount,
            "accountType": accountType,
            "phone": phone,
            "privateType": privateType,
            "settlerCertificateCode": settlerCertificateCode,
            "settlerCertificateStartDate": settlerCertificateStartDate,
            "settlerCertificateEndDate": settlerCertificateEndDate
        }
        sendData = {_k: str(_v) for _k, _v in data.items() if _v}
        sendData["payBankList"] = payBankList
        return self._post(url, data=sendData)

    def modify_account(
            self, settleNum, customerNum, bankAccountName, bankAccountNum, province, city, bankName, bankBranchName,
            settleAmount, payBankList, accountType, phone,
            privateType=None, settlerCertificateCode=None, settlerCertificateStartDate=None, settlerCertificateEndDate=None
    ):
        """
        修改结算账户
        """
        url = "/v2/agent/declare/settleinfo/modify"
        data = {
            "settleNum": settleNum,
            "customerNum": customerNum,
            "bankAccountName": bankAccountName,
            "bankAccountNum": bankAccountNum,
            "province": province,
            "city": city,
            "bankName": bankName,
            "bankBranchName": bankBranchName,
            "settleAmount": settleAmount,
            "accountType": accountType,
            "phone": phone,
            "privateType": privateType,
            "settlerCertificateCode": settlerCertificateCode,
            "settlerCertificateStartDate": settlerCertificateStartDate,
            "settlerCertificateEndDate": settlerCertificateEndDate
        }
        sendData = {_k: str(_v) for _k, _v in data.items() if _v}
        sendData["payBankList"] = payBankList
        return self._post(url, data=sendData)

    def get_account(self, settleNum):
        """
        获取结算账户信息
        """
        return self._post("/v1/agent/declare/settleinfo/{}".format(settleNum))