manyi/apps/web/agent/models.py

# -*- coding: utf-8 -*-
# !/usr/bin/env python
"""
    web.agent.models
    ~~~~~~~~~
"""

import simplejson as json
import datetime
import itertools
import logging
import urllib

from bson.objectid import ObjectId
from django.conf import settings
from django.utils.module_loading import import_string
from mongoengine import MapField, LazyReferenceField, IntField
from mongoengine.errors import DoesNotExist
from mongoengine.fields import (StringField, DictField, BooleanField, DateTimeField, EmbeddedDocumentField,
                                ListField, ReferenceField)

from typing import Any, Dict, TYPE_CHECKING, Optional, cast
from collections import namedtuple

from apilib.monetary import RMB, sum_rmb, Permillage, Percent
from apps.web.agent.define import AgentConst, AGENT_INCOME_SOURCE, AGENT_INCOME_TYPE
from apps.web.agent.errors import PrimaryAgentDoesNotExist
from apps.web.common.models import WithdrawRecord, CapitalUser, Balance

from apps.web.constant import Const, AppPlatformType, DEALER_CONSUMPTION_AGG_KIND, MoniAppStatus
from apps.web.core.db import Searchable, MonetaryField, StrictDictField, PermillageField, PercentField
from apps.web.core.exceptions import InvalidParameter, NoAgentFound, NoManagerFound, MerchantError
from apps.web.core.messages.sms import agentWithdrawSMSProvider
from apps.web.core.models import WechatManagerApp, WechatAuthApp, BoundOpenInfo, AliApp, WechatPayApp, \
    WechatMiniApp, WechatUserManagerApp, JDAggrePayApp, JDAuthApp, BankCard, \
    WithdrawEntity, WechatUserSubscribeManagerApp, WechatDealerSubscribeManagerApp
from apps.web.core import PayAppType, APP_KEY_DELIMITER, ROLE
from apps.web.core.payment import WithdrawGateway
from apps.web.device.models import DeviceType
from apps.web.management.models import Manager
from apps.web.common.transaction import WITHDRAW_PAY_TYPE
from apps.web.exceptions import UserServerException

logger = logging.getLogger(__name__)

if TYPE_CHECKING:
    from pymongo.results import UpdateResult
    from apps.web.common.transaction import WithdrawHandler
    from apps.web.core import PayAppBase

AgentDisclaimer = namedtuple("AgentDisclaimer", ["content", "version"])


class Agent(CapitalUser):
    """
    代理商, 管理经销商
    """

    INCOME_SOURCE_LIST = AGENT_INCOME_SOURCE.choices()
    INCOME_SOURCE_TO_TYPE = AgentConst.MAP_SOURCE_TO_TYPE

    INCOME_TYPE_LIST = AGENT_INCOME_TYPE.choices()
    INCOME_TYPE_TO_FIELD = AgentConst.MAP_TYPE_TO_FIELD

    #: 默认的收入频道的分布情况
    DEFAULT_INCOME_MAP = {
        AGENT_INCOME_SOURCE.AD: 0,
        AGENT_INCOME_SOURCE.DEALER_WITHDRAW_FEE: 0,
        AGENT_INCOME_SOURCE.DEALER_CARD_FEE: 0,
        AGENT_INCOME_SOURCE.DEALER_DEVICE_FEE: 0
    }

    #: 默认管理平台微信公众号授权用户信息
    DEFAULT_WECHAT_AUTH_USER_INFO = {
        'avatar': '',
        'sex': 0
    }

    DEFAULT_PAY_TYPE = {
        ROLE.agent: {
            AppPlatformType.ALIPAY: PayAppType.ALIPAY,
            AppPlatformType.WECHAT: PayAppType.WECHAT,
            AppPlatformType.JD: PayAppType.JD_AGGR
        },
        ROLE.dealer: {
            AppPlatformType.ALIPAY: PayAppType.ALIPAY,
            AppPlatformType.WECHAT: PayAppType.WECHAT,
            AppPlatformType.JD: PayAppType.JD_AGGR
        },
        ROLE.myuser: {
            AppPlatformType.ALIPAY: PayAppType.ALIPAY,
            AppPlatformType.WECHAT: PayAppType.WECHAT,
            AppPlatformType.JD: PayAppType.JD_AGGR,
            AppPlatformType.WECHAT_MINI: PayAppType.WECHAT_MINI
        }
    }

    DEFAULT_CHECKPOINT = {
        'gerenzhongxin': False, 'yue': False, 'baogaolaoban': False, 'fukuan': False
    }

    # CAPITAL_PAY_APP_LIST = [WechatPayApp, JDAggrePayApp, JDOpenPayApp]

    #: 收入相关
    deviceBalance = MapField(field = EmbeddedDocumentField(Balance))
    trafficBalance = MapField(field = EmbeddedDocumentField(Balance))
    adBalance = MapField(field = EmbeddedDocumentField(Balance))
    withdrawBalance = MapField(field = EmbeddedDocumentField(Balance))
    insuranceBalance = MapField(field = EmbeddedDocumentField(Balance))
    apiQuotaBalance = MapField(field = EmbeddedDocumentField(Balance))
    disableAdBalance = MapField(field = EmbeddedDocumentField(Balance))

    incomeMap = DictField(verbose_name = '收入字典', default = DEFAULT_INCOME_MAP)
    aggregatedIncome = DictField(verbose_name = '累计收入，只增不减', default = DEFAULT_INCOME_MAP)

    #: 代理自定义自己的产品名称和logo
    productLogo = StringField(verbose_name = "产品logo", default = "")
    productName = StringField(verbose_name = "产品名称", default = "")

    #: 公众号相关
    gzhServiceQrcodeUrl = StringField(verbose_name = "公众号二维码", default = "", max_length = 256)
    gzhServiceLinkUrl = StringField(verbose_name = "公众号链接二维码", default = "")
    forceFollowGzh = BooleanField(verbose_name = "是否强制关注公众号", default = False)
    forceFollowGzhForDealer = StringField(verbose_name = u'经销商的强制关注开关',
                                          default = 'free')  # never:此代理商下的经销商，永远不准强制关注,free：经销商自由允许关注
    title = StringField(verbose_name = u'公众号强制关注的时候，显示的title', default = u'为了充分保障您的支付权益，首次使用需要您关注服务公众号辅助使用设备。')
    desc = StringField(verbose_name = u'公众号的加粉描述', default = u'关注步骤：手指按在上面二维码上，弹出窗口后，点公众号关注后您再次扫描设备上二维码启动设备。')

    #: 用于客服的联系方式
    serviceName = StringField(verbose_name = "客服名称", default = "", max_length = 32)
    servicePhone = StringField(verbose_name = "客服电话", default = "", max_length = 32)
    serviceQrcodeUrl = StringField(verbose_name = "二维码图片链接", default = "", max_length = 256)

    #: 特性列表
    features = ListField(field = StringField(), verbose_name = u'支持的特性', default = [])

    customizedUserGzhAllowable = BooleanField(verbose_name = '用户自定义公众号', default = False)
    customizedUserSubGzhAllowable = BooleanField(verbose_name = '用户订阅通知自定义公众号', default = False)
    customizedDealerGzhAllowable = BooleanField(verbose_name = '经销商自定义公众号', default = False)
    customizedDealerSubGzhAllowable = BooleanField(verbose_name = '经销商订阅通知自定义公众号', default = False)

    managerId = StringField(verbose_name = "从属于哪个管理员", null = False)

    adShow = BooleanField(verbose_name = "广告收入显示与否的选项", default = False)

    # 厂商参与设备运营分成的商户比例 资金池模式
    managerProfitShare = PercentField(verbose_name='代理商设备运营分成比例', default = Percent('0.00'))
    # 厂商参与设备运营分成的商户比例 暂不支持，但是可以先预留
    managerMerProfitShare = PercentField(verbose_name='代理商设备运营分成比例 商户收款', default = Percent('0.00'))

    # 提现OPENID映射
    payOpenIdMap = MapField(EmbeddedDocumentField(BoundOpenInfo))

    #: 用于API接口
    openAPI = BooleanField(verbose_name = "是否支持openAPI", default = False)
    agentSign = StringField(verbose_name = '唯一签名用于API调用,agent的签名，用于核对对方的签名')
    mySign = StringField(verbose_name = '唯一签名用于API调用，我们的签名，用于请求对方URl，对方鉴权')
    domain = StringField(verbose_name = '域名调用URL')

    wechatLoginAuthApp = EmbeddedDocumentField(verbose_name='授权APP', document_type = WechatAuthApp)

    oldWechatLoginAuthApp = EmbeddedDocumentField(verbose_name = '老的授权APP', document_type = WechatAuthApp, default = None)

    wechatUserManagerialApp = EmbeddedDocumentField(verbose_name = '用户管理APP', document_type = WechatUserManagerApp)
    wechatDealerManagerialApp = EmbeddedDocumentField(verbose_name = '经销商管理APP', document_type = WechatManagerApp)

    wechatUserSubscribeManagerApp = EmbeddedDocumentField(verbose_name = '用户订阅通知APP',
                                                          document_type = WechatUserSubscribeManagerApp)
    wechatDealerSubscribeManagerApp = EmbeddedDocumentField(verbose_name = '经销商订阅通知APP',
                                                            document_type = WechatDealerSubscribeManagerApp)

    jdAuthApp = EmbeddedDocumentField(verbose_name = '管理APP', document_type = JDAuthApp, default = None)

    # 小程序实际上由前台绑定了, 设计需要改变
    # 小程序只能在主AGENT配置, 其他代理商不能配置任何资金池
    customizedWechatMiniAllowable = BooleanField(verbose_name = '自定义MINI', default = False)
    wechatMiniApp = EmbeddedDocumentField(verbose_name = u'微信小程序APP', document_type = WechatMiniApp, default = None)
    payAppWechatMini = LazyReferenceField(document_type = JDAggrePayApp, default = None)

    # payAppMini = GenericLazyReferenceField(choices = CAPITAL_PAY_APP_LIST, verbose_name = u'资金池APP', default = None)

    # 支付APP配置
    customizedAlipayCashflowAllowable = BooleanField(verbose_name = '是否开启自主支付宝收款权限', default = False)
    payAppAli = LazyReferenceField(document_type = AliApp, default = None)

    customizedWechatCashflowAllowable = BooleanField(verbose_name = '是否开启微信自主收款权限', default = False)
    payAppWechat = LazyReferenceField(document_type = WechatPayApp, verbose_name = '微信支付APP(用于提现和支付)', default = None)

    customizedJDAggreAllowable = BooleanField(verbose_name = '是否使用京东聚合支付', default = False)
    payAppJDAggre = LazyReferenceField(document_type = JDAggrePayApp, verbose_name = '京东聚合支付(仅用于支付)', default = None)

    bankcards = ListField(ReferenceField(BankCard), default = [])

    featureToggles = DictField(verbose_name = '特性开关', default = {})

    # 厂商给代理商的流量卡成本价. 提交经销商的时候，经销商的默认价格从这个继承，代理商也可以更改
    annualTrafficCost = MonetaryField(verbose_name = '厂商给代理商的流量卡成本价', default = Const.PLATFORM_DEFAULT_TRAFFIC_COST)

    # 平台给厂商的流量卡成本. 添加厂商的时候设置, 添加代理商的时候从厂商配置继承.
    # 如果有更改，更改之前的代理商不会修改这个费用，如果修改需要直接修改
    trafficCardCost = MonetaryField(verbose_name = '平台给厂商的流量卡成本价', default = Const.PLATFORM_DEFAULT_TRAFFIC_COST)

    # 厂商给代理商的提现费率. 代理商给经销商设置的提现费率必须大于该值. 配置资金池的代理商不受限制
    withdrawFeeRatio = PermillageField(verbose_name = '厂商给代理商的提现费率下限',
                                       default = Const.PLATFORM_DEFAULT_WITHDRAW_FEE_RATIO,
                                       min_value = Const.MIN_DEALER_WITHDRAW_FEE_RATIO,
                                       max_value = Const.MAX_DEALER_WITHDRAW_FEE_RATIO)

    # 平台给厂商的提现费率. 厂商给代理商的提现费率下限必须大于该值. 这个是平台成本价.
    withdrawFeeRatioCost = PermillageField(verbose_name = '平台给厂商的提现费率',
                                           default = Const.PLATFORM_DEFAULT_WITHDRAW_FEE_RATIO,
                                           min_value = Const.MIN_DEALER_WITHDRAW_FEE_RATIO,
                                           max_value = Const.MAX_DEALER_WITHDRAW_FEE_RATIO)

    bannerList = ListField(field = DictField(), verbose_name = "banner列表", default = [])

    #: 大部分情况下，是不允许直接变更支付网关的
    #: 手工给其转账的时候把该开关打开
    isChangingPaymentGateway = BooleanField(default = False)

    payType = DictField(verbose_name = u'各应用支付方式', default = DEFAULT_PAY_TYPE)

    moniAppList = ListField(verbose_name = u'监督公众号的清单', default = [])
    moniAppCheckPointDict = DictField(verbose_name = u'是否需要弹出监督公众号的代理商检查点', default = DEFAULT_CHECKPOINT)

    maxPayLimit = IntField(verbose_name = u'最大充值金额限制', default = 500)

    cardWechatInfo = DictField(verbose_name = u'批量导入实体卡绑定内勤微信账号信息', default = {})

    boundCardName = StringField(verbose_name = u'默认绑定卡主姓名', default = '')
    boundCardPhone = StringField(verbose_name = u'默认绑定卡主手机号', default = '')

    withdrawApps = MapField(EmbeddedDocumentField(document_type = WithdrawEntity), default = None)

    # 寻找 免责声明的 流程是 代理商--->主代理商--->系统代理商
    needDisclaimer = BooleanField(verbose_name = u"代理商是否需要免责声明", default = True)
    disclaimer = StringField(verbose_name = u"代理商设置的用户的免责声明", default = "")
    disclaimer_version = StringField(verbose_name = u"当前免责声明的版本，声明更新的时候须将版本号更新", default = "v1.0.0")

    dealerBankWithdrawFee = BooleanField(verbose_name = u'经销商提现到银行卡的手续费，计算方式', default = False)

    bankWithdrawFee = BooleanField(verbose_name = u"银行卡提现手续开关（资金池代理商开关和这个开关为双开关）", default = True)

    ledgerAppJDAggre = LazyReferenceField(document_type=JDAggrePayApp, verbose_name=u'京东聚合支付(仅用于设备收益分账)', default=None)

    meta = {
        'indexes': [
            {
                'fields': ['username'],
                'unique': True
            }
        ],

        "collection": "Agent",
        "db_alias": "default"
    }

    search_fields = ('username', 'nickname', 'openId', 'remarks')

    def __str__(self):
        return 'Agent<id={} username={} nickname={}>'.format(str(self.id), self.username, self.nickname)

    @property
    def service_phone(self):
        return self.servicePhone

    def my_pay_type(self, role, gateway_type):
        if role in self.payType and gateway_type in self.payType[role]:
            return self.payType[role][gateway_type]
        else:
            return self.DEFAULT_PAY_TYPE[role][gateway_type]

    @property
    def my_wechat_pay_app(self):
        # type: ()->WechatPayApp

        if self.customizedWechatCashflowAllowable:
            if not self.payAppWechat:
                raise Exception(u'第三方支付配置错误(1001)')
            else:
                my_app = self.payAppWechat.fetch()  # type: WechatPayApp
                if not my_app.valid:
                    raise Exception(u'第三方支付配置错误(1002)')
                else:
                    if (str(self.id) == settings.MY_PRIMARY_AGENT_ID) and (not my_app.inhouse):
                        raise Exception(u'第三方支付配置错误(1003)')

                    my_app.occupantId = str(self.id)
                    my_app.occupant = self
                    return my_app

        my_app = WechatPayApp.get_null_app()
        my_app.occupantId = str(self.id)
        my_app.occupant = self
        return my_app

    @property
    def my_ali_pay_app(self):
        # type: ()->AliApp
        if self.customizedAlipayCashflowAllowable:
            if not self.payAppAli:
                raise Exception(u'第三方支付配置错误(1001)')
            else:
                my_app = self.payAppAli.fetch()  # type: AliApp
                if not my_app.valid:
                    raise Exception(u'第三方支付配置错误(1002)')
                else:
                    if (str(self.id) == settings.MY_PRIMARY_AGENT_ID) and (not my_app.inhouse):
                        raise Exception(u'第三方支付配置错误(1003)')

                    my_app.occupantId = str(self.id)
                    my_app.occupant = self
                    return my_app

        my_app = AliApp.get_null_app()
        my_app.occupantId = str(self.id)
        my_app.occupant = self
        return my_app

    @property
    def my_jd_aggre_pay_app(self):
        # type: ()->JDAggrePayApp
        if self.customizedJDAggreAllowable:
            if not self.payAppJDAggre:
                raise Exception(u'第三方支付配置错误(1001)')
            else:
                my_app = self.payAppJDAggre.fetch()  # type: JDAggrePayApp
                if not my_app.valid:
                    raise Exception(u'第三方支付配置错误(1002)')
                else:
                    if (str(self.id) == settings.MY_PRIMARY_AGENT_ID) and (not my_app.inhouse):
                        raise Exception(u'第三方支付配置错误(1003)')

                    my_app.occupantId = str(self.id)
                    my_app.occupant = self
                    return my_app

        my_app = JDAggrePayApp.get_null_app()
        my_app.occupantId = str(self.id)
        my_app.occupant = self
        return my_app

    def to_dict(self, shadow = False):
        rv = super(Agent, self).to_dict()

        rv.update({
            'id': str(self.id),

            'productLogo': self.product_logo,
            'productName': self.product_name,
            'serviceName': self.serviceName,
            'servicePhone': self.servicePhone,
            'serviceQrcodeUrl': self.serviceQrcodeUrl,
            'gzhServiceQrcodeUrl': self.gzhServiceQrcodeUrl,

            'adShow': self.adShow,

            'forceFollowGzh': self.forceFollowGzh,

            'managerId': self.managerId,
            'isPrimary': self.is_primary,
            'featureList': self.feature_list,

            'annualTrafficCost': self.annualTrafficCost,
            'trafficCardCost': self.trafficCardCost,

            'title': self.title,
            'desc': self.desc,

            'smsVendor': self.smsVendor,
            'dealerBankWithdrawFee': self.dealerBankWithdrawFee,
            'bankWithdrawFee': self.bankWithdrawFee
        })

        return rv

    @classmethod
    def filter(cls, manager_id, search_key, page_index, page_size, moniAppId = None, shadow = False):
        query = {'managerId': manager_id} if manager_id is not None else {}

        if moniAppId:
            query.update({'moniAppList': moniAppId})

        agents = cls.objects(**query).search(search_key).order_by('-dateTimeAdded')

        total = agents.count()

        dataList = []
        for agent in agents.paginate(pageIndex = page_index, pageSize = page_size):  # type: Agent

            deviceTypes = DeviceType.objects(agentId = str(agent.id)).all()
            if len(deviceTypes):
                devTypeList = [_.to_dict() for _ in deviceTypes]
            else:
                devTypeList = []

            if Manager.objects(primeAgentId = str(agent.id)).first():
                isPrimary = True
            else:
                isPrimary = False

            zhejiangDict = {}

            try:
                from apps.web.south_intf.zhejiang_fire import ZhejiangNorther
                obj = ZhejiangNorther.objects.get(tokenId = str(agent.id))
                ip, port = obj.northPort.split(":")
                zhejiangDict = {
                    'loginUsername': obj.usernameFromHear,
                    'loginPassword': obj.passwordFromHear,
                    'mqUsername': obj.mqUser,
                    'mqPassword': obj.mqPassword,
                    'code': obj.serviceCodeFromNorth,
                    "northIp": ip,
                    "northPort": port
                }
            except Exception:
                pass

            item = {
                'id': str(agent.id),
                'nickname': agent.nickname,
                'username': '******' if shadow else agent.username,

                'annualTrafficCost': agent.annualTrafficCost,
                'trafficCardCost': agent.trafficCardCost,

                'withdrawFeeRatio': float(agent.withdrawFeeRatio.amount),
                'withdrawFeeRatioCost': float(agent.withdrawFeeRatioCost.amount),
                'managerProfitShare': float(agent.managerProfitShare.amount),

                'dealerTotal': agent.get_dealers().count(),

                'dateTimeAdded': agent.dateTimeAdded,

                'featureList': agent.feature_list,
                'bannerImgList': agent.bannerList,

                'customizedAlipayCashflowAllowable': agent.customizedAlipayCashflowAllowable,
                'customizedWechatCashflowAllowable': agent.customizedWechatCashflowAllowable,
                'customizedDealerGzhAllowable': agent.customizedDealerGzhAllowable,
                'customizedDealerSubGzhAllowable': agent.customizedDealerSubGzhAllowable,
                'customizedUserGzhAllowable': agent.customizedUserGzhAllowable,
                'customizedUserSubGzhAllowable': agent.customizedUserSubGzhAllowable,

                'isPrimary': isPrimary,
                'deviceType': devTypeList,
                'detail': {
                    'agentId': str(agent.id),
                    'managerId': agent.managerId,
                    'features': json.dumps(agent.feature_boolean_map)
                },
                'ZJFirePlatform': zhejiangDict,
                'forceFollowGzh': 'yes' if agent.forceFollowGzh else 'no',
                'forceFollowGzhForDealer': agent.forceFollowGzhForDealer,
                'productName': agent.productName,
                'maxPayLimit': agent.maxPayLimit
            }

            try:
                pay_app_ali = agent.my_ali_pay_app  # type: Optional[AliApp]
            except Exception as e:
                pay_app_ali = None

            if pay_app_ali:  # type
                item.update({'aliPayApp': pay_app_ali.to_dict()})  # type: AliApp
            else:
                item.update({'aliPayApp': AliApp.get_null_app().to_dict()})

            try:
                pay_app_wechat = agent.my_wechat_pay_app
            except Exception as e:
                pay_app_wechat = None

            if pay_app_wechat:
                item.update({'wechatPayApp': pay_app_wechat.to_dict()})
            else:
                item.update({'wechatPayApp': WechatPayApp.get_null_app().to_dict()})

            dealer_manager_app = agent.wechatDealerManagerialApp or WechatManagerApp.get_null_app()  # type: WechatManagerApp
            item.update({'dealer': dealer_manager_app.to_dict()})

            user_manager_app = agent.wechatUserManagerialApp or WechatUserManagerApp.get_null_app()  # type: WechatUserManagerApp
            item.update({'user': user_manager_app.to_dict()})

            user_sub_manager_app = agent.wechatUserSubscribeManagerApp or WechatUserSubscribeManagerApp.get_null_app()  # type: WechatUserSubscribeManagerApp
            item.update({'user_sub': user_sub_manager_app.to_dict()})

            dealer_sub_manager_app = agent.wechatDealerSubscribeManagerApp or WechatDealerSubscribeManagerApp.get_null_app()  # type: WechatDealerSubscribeManagerApp
            item.update({'dealer_sub': dealer_sub_manager_app.to_dict()})

            # 监督号的配置
            moniAppList = []
            for appId in agent.moniAppList:
                try:
                    moniApp = MoniApp.objects.get(appId = appId)
                except Exception, e:
                    continue
                moniAppList.append({'appId': moniApp.appId, 'appName': moniApp.appName})
            item.update({'moniApps': moniAppList})

            pointList = []
            for key, switch in agent.moniAppCheckPointDict.items():
                try:
                    point = MoniAppPoint.objects.get(key = key)
                except Exception, e:
                    continue
                pointList.append({'name': point.name, 'key': key, 'switch': switch})
            item.update({'pointDict': pointList})

            if agent.is_primary:
                item.update({'isPrimary': True})
                dataList.insert(0, item)
            else:
                item.update({'isPrimary': False})
                dataList.append(item)

        return total, dataList

    @property
    def product_logo(self):
        if not self.productLogo:
            return self.primary_agent.productLogo
        else:
            return self.productLogo

    @property
    def product_name(self):
        if not self.productName:
            return self.primary_agent.productName
        else:
            return self.productName

    @staticmethod
    def get_agent(agentId):
        try:
            agent = Agent.objects.get(id = ObjectId(agentId))  # type: Agent
            return agent.to_dict()
        except DoesNotExist:
            logger.exception('can not find agent from db,id=%s' % agentId)
            return None

    def update(self, **kwargs):
        # type: (**Any)->int
        """
        每次更新的时候确保正在更改支付网关的选项设置为关闭
        :param kwargs:
        :return:
        """

        return super(Agent, self).update(**kwargs)

    @staticmethod
    def update_agent(agentId, valueDict):
        """TODO REFACTOR"""
        if not valueDict:
            return True

        try:
            agent = Agent.objects(id = agentId).get()
            updated = agent.update(**valueDict)
            assert updated, u'更新失败'
        except Exception as e:
            logger.exception('update agent error=%s' % e)
            return False

        return True

    @staticmethod
    def record_cookie(agentId, response):
        agent = Agent.get_agent(agentId)
        if agent is None:
            return response

        response.set_cookie(key = 'agentLogoUrl',
                            value = agent['productLogo'],
                            max_age = 24 * 3600 * 30,
                            domain = settings.COOKIE_DOMAIN)

        pn = urllib.quote(agent['productName'].encode('utf-8'))
        response.set_cookie(key = 'agentBrandName', value = pn, max_age = 24 * 3600 * 30,
                            domain = settings.COOKIE_DOMAIN)
        response.set_cookie(key = 'agentId', value = agentId, max_age = 24 * 3600 * 30, domain = settings.COOKIE_DOMAIN)
        return response

    def put_cookie(self, response):
        logo = self.product_logo

        if not logo:
            logo = '/app/img/logo.png'

        response.set_cookie(key = 'agentLogoUrl', value = logo,
                            max_age = 24 * 3600 * 30,
                            domain = settings.COOKIE_DOMAIN)

        productName = self.product_name

        response.set_cookie(key = 'agentBrandName',
                            value = urllib.quote(productName.encode('utf-8')),
                            max_age = 24 * 3600 * 30,
                            domain = settings.COOKIE_DOMAIN)
        response.set_cookie(key = 'agentId',
                            value = str(self.id),
                            max_age = 24 * 3600 * 30,
                            domain = settings.COOKIE_DOMAIN)

        return response

    def get_dealers(self):
        """
        获取旗下经销商
        :return:
        """
        Dealer = import_string('apps.web.dealer.models.Dealer')
        return Dealer.objects(agentId = str(self.id))

    @property
    def primary_agent_id(self):
        return self.manager.primeAgentId

    @property
    def manager(self):      # type:() -> Manager
        if not hasattr(self, '__manager__'):
            manager = Manager.objects(id=self.managerId).first()
            if not manager:
                raise NoManagerFound(manager_id=self.managerId)
            setattr(self, '__manager__', manager)

        return getattr(self, '__manager__')

    @property
    def primary_agent(self):    # type:() -> Agent
        primeAgent = Agent.objects(id=self.primary_agent_id).first()
        if not primeAgent:
            raise NoAgentFound(agent_id=self.primary_agent_id)

        setattr(self, '__my_prime_agent__', primeAgent)

        return getattr(self, '__my_prime_agent__')

    @property
    def is_primary(self):
        """
        不需要去多查询一次数据库 只需要找到manager即可
        """
        return self.is_equal(self.primary_agent)

    @property
    def is_in_domain(self):
        try:
            return self.manager.domain == settings.MY_DOMAIN
        except Exception as e:
            logger.error('get manager failure. error = %s; id = %s' % (str(e), str(self.id)))
            return False

    @classmethod
    def from_agent(cls, agent, app_type, **kwargs):
        param_key = '{app_type}_app'.format(app_type = app_type)

        attr_or_func = getattr(agent, param_key)

        if hasattr(attr_or_func, '__call__'):
            return attr_or_func(**kwargs)
        else:
            return attr_or_func

    @classmethod
    def factory(cls, **kwargs):
        factory_type = kwargs.pop('factory_type')
        if factory_type == 'app':
            app_type = kwargs.pop('app_type')
            return lambda agent: cls.from_agent(agent = agent, app_type = app_type, **kwargs)
        elif factory_type == 'withdraw_source_key':
            pay_app = kwargs.pop('pay_app')
            return lambda agent: getattr(agent, 'withdraw_source_key')(pay_app)
        else:
            raise InvalidParameter(u'参数错误')

    @staticmethod
    def get_inhouse_prime_agent():
        # type: ()->Agent
        """
        为了方便识别，如果获取默认代理商失败，需要报错为默认代理商找不到
        """
        try:
            return Agent.objects(id = str(settings.MY_PRIMARY_AGENT_ID)).get()
        except DoesNotExist:
            raise PrimaryAgentDoesNotExist('failed to get default primary agent')

    @property
    def inhouse_prime_agent(self):
        if str(self.id) == str(settings.MY_PRIMARY_AGENT_ID):
            return self
        else:
            return Agent.get_inhouse_prime_agent()

    @property
    def customizedCashflowAllowable(self):
        """
        目前withdrawSourceKey是使用微信商户来标记. 所以资金池场景下仅
        判断微信商户是否支持就可以。在资金池场景下, 必须配置微信支付
        :return:
        """
        return self.customizedWechatCashflowAllowable

    def wechat_mini_env_pay_app(self, role = None, pay_app_type = None):
        # type: (str, str)->WechatMiniApp

        if not self.customizedWechatMiniAllowable:
            primary_agent = self.primary_agent
            if self.is_equal(primary_agent):
                return self.inhouse_prime_agent.wechat_mini_env_pay_app(role, pay_app_type)
            else:
                return primary_agent.wechat_mini_env_pay_app(role, pay_app_type)

        if self.customizedWechatMiniAllowable:
            if not self.customizedCashflowAllowable:
                raise Exception(u'第三方支付配置错误(1001)')

            app = self.payAppWechatMini.fetch()
            if app and app.enable and app.valid:
                app.occupantId = str(self.id)
                app.occupant = self
                return app

            app = self.wechatMiniApp  # type: WechatMiniApp
            if app and app.enable and app.valid:
                app.occupantId = str(self.id)
                app.occupant = self
                return app

            raise Exception(u'第三方支付配置错误(1002)')
        else:
            return WechatMiniApp.get_null_app()

    def wechat_env_pay_app(self, role = None, pay_app_type = None):
        # type: (Optional[None, str], Optional[None, str])->Optional[cast(PayAppBase)]

        assert not (role and pay_app_type), 'role and app_type must not have value in the same time.'

        if not self.customizedCashflowAllowable:
            if pay_app_type:
                _pay_app_type = pay_app_type
                _role = None
            else:
                custom = self.payType.get('custom', False)

                if custom:
                    if role in self.payType and AppPlatformType.WECHAT in self.payType[role]:
                        _pay_app_type = self.payType[role][AppPlatformType.WECHAT]
                        _role = None
                    else:
                        _pay_app_type = None
                        _role = role
                else:
                    _pay_app_type = None
                    _role = role

            primary_agent = self.primary_agent
            if self.is_equal(primary_agent):
                return self.inhouse_prime_agent.wechat_env_pay_app(_role, _pay_app_type)
            else:
                return primary_agent.wechat_env_pay_app(_role, _pay_app_type)

        if role:
            app_pay_type = self.my_pay_type(role, AppPlatformType.WECHAT)
        else:
            app_pay_type = pay_app_type

        if app_pay_type == PayAppType.WECHAT:
            app = self.my_wechat_pay_app  # type: WechatPayApp

            if not app.enable or not app.valid:
                raise Exception(u'系统配置错误(第三方支付)')

            return app

        if app_pay_type == PayAppType.JD_AGGR:
            app = self.my_jd_aggre_pay_app  # type: JDAggrePayApp

            if not app.enable or not app.valid:
                raise Exception(u'系统配置错误(第三方支付)')

            return app

        raise Exception(u'系统配置错误(第三方支付)')

    def alipay_env_pay_app(self, role = None, pay_app_type = None):
        assert not (role and pay_app_type), 'role and app_type must not have value in the same time.'

        if not self.customizedCashflowAllowable:
            # 如果代理商配置了需要的支付类型, 以代理商配置的为准
            if pay_app_type:
                _pay_app_type = pay_app_type
                _role = None
            else:
                custom = self.payType.get('custom', False)

                if custom:
                    if role in self.payType and AppPlatformType.ALIPAY in self.payType[role]:
                        _pay_app_type = self.payType[role][AppPlatformType.ALIPAY]
                        _role = None
                    else:
                        _pay_app_type = None
                        _role = role
                else:
                    _pay_app_type = None
                    _role = role

            primary_agent = self.primary_agent
            if self.is_equal(primary_agent):
                return self.inhouse_prime_agent.alipay_env_pay_app(_role, _pay_app_type)
            else:
                return primary_agent.alipay_env_pay_app(_role, _pay_app_type)

        if role:
            app_pay_type = self.my_pay_type(role, AppPlatformType.ALIPAY)
        else:
            app_pay_type = pay_app_type

        if app_pay_type == PayAppType.ALIPAY:
            app = self.my_ali_pay_app  # type: AliApp
            if not app.enable or not app.valid:
                raise Exception(u'系统配置错误(第三方支付)')
            return app

        if app_pay_type == PayAppType.JD_AGGR:
            app = self.my_jd_aggre_pay_app  # type: JDAggrePayApp
            if not app.enable or not app.valid:
                raise Exception(u'系统配置错误(第三方支付)')
            return app

        raise Exception(u'系统配置错误(第三方支付)')

    def jd_env_pay_app(self, role = None, pay_app_type = None):
        assert not (role and pay_app_type), 'role and app_type must not have value in the same time.'

        if not self.customizedCashflowAllowable:
            if pay_app_type:
                _pay_app_type = pay_app_type
                _role = None
            else:
                custom = self.payType.get('custom', False)

                if custom:
                    if role in self.payType and AppPlatformType.JD in self.payType[role]:
                        _pay_app_type = self.payType[role][AppPlatformType.JD]
                        _role = None
                    else:
                        _pay_app_type = None
                        _role = role
                else:
                    _pay_app_type = None
                    _role = role

            primary_agent = self.primary_agent
            if self.is_equal(primary_agent):
                return self.inhouse_prime_agent.jd_env_pay_app(_role, _pay_app_type)
            else:
                return primary_agent.jd_env_pay_app(_role, _pay_app_type)

        if role:
            app_pay_type = self.my_pay_type(role, AppPlatformType.JD)
        else:
            app_pay_type = pay_app_type

        if app_pay_type == PayAppType.JD_AGGR:
            app = self.my_jd_aggre_pay_app  # type: JDAggrePayApp
            if not app.enable or not app.valid:
                raise Exception(u'系统配置错误(第三方支付)')
            return app

        raise Exception(u'系统配置错误(第三方支付)')

    def _check_wechat_withdraw(self):

        my_app = self.my_wechat_pay_app
        if not my_app.enable:
            raise Exception(u'系统配置错误(三方支付)')

    def withdraw_source_key(self, pay_app = None):
        # type:(PayAppBase)->str

        if not pay_app:
            my_app = self.wechat_env_pay_app(pay_app_type = PayAppType.WECHAT)
            if (not my_app) or (not my_app.valid) or (not my_app.enable):
                raise Exception(u'配置错误(第三方支付)')
        else:
            agent = pay_app.occupant  # type: Agent

            my_app = agent.my_wechat_pay_app  # type: WechatPayApp
            if (not my_app) or (not my_app.valid) or (not my_app.enable):
                raise Exception(u'配置错误(第三方支付)')

        if hasattr(my_app, '__source_key__'):
            return APP_KEY_DELIMITER.join(
                [WithdrawGateway.LEDGER_PREFIX, my_app.pay_app_type, getattr(my_app, '__source_key__')])
        else:
            raise AttributeError('no __source_key__ attribute')

        # if my_app.inhouse:
        #     return settings.MY_PRIMARY_AGENT_WALLET_KEY # 所有平台inhouse资金池全部使用固定的资金池KEY
        # else:
        #     return APP_KEY_DELIMITER.join(
        #         [WithdrawGateway.LEDGER_PREFIX, my_app.pay_app_type, getattr(my_app, '__source_key__')])

    @classmethod
    def _parse_source_key(cls, source_key):
        # type: (str)->tuple
        tokens = source_key.split(APP_KEY_DELIMITER)
        return True if tokens[0].startswith(WithdrawGateway.LEDGER_PREFIX) else False, tokens[1], tokens[2], tokens[3:]

        # if source_key == settings.MY_PRIMARY_AGENT_WALLET_KEY:
        #     is_ledger, pay_app_type, occupant_id = True, PayAppType.WECHAT, settings.MY_PRIMARY_AGENT_WALLET_KEY
        # else:
        #     tokens = source_key.split(APP_KEY_DELIMITER)
        #     is_ledger, pay_app_type, occupant_id = True if tokens[0].startswith(
        #         WithdrawGateway.LEDGER_PREFIX) else False, tokens[1], tokens[2]

    @property
    def disclaimerVersionFormat(self):
        return "{}_{}"

    @property
    def disclaimerVersion(self):
        verFormat = self.disclaimerVersionFormat
        return verFormat.format(str(self.id), self.disclaimer_version)

    @disclaimerVersion.setter
    def disclaimerVersion(self, value):
        try:
            self.update(disclaimer_version = value)
        except Exception as e:
            logger.error(
                "set disclaimer version error, agent is <{}>, version is <{}>, error is ".format(self, value, e))
            raise e

    @property
    def disclaimerContent(self):
        return self.disclaimer

    @disclaimerContent.setter
    def disclaimerContent(self, value):
        try:
            self.update(disclaimer = value)
        except Exception as e:
            logger.error("set disclaimer content error, agent is <{}>. error is <{}>".format(self, e))
            raise e

    @property
    def disclaimerAgent(self):
        """
        获取设置了免责声明的代理商
        :return:
        """
        if self.disclaimer:
            return self

        primeAgent = self.primary_agent
        if primeAgent.disclaimer:
            return primeAgent

        inhousePrimeAgent = self.inhouse_prime_agent

        return inhousePrimeAgent

    @classmethod
    def get_disclaimer(cls, agentId):
        """
        获取 代理商的 免责声明
        :param agentId:
        :return:
        """
        agent = cls.objects.get(id = agentId)
        if not agent.needDisclaimer:
            return AgentDisclaimer(content = "", version = agent.disclaimerVersion)

        agent = agent.disclaimerAgent
        content = agent.disclaimerContent
        disclaimer = AgentDisclaimer(content = content, version = agent.disclaimerVersion)
        return disclaimer

    @classmethod
    def withdraw_gateway_list(cls, source_key):
        is_ledger, pay_app_type, occupant_id, tokens = cls._parse_source_key(source_key)
        if not is_ledger:
            return {
                'alipay': WithdrawGateway(AliApp.get_null_app(), False),
                'wechat': WithdrawGateway(WechatPayApp.get_null_app(), False)
            }

        if pay_app_type != PayAppType.WECHAT:
            raise UserServerException(u'系统配置错误(第三方支付)，请联系平台客服(1001)')

        agent = cls.objects(id = occupant_id).first()
        if not agent:
            raise UserServerException(u'系统配置错误(第三方支付)，请联系平台客服(1002)')

        if not agent.withdrawApps:
            return {
                'alipay': None,
                'wechat': agent.my_wechat_pay_app.new_withdraw_gateway(is_ledger = is_ledger)}
        else:
            if source_key not in agent.withdrawApps:
                raise UserServerException(u'系统配置错误(第三方支付)，请联系平台客服(1003)')

            withdraw_entity = agent.withdrawApps[source_key]  # type: WithdrawEntity

            withdraw_entity.alipay_app.occupantId = str(agent.id)
            withdraw_entity.alipay_app.occupant = agent

            withdraw_entity.wechat_app.occupantId = str(agent.id)
            withdraw_entity.wechat_app.occupant = agent

            return {
                'alipay': withdraw_entity.alipay_app.new_withdraw_gateway(is_ledger = is_ledger),
                'wechat': withdraw_entity.wechat_app.new_withdraw_gateway(is_ledger = is_ledger)
            }

    @staticmethod
    def get_platform_wechat_manager_app():
        agent = Agent.get_inhouse_prime_agent()
        return agent.platform_wechat_manager_app

    @property
    def platform_wechat_manager_app(self):
        agent = self.inhouse_prime_agent
        app = agent.wechatDealerManagerialApp
        if not app:
            raise Exception(u'公众号配置错误')

        app.occupantId = str(agent.id)
        return app

    @staticmethod
    def get_inhouse_wechat_user_manager_app():
        agent = Agent.get_inhouse_prime_agent()
        return agent.inhouse_wechat_user_manager_app

    @property
    def inhouse_wechat_user_manager_app(self):
        agent = self.inhouse_prime_agent
        app = agent.wechatUserManagerialApp
        if not app:
            raise Exception(u'公众号配置错误')

        app.occupantId = str(agent.id)
        return app

    @property
    def wechat_auth_app(self):
        """
        标识用户只用平台的公众号对应APPID. 部分用户由于前期原因配置了自己的
        该接口仅做兼容
        :return:
        """
        if self.wechatLoginAuthApp:
            app = self.wechatLoginAuthApp  # type: WechatAuthApp
            app.occupantId = str(self.id)
            return app
        else:
            primary_agent = self.primary_agent
            if self.is_equal(primary_agent):
                return self.inhouse_prime_agent.wechat_auth_app
            else:
                return primary_agent.wechat_auth_app

    @property
    def wechat_user_manager_app(self):
        if self.customizedUserGzhAllowable:
            if self.wechatUserManagerialApp:
                app = self.wechatUserManagerialApp  # type: WechatManagerApp
                app.occupantId = str(self.id)
                app.occupant = self
                return app
            else:
                raise Exception(u'系统配置错误(wechat_user_manager_app)')
        else:
            primary_agent = self.primary_agent
            if self.is_equal(primary_agent):
                inhouse_agent = self.inhouse_prime_agent
                if inhouse_agent.customizedUserGzhAllowable:
                    return inhouse_agent.wechat_user_manager_app
                else:
                    raise Exception(u'系统未配置(wechat_user_manager_app)')
            else:
                return primary_agent.wechat_user_manager_app

    @property
    def wechat_user_messager_app(self):
        if self.customizedUserGzhAllowable or self.customizedUserSubGzhAllowable:
            if self.customizedUserGzhAllowable:
                if self.wechatUserManagerialApp:
                    app = self.wechatUserManagerialApp  # type: WechatUserManagerApp
                    app.occupantId = str(self.id)
                    app.occupant = self
                    return app
                else:
                    raise Exception(u'系统配置错误(wechat_user_manager_app)')

            if self.customizedUserSubGzhAllowable:
                if self.wechatUserSubscribeManagerApp:
                    app = self.wechatUserSubscribeManagerApp  # type: WechatUserSubscribeManagerApp
                    app.occupantId = str(self.id)
                    app.occupant = self
                    return app
                else:
                    raise Exception(u'系统配置错误(wechat_user_subscribe_manager_app)')
        else:
            primary_agent = self.primary_agent
            if self.is_equal(primary_agent):
                inhouse_agent = self.inhouse_prime_agent
                if inhouse_agent.customizedUserGzhAllowable or inhouse_agent.customizedUserSubGzhAllowable:
                    return inhouse_agent.wechat_user_messager_app
                else:
                    raise Exception(u'系统未配置(wechat_user_manager_app|wechat_user_subscribe_manager_app)')
            else:
                return primary_agent.wechat_user_messager_app

    @property
    def wechat_manager_app(self):
        if self.customizedDealerGzhAllowable:
            if self.wechatDealerManagerialApp:
                app = self.wechatDealerManagerialApp  # type: WechatManagerApp
                app.occupantId = str(self.id)
                return app
            else:
                raise Exception(u'系统配置错误(wechat_dealer_manager_app, %s)' % str(self.id))
        else:
            primary_agent = self.primary_agent
            if self.is_equal(primary_agent):
                return self.inhouse_prime_agent.wechat_manager_app
            else:
                return primary_agent.wechat_manager_app

    @property
    def wechat_user_subscribe_manager_app(self):
        if self.customizedUserSubGzhAllowable:
            if self.wechatUserSubscribeManagerApp:
                app = self.wechatUserSubscribeManagerApp  # type: WechatManagerApp
                app.occupantId = str(self.id)
                app.occupant = self
                return app
            else:
                raise Exception(u'系统配置错误(wechat_user_subscribe_manager_app)')
        else:
            primary_agent = self.primary_agent
            if self.is_equal(primary_agent):
                inhouse_agent = self.inhouse_prime_agent
                if inhouse_agent.customizedUserSubGzhAllowable:
                    return inhouse_agent.wechat_user_subscribe_manager_app
                else:
                    raise Exception(u'系统未配置(wechat_user_subscribe_manager_app)')
            else:
                return primary_agent.wechat_user_subscribe_manager_app

    @property
    def wechat_dealer_subscribe_manager_app(self):
        if self.customizedDealerSubGzhAllowable:
            if self.wechatDealerSubscribeManagerApp:
                app = self.wechatDealerSubscribeManagerApp  # type: WechatManagerApp
                app.occupantId = str(self.id)
                app.occupant = self
                return app
            else:
                raise Exception(u'系统配置错误(wechat_user_manager_app)')
        else:
            primary_agent = self.primary_agent
            if self.is_equal(primary_agent):
                return self.inhouse_prime_agent.wechatDealerSubscribeManagerApp
            else:
                return primary_agent.wechatDealerSubscribeManagerApp

    def get_user_sub_template_id_list(self):
        if not self.customizedUserSubGzhAllowable:
            return []

        if not self.wechatUserSubscribeManagerApp.templateIdMap:
            return []

        else:
            try:
                return map(lambda _: _['templateId'], self.wechatUserSubscribeManagerApp.templateIdMap.values())
            except Exception as e:
                logger.info('error  e=<{}>'.format(e))
                return []

    def get_dealer_sub_template_id_list(self):
        if not self.customizedDealerSubGzhAllowable:
            return []

        if not self.wechatUserSubscribeManagerApp.templateIdMap:
            return []

        else:
            try:
                return map(lambda _: _['templateId'], self.wechatDealerSubscribeManagerApp.templateIdMap.values())
            except Exception as e:
                logger.info('error  e=<{}>'.format(e))
                return []

    @property
    def alipay_auth_app(self):
        if not self.customizedAlipayCashflowAllowable:
            primary_agent = self.primary_agent
            if self.is_equal(primary_agent):
                return self.inhouse_prime_agent.alipay_auth_app
            else:
                return primary_agent.alipay_auth_app

        app = self.my_ali_pay_app  # type: AliApp

        logger.debug("app id is:{}".format(str(app.id)))

        if not app.valid:
            raise Exception(u'系统配置错误(第三方支付)')

        app.occupantId = str(self.id)
        return app

    @property
    def jd_auth_app(self):
        """
        标识用户只用平台的公众号对应APPID. 部分用户由于前期原因配置了自己的
        该接口仅做兼容
        :return:
        """
        if self.jdAuthApp:
            app = self.jdAuthApp  # type: JDAuthApp
            app.occupantId = str(self.id)
            return app
        else:
            primary_agent = self.primary_agent
            if self.is_equal(primary_agent):
                return self.inhouse_prime_agent.jdAuthApp
            else:
                return primary_agent.jdAuthApp

    @property
    def wechat_mini_auth_app(self):
        if not self.customizedWechatMiniAllowable:
            primary_agent = self.primary_agent
            if self.is_equal(primary_agent):
                return self.inhouse_prime_agent.wechat_mini_auth_app
            else:
                return primary_agent.wechat_mini_auth_app

        app = self.wechatMiniApp

        if not app.valid:
            raise Exception(u'系统配置错误(第三方支付)')

        app.occupantId = str(self.id)
        return app

    def income_by_date(self, date, specific_source = None):
        """
        :param date:
        :param specific_source:
        :return:
        """
        reports = AgentIncomeReport.objects(agentId = str(self.id), date = date)

        if specific_source:
            reports = reports.filter(source = specific_source)

        return RMB(reports.sum('amount'))

    def today_income(self, specific_source = None):
        today = datetime.datetime.now().strftime(Const.DATE_FMT)
        return self.income_by_date(date = today, specific_source = specific_source)

    def yesterday_income(self, specific_source = None):
        yesterday = (datetime.datetime.now() - datetime.timedelta(days = 1)).strftime(Const.DATE_FMT)
        return self.income_by_date(date = yesterday, specific_source = specific_source)

    def aggregate_income(self, specific_source = None):
        """
        获取聚合收入
        :param specific_source: 特定source
        :return:
        """
        if not specific_source:
            return sum_rmb(self.aggregatedIncome.values())
        else:
            return sum_rmb([v for k, v in self.aggregatedIncome.items() if k == specific_source])

    def _single_month_income(self, year, month, specific_source):
        if not specific_source:
            reports = AgentIncomeReport.objects(agentId = str(self.id), date__startswith = '%d-%02d' % (year, month))
        else:
            reports = AgentIncomeReport.objects(agentId = str(self.id), date__startswith = '%d-%02d' % (year, month),
                                                source = specific_source)
        return RMB(reports.sum('amount'))

    def current_month_income(self, specific_source = None):
        now = datetime.datetime.now()
        return self._single_month_income(now.year, now.month, specific_source)

    def monthly_income(self, specific_source = None):
        """
        获得月报表
        :param specific_source:
        :return:
        """
        now = datetime.datetime.now()
        years = {_ for _ in range(2017, now.year + 1)}
        months = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12]

        return {(year, month): self._single_month_income(year, month, specific_source)
                for year, month in itertools.product(years, months)}

    def is_equal(self, agent):
        return str(self.id) == str(agent.id)

    @property
    def supported_device_types(self):
        result = DeviceType.all(str(self.id))
        if result:
            return result

        primary_agent = self.primary_agent
        if not self.is_equal(primary_agent):
            return primary_agent.supported_device_types
        else:
            return []

    @property
    def feature_boolean_map(self):
        # type: ()->Dict[str, bool]
        features = super(Agent, self).feature_boolean_map
        if self.customizedWechatCashflowAllowable:
            features.update({
                'show_withdraw_management': True
            })

        return features

    @property
    def hide_consume_kinds_dealer(self):
        rv = []

        if 'hiddenUsedTime' in self.features:
            rv.append(DEALER_CONSUMPTION_AGG_KIND.DURATION)

        if 'hidden_coins' in self.features:
            rv.append(DEALER_CONSUMPTION_AGG_KIND.COIN)

        left_features = set(self.features) - {'hiddenUsedTime', 'hidden_coins'}
        for feature in left_features:
            if feature.startswith('hide_') and feature.endswith('_for_dealer'):
                kind = feature.replace('hide_', '').replace('_for_dealer', '')
                if kind in DEALER_CONSUMPTION_AGG_KIND.choices():
                    rv.append(kind)

        return rv

    @property
    def hide_consume_kinds_user(self):
        rv = []

        if 'hiddenUsedTime' in self.features:
            rv.append(DEALER_CONSUMPTION_AGG_KIND.DURATION)

        if 'hidden_coins' in self.features:
            rv.append(DEALER_CONSUMPTION_AGG_KIND.COIN)

        left_features = set(self.features) - {'hiddenUsedTime', 'hidden_coins'}
        for feature in left_features:
            if feature.startswith('hide_') and feature.endswith('_for_user'):
                kind = feature.replace('hide_', '').replace('_for_user', '')
                if kind in DEALER_CONSUMPTION_AGG_KIND.choices():
                    rv.append(kind)

        return rv

    @property
    def withdraw_sms_provider(self):
        return agentWithdrawSMSProvider

    @property
    def withdraw_sms_phone_number(self):
        return str(self.username)

    def incr_income(self, source, source_key, money):
        # type: (str, str, RMB)->bool

        assert isinstance(money, RMB), 'money has to be a RMB instance'
        assert source in AGENT_INCOME_SOURCE.choices(), 'not support this source'
        assert source_key, 'source key must not be none'

        income_type = AgentConst.MAP_SOURCE_TO_TYPE[source]

        query = {'_id': ObjectId(self.id)}

        update = {
            '$inc': {
                'aggregatedIncome.{source}'.format(source=source): money.mongo_amount,
                'incomeMap.{source}'.format(source=source): money.mongo_amount,
                '{filed}.{key}.balance'.format(filed=AgentConst.MAP_TYPE_TO_FIELD[income_type],
                                               key=source_key): money.mongo_amount
            },
        }

        result = self.get_collection().update_one(query, update, upsert=False)  # type: UpdateResult
        return bool(result.modified_count == 1)

    def decr_income(self, source, source_key, money):  # type:(str, str, RMB) -> bool
        """
        扣除代理商的收益
        必须由调用方保证不会调用重入
        """
        return self.incr_income(source, source_key, -money)

    def new_withdraw_record(self, withdraw_gateway, pay_entity, source_key, income_type, amount, pay_type, manual,
                            recurrent):
        # type: (WithdrawGateway, BankCard, str, str, RMB, str, bool, bool) -> WithdrawRecord

        if income_type == AGENT_INCOME_TYPE.DEALER_WITHDRAW_FEE:
            withdraw_fee_ratio = Permillage('0.00')  # type: Permillage
        else:
            withdraw_fee_ratio = Const.PLATFORM_DEFAULT_WITHDRAW_FEE_RATIO  # type: Permillage

        service_fee = amount * withdraw_fee_ratio.as_ratio  # type: RMB

        withdraw_agent = withdraw_gateway.occupant  # type: Agent

        has_bank_fee = withdraw_agent.dealerBankWithdrawFee and \
                       self.bankWithdrawFee and \
                       pay_type == WITHDRAW_PAY_TYPE.BANK
        if has_bank_fee:
            bank_trans_fee = min(RMB('25.00'), max(RMB('0.10'), amount * Permillage('1').as_ratio))
        else:
            bank_trans_fee = RMB('0.00')

        actual_pay = amount - service_fee - bank_trans_fee  # type: RMB

        return WithdrawRecord.create(self,
                                     withdraw_gateway = withdraw_gateway,
                                     pay_entity = pay_entity,
                                     source_key = source_key,
                                     income_type = income_type,
                                     pay_type = pay_type,
                                     fund_map = {
                                         'amount': amount,
                                         'serviceFee': service_fee,
                                         'actualPay': actual_pay,
                                         'bankTransFee': bank_trans_fee,
                                         'withdrawFeeRatio': withdraw_fee_ratio,
                                         'partition': []
                                     },
                                     manual = manual,
                                     recurrent = recurrent)

    def withdraw_bank_card(self, bank_card_no = None):
        # type: (str)->BankCard

        bank_card = None

        if not bank_card_no:
            if len(self.bankcards) > 0:
                bank_card = self.bankcards[0]
        else:
            bank_card = BankCard.objects(cardNo = bank_card_no).first()

        return bank_card

    def new_withdraw_handler(self, record):
        # type: (WithdrawRecord) -> WithdrawHandler
        from apps.web.agent.withdraw import AgentWithdrawHandler
        return AgentWithdrawHandler(self, record)

    def get_bound_pay_openid(self, key):
        # type: (str)->str
        pay_openid_map = self.payOpenIdMap  # type: dict
        bound = pay_openid_map.get(key, BoundOpenInfo(openId = ''))  # type: BoundOpenInfo
        return str(bound.openId)

    def set_bound_pay_openid(self, key, **payload):
        # type: (str, Dict)->None
        self.payOpenIdMap[key] = BoundOpenInfo(**payload)

    def get_online_moni_app(self):
        apps = MoniApp.objects.filter(appId__in = self.moniAppList, status=MoniAppStatus.ADDING).order_by('-priority')
        if apps.count() == 0:
            return {}

        app = apps.first()  # type: MoniApp
        return {
            'appId': app.appId,
            'secret': app.secret,
            'title': app.title,
            'appName': app.appName,
            'desc': app.desc
        }

    @property
    def agentIds(self):
        # 自身 有agentId 配置的有公众号的，返回自身的id， 没有配置公众号的， 找到厂商的祝代理商的id 以及自身的id返回
        if hasattr(self, "wechatUserManagerialApp") and self.wechatUserManagerialApp:
            return [str(self.id)]

        manager = Manager.objects.get(id = self.managerId)
        agent = Agent.objects.get(id = str(manager.primeAgentId))
        return [str(agent.id), str(self.id)]

    def is_my_product_user(self, user):
        return str(self.id) == user.productAgentId

    @property
    def my_avatar(self):
        if self.avatar:
            return self.avatar

        logo = self.product_logo

        if not logo:
            logo = ''

        return logo

    def check_merchant_conditions(self):    # type:()->None
        """ 商户确认开通前的检查 """
        if not self.ledgerAppJDAggre:
            raise MerchantError(u"商户尚未绑定，开通失败")

    def confirm_merchant(self, source):
        """ 商户已经是确认状态所需要处理的事情 """
        if not source.support_jdaggre():
            return

        if not source.owner == self:
            return

        # 首先获取自己的商户
        app = source.create_app()
        self.ledgerAppJDAggre = app

        return self.save()

    def success_merchant(self):
        """ 确认商户开通的动作 """
        return

    @property
    def merchantRegister(self):
        return self

    @property
    def abnormal(self):
        return self.status == self.Status.ABNORMAL

    @property
    def myLedgerMerchantApp(self):
        """
        商户分账APP
        """
        if not self.ledgerAppJDAggre:
            if not self.payAppJDAggre:
                return None
            else:
                return self.payAppJDAggre.fetch()

        return self.ledgerAppJDAggre.fetch()

    def get_merchant_split_id(self):
        """
        获取商户号绑定的email
        :return:
        """
        app = self.myLedgerMerchantApp  # type: PayAppBase
        if not app:
            return None
        else:
            return app.split_id

    @property
    def deviceIncomeShow(self):
        """
        兼容之前数据库字段的方式
        """
        if "deviceIncomeShow" in self.features:
            return True

        return getattr(self, '_deviceIncomeShow', False)

    @deviceIncomeShow.setter
    def deviceIncomeShow(self, value):
        """
        兼容之前的数据库 防止报错
        """
        setattr(self, '_deviceIncomeShow', value)


class AgentIncomeReport(Searchable):
    agentId = StringField(verbose_name = '代理商ID')
    detail = StrictDictField(verbose_name = '详情', default = {})
    source = StringField(verbose_name = '收入来源')
    amount = MonetaryField(verbose_name = '数额', default = RMB(0))
    mchid = StringField(verbose_name = '资金账号')
    date = StringField(verbose_name = '日期', default = lambda: datetime.datetime.now().strftime(Const.DATE_FMT))
    dateTimeAdded = DateTimeField(verbose_name = '生成时间', default = datetime.datetime.now)

    meta = {"collection": "agent_income_reports", "db_alias": "report"}

    def __repr__(self):
        return '<AgentIncomeReport agentId=%s, source=%s, date=%s>' % (self.agentId, self.source, self.date)

    @property
    def title(self):
        if self.source == AGENT_INCOME_SOURCE.DEALER_WITHDRAW_FEE:
            return u'提现收益-经销商({}) 提现金额({})'.format(str(self.detail.get('name', '')), str(self.detail.get('withdrawAmount', '')))

        elif self.source == AGENT_INCOME_SOURCE.DEALER_CARD_FEE:
            return u'流量卡收益-经销商(%s) 充值(%s)' % (self.detail.get('name', ''), self.detail.get('sum_of_price', ''))

        elif self.source == AGENT_INCOME_SOURCE.AD:
            return u'广告收益-广告(%s) 设备(%s) 经销商(%s)' \
                   % (self.detail.get('adId', ''), self.detail.get('logicalCode', ''), self.detail.get('dealer', ''))

        elif self.source == AGENT_INCOME_SOURCE.DEALER_DEVICE_FEE:
            return u'设备收益-经销商(%s) 设备(%s) 地址(%s)' \
                   % (self.detail.get('name', ''), self.detail.get('logicalCode', ''), self.detail.get('groupName', ''))

        elif self.source == AGENT_INCOME_SOURCE.INSURANCE:
            return u'保险收益-经销商({}) 地址({})'.format(self.detail.get('name', ''), self.detail.get('groupName', ''))

    @classmethod
    def get_income_list(cls, **filters):
        if 'endDate' in filters:
            filters['date__lt'] = filters.pop('endDate')

        return cls.objects(**filters).order_by('-createdTime')


class MoniApp(Searchable):
    # 公众号的原始属性 其中appToken用于微信验证服务器 由我们自行设置
    appName = StringField(verbose_name=u"公众号的名称", default="")
    appid = StringField(verbose_name=u"appId", unique=True)
    rawAppId = StringField(verbose_name=u"公众号的微信号", unique=True)
    secret = StringField(verbose_name=u"秘钥", default="")
    appToken = StringField(verbose_name="token", default="")

    appType = StringField(verbose_name="公众号类型", default="wechat")
    agentId = StringField(verbose_name=u"公众号所属的Agent")
    priority = IntField(verbose_name="加粉的数量", default=0)
    status = IntField(verbose_name="当前的状态", default=MoniAppStatus.ADDING, choices=MoniAppStatus.choices())

    desc = StringField(verbose_name='展示描述', default = '')
    title = StringField(verbose_name=u'展示加粉的说明title', default="")

    # TODO 需要将agentId 以及status 设置为联合索引确保唯一性
    meta = {"collection": "moni_app", "db_alias": "default"}

    def __str__(self):
        return "<{}>-<{}>".format(self.appName, self.rawAppId)

    @property
    def appId(self):
        return self.appid

    @property
    def occupantId(self):
        return self.agentId

    @classmethod
    def get_app_by_raw(cls, rawAppId):
        try:
            app = cls.objects.get(rawAppId=rawAppId)
        except DoesNotExist:
            return None
        return app

    @classmethod
    def get_app_by_agent(cls, agentId):
        """
        获取代理商设置的 moniApp
        找到加粉符合条件的最少的一个
        """
        agentId = str(agentId)
        return cls.objects.filter(agentId=agentId, status=MoniAppStatus.ADDING).first()

    @classmethod
    def get_inhouse_app(cls):
        return cls.get_app_by_agent(agentId=settings.MY_PRIMARY_AGENT_ID)

    def to_dict(self):
        return {
            'appid': self.appid,
            'secret': self.secret,
            'title': self.title,
            'appName': self.appName,
            'desc': self.desc
        }

    @classmethod
    def subscribe(cls, app):
        cls.objects.filter(appid=str(app.appid)).update(inc__priority=1)

    @classmethod
    def unSubscribe(cls, app):
        cls.objects.filter(appid=str(app.appid)).update(dec__priority=1)


class MoniAppPoint(Searchable):
    key = StringField(verbose_name = 'key', default = '')
    name = StringField(verbose_name = 'name', default = '')
    desc = StringField(verbose_name = 'desc', default = '')

    meta = {"collection": "moni_app_point", "db_alias": "default"}