manyi/apps/web/dealer/models.py

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

"""
    web.dealer.models
    ~~~~~~~~~
"""

import datetime
import logging
import uuid
from operator import itemgetter

from bson.objectid import ObjectId
from django.conf import settings
from django.core.cache import cache
from mongoengine import EmbeddedDocument, StringField, BooleanField, EmbeddedDocumentField, MapField, DynamicDocument, \
    DateTimeField, Document, IntField, ListField, DictField, FloatField, ObjectIdField, EmbeddedDocumentListField, \
    LazyReferenceField, DynamicEmbeddedDocument
from typing import Dict, AnyStr, Any, List, Optional, TYPE_CHECKING

from apilib.monetary import RMB, Percent, sum_rmb, Permillage
from apilib.systypes import IterConstant
from apilib.utils_string import cn
from apilib.utils_url import add_query
from apps.web.agent.models import Agent, MoniAppPoint
from apps.web.common.models import UserSearchable, MonthlyPackageTemp
from apps.web.common.models import WithdrawRecord, CapitalUser, Balance
from apps.web.common.transaction import OrderNoMaker, OrderMainType, DealerPaySubType, WITHDRAW_PAY_TYPE, RefundSubType
from apps.web.constant import TYPE_ADJUST_USER_VIRTUAL_CARD, DEALER_CONSUMPTION_AGG_KIND, \
    DEALER_CONSUMPTION_AGG_KIND_TRANSLATION, Const, APP_TYPE
from apps.web.core import PayAppType, ROLE, APP_KEY_DELIMITER
from apps.web.core.db import Searchable, MonetaryField, PercentField, PermillageField
from apps.web.core.exceptions import UpdateError, ServiceException, InvalidParameter
from apps.web.core.messages.sms import dealerWithdrawSMSProvider, dealerMonitorWithdrawSMSProvider
from apps.web.core.models import BoundOpenInfo
from apps.web.core.payment import WithdrawGateway
from apps.web.dealer.constant import TodoDone, TodoTypeEnum, LinkageSwitchEnum
from apps.web.dealer.define import DEALER_INCOME_SOURCE, DEALER_INCOME_TYPE, DealerConst, \
    DEALER_INCOME_SOURCE_TRANSLATION
from apps.web.device.models import Device, Group, GroupDict, DeviceType
from apps.web.helpers import get_app, get_user_manager_agent
from apps.web.utils import concat_dealer_access_entry_url, concat_front_end_url, concat_server_end_url

if TYPE_CHECKING:
    from apps.web.common.transaction import WithdrawHandler
    from apps.web.core.payment import PaymentGateway
    from apps.web.core import PayAppBase


logger = logging.getLogger(__name__)


class DealerCacheMgr():
    pass


class DealerAddr(EmbeddedDocument):
    id = StringField(verbose_name = 'id', default = str(uuid.uuid4()))
    default = BooleanField(verbose_name = u'是否默认地址', default = False)
    name = StringField(verbose_name = 'name', default = '')
    tel = StringField(verbose_name = 'telno', default = '')
    addr = StringField(verbose_name = 'addr', default = '')


class ApiAppInfo(DynamicDocument):
    people = StringField(verbose_name=u"联系人", default='')
    tel = StringField(verbose_name=u"联系电话", default='')

    callbackUrl = StringField(verbose_name=u"回调地址Url", default='')

    apiDeviceMax = IntField(verbose_name=u'API最大接入数量', default=0)
    apiDevicePerCost = MonetaryField(verbose_name=u'API配额单价', default=RMB(50))

    meta = {
        'collection': 'api_app_info',
        'db_alias': 'default'
    }


class DisableAdPlan(DynamicDocument):
    cycle = IntField(verbose_name=u'单次购买周期(单位:月)', default=12)
    disableAdCost = MonetaryField(verbose_name=u'纯净版单价', default=RMB(20))

    meta = {
        'collection': 'disable_ad_plan',
        'db_alias': 'default'
    }


class ConfigTemplate(EmbeddedDocument):
    rechargeDiscount = ListField(verbose_name=u'优惠充值模板', default=[])
    cardRechargeDiscount = ListField(verbose_name=u'卡优惠充值模板', default=[])


class DealerLinkageSwitch(DynamicEmbeddedDocument):
    """
    经销商相关开关项
    """
    chargeInsurance = BooleanField(default=False, verbose_name=u"充电险开关")
    agentProxyServicePhone = BooleanField(verbose_name=u'代理商是否接管客服', default=False)

    def to_dict(self):
        return {"chargeInsurance": self.chargeInsurance, 'agentProxyServicePhone': self.agentProxyServicePhone}


class Dealer(CapitalUser):
    """
        经销商
    """

    INCOME_SOURCE_LIST = DEALER_INCOME_SOURCE.choices()
    INCOME_SOURCE_TO_TYPE = DealerConst.MAP_SOURCE_TO_TYPE

    INCOME_TYPE_LIST = DEALER_INCOME_TYPE.choices()
    INCOME_TYPE_TO_FIELD = DealerConst.MAP_TYPE_TO_FIELD

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

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

    CACHE_KEY = 'dealer-info-{id}'
    CacheMgr = DealerCacheMgr

    wechat = StringField(verbose_name = "微信", default = "")
    description = StringField(verbose_name = "描述", default = "")
    phone = StringField(verbose_name = "电话", default = "")

    # 相当于 经销商的资金池
    deviceBalance = MapField(EmbeddedDocumentField(document_type=Balance))
    # 经销商获取的分账收益 在消费订单结束的时候进行记录
    ledgerBalance = MapField(EmbeddedDocumentField(document_type=Balance))
    adBalance = MapField(EmbeddedDocumentField(document_type=Balance))

    payOpenIdMap = MapField(EmbeddedDocumentField(BoundOpenInfo))

    isRead = StringField(verbose_name = "是否已经读了消息", default = "N")
    noPassReason = StringField(verbose_name = "未通过原因", default = "")
    cibMerchant = BooleanField(verbose_name = "是否立即生效", default = False)

    # 推送消息
    managerialAppId = StringField(verbose_name = "auth,notify app id", default = "")
    managerialOpenId = StringField(verbose_name = "后台，接受推送消息", default = "")

    wechatAuthUserInfo = DictField(verbose_name = "经销商授权后获取的用户信息", default = DEFAULT_WECHAT_AUTH_USER_INFO)
    managerialWechatBoundTime = DateTimeField(verbose_name = "绑定微信的时间")

    agentId = StringField(verbose_name = "从属于哪个代理商", null = False)

    # 上级代理商设置的经营分成比例策略
    agentProfitShare = PercentField(verbose_name='代理商设备运营分成比例', default = Percent('0.00'))
    agentMerProfitShare = PercentField(verbose_name='代理商设备运营分成比例 商户收款', default = Percent('0.00'))

    #: 默认选项|开关
    #: 选项
    adShow = BooleanField(verbose_name = "广告显示与否的选项", default = True)
    noAdPolicy = StringField(verbose_name = "不打开广告情况下的显示策略(notshow,noshow,banner)", default = 'noshow')
    beforeCharge = BooleanField(verbose_name = "使用前必须充值开关", default = False)
    noRecharge = BooleanField(verbose_name = "不需要充值功能", default = False)

    defaultWashConfig = DictField(verbose_name = "与经销商绑定的washconfig", default = {})

    #: 通知选项
    withdrawlNotify = BooleanField(verbose_name = "提现通知", default = False)
    offlineNotifySwitch = BooleanField(verbose_name = "设备离线通知开关", default = False)
    offlineNotifyTime = StringField(verbose_name = '设备离线通知时间', default = '')
    dailyIncomeReportPushSwitch = BooleanField(verbose_name = '是否开启次日9点推送昨日收益报表', default = False)
    newUserPaymentOrderPushSwitch = BooleanField(verbose_name = '是否开启', default = False)
    devFaultPushDealerSwitch = BooleanField(verbose_name = '设备故障通知开关,是否推送给经销商', default = False)
    devFaultPushUserSwitch = BooleanField(verbose_name = '设备故障通知开关,是否推送给用户', default = False)

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

    # 提现费率. 由代理商设置
    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
    )

    # 流量卡年费
    annualTrafficCost = MonetaryField(default = Const.PLATFORM_DEFAULT_TRAFFIC_COST, verbose_name = '流量卡年费')

    # 该值由平台调整. 可以单独调整某些经销商的成本价格和卡年费
    trafficCardCost = MonetaryField(verbose_name = u'平台提供给经销商的流量卡成本价', default = None)

    limitAdLocation = BooleanField(default = False, verbose_name = u'是否限制广告领取的地域')

    features = ListField(verbose_name = u'支持的特性', default = [])

    defaultDiscountConfig = DictField(verbose_name = u"经销商默认优惠充值套餐", default = Const.DEFAULT_DISCOUNT_RULE)

    defaultCardDiscountConfig = DictField(verbose_name = u"经销商默认卡优惠充值套餐", default = Const.DEFAULT_DISCOUNT_RULE)

    supportedIncomeAggregate = ListField(verbose_name = u'支持的收入聚合', default = DEALER_INCOME_SOURCE.choices())

    supportedConsumptionAggregate = ListField(verbose_name=u'支持的消费聚合', default=DEALER_CONSUMPTION_AGG_KIND.choices())

    supportedConsumptionShow = ListField(verbose_name=u"用户端的消费信息显示", default=[])

    bottomAd = DictField(verbose_name = u"底部banner", default = {})

    pushBrokerUrl = StringField(verbose_name = u'经销商订阅的消息推送URL', default = settings.DEFAULT_DEALER_PUSH_BROKER_URL)

    limitDevNum = IntField(verbose_name = u'限制经销商最大设备数量', default = 9999)

    isPurePartner = BooleanField(verbose_name = u'是否为纯合伙人', default = False)

    userCount = IntField(verbose_name = u'总的用户数量', default = 0)

    moniAppCheckPointDict = DictField(verbose_name = u'是否需要弹出监督公众号的经销商检查点', default = DEFAULT_CHECKPOINT)

    forceFollowGzh = StringField(verbose_name=u'是否强制关注公众号', default='agent')#以代理商为准：'agent',强制关注：'yes',不允许强制关注：'no'，‘永不关注’：‘never’，‘关不关注无所谓’:free

    devCount = IntField(verbose_name = u'设备数量，方便统计查询', default = 0)

    maxPackagePrice = MonetaryField(verbose_name = u"最大套餐价格", default = RMB(100.00))

    expressAddrList = EmbeddedDocumentListField(verbose_name = u'经销商的收货地址', document_type = DealerAddr)

    showServicePhone = BooleanField(verbose_name = u'是否允许使用服务电话', default = False)

    isShowBanner = BooleanField(verbose_name = u'是否显示代理商banner', default = True)

    hasTempPackage = BooleanField(verbose_name = u'是否启用临时套餐', default = False)

    freeModeDisplayedIdle = BooleanField(verbose_name = u'免费地址下端口状态显示为空闲', default = False)

    domain = StringField(verbose_name = u'api推送地址url', default = '')

    monitorPhone = StringField(verbose_name = u"提现审批员的手机号", default = "", max_length = 32)

    currencyCoins = BooleanField(verbose_name = u'金币是否通用(deprecated)', default = True)
    currencyMode = StringField(verbose_name = u'金币通用模式(allYes, allNo, asGroup)', default = None)

    # 有种情况下比如同一个手机号在两个代理商下申请的都有经销商 商户其实只用申请一个就行了
    bindMerSrc = StringField(verbose_name=u"共用的商户经销商ID")

    # 联动开关
    linkageSwitch = EmbeddedDocumentField(verbose_name=u"开关集合", document_type=DealerLinkageSwitch, default=DealerLinkageSwitch)

    apiInfo = LazyReferenceField(verbose_name=u'api应用信息', document_type=ApiAppInfo, default=None)
    disableAdPlan = LazyReferenceField(verbose_name=u'纯净版信息', document_type=DisableAdPlan, default=None)

    # 配置模板集合
    templateSet = EmbeddedDocumentField(verbose_name=u"开关集合", document_type=ConfigTemplate, default=ConfigTemplate)

    meta = {
        'indexes': [
            {
                'fields': ['username', 'agentId'], 'unique': True
            },
        ],
        "collection": "Dealer",
        "db_alias": "default"
    }

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

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

    @property
    def currency_mode(self):
        if not self.currencyMode:
            if self.currencyCoins:
                return 'allYes'
            else:
                return 'allNo'
        else:
            return self.currencyMode

    @property
    def bossId(self):
        return self.id

    @property
    def myBoss(self):
        return self

    # 覆写. 需要对ruleDict参数进行一次转换
    def save(self, force_insert = False, validate = True, clean = True,
             write_concern = None, cascade = None, cascade_kwargs = None,
             _refs = None, save_condition = None, **kwargs):

        if 'defaultDiscountConfig' in kwargs:
            newRuleDict = {}
            for k, v in self.format_default_discount.items():
                newRuleDict[k.replace('.', '-')] = v
            self.defaultDiscountConfig = newRuleDict

        if "defaultCardDiscountConfig" in kwargs:
            newCardRuleDict = {}
            for k, v in self.format_card_discount.items():
                newCardRuleDict[k.replace('.', '-')] = v
            self.defaultCardDiscountConfig = newCardRuleDict

        return super(Dealer, self).save(**kwargs)

    def update(self, **kwargs):
        try:
            ruleDict = kwargs.pop('defaultDiscountConfig', None)
            if ruleDict is not None:
                newRuleDict = {}
                for k, v in ruleDict.items():
                    newRuleDict[k.replace('.', '-')] = v
                kwargs.update({'defaultDiscountConfig': newRuleDict})

            cardRuleDict = kwargs.pop('defaultCardDiscountConfig', None)
            if cardRuleDict is not None:
                newCardRuleDict = {}
                for k, v in cardRuleDict.items():
                    newCardRuleDict[k.replace('.', '-')] = v
                kwargs.update({'defaultCardDiscountConfig': newCardRuleDict})

            updated = super(Dealer, self).update(**kwargs)

            if not updated:
                raise UpdateError('failed to update dealer query kwargs(%s)' % (kwargs,))
            else:
                return updated

        finally:
            self.invalid_cache(str(self.id))

    @property
    def permissionList(self):
        return []

    @property
    def virtualCardOnlyCardSwitch(self):
        """
        虚拟卡仅用于 实体卡的开关
        TODO zjl 这个是临时加的一个开关，目的在于不修改前台 达到 虚拟卡只用于实体卡而不能微信扫码使用的目的
        等虚拟卡这一块儿整改的时候直接将这个去掉
        :return:
        """
        return "virtualCardOnlyCardSwitch" in self.features

    @property
    def switches(self):
        return {
            'adShow': self.adShow,

            'beforeCharge': self.beforeCharge,
            'noRecharge': self.noRecharge,

            'withdrawlNotify': self.withdrawlNotify,
            'offlineNotify': self.offlineNotifySwitch,
            'dailyIncomeReportPushSwitch': self.dailyIncomeReportPushSwitch,
            'newUserPaymentOrderPushSwitch': self.newUserPaymentOrderPushSwitch,
            'devFaultPushDealerSwitch': self.devFaultPushDealerSwitch,
            'devFaultPushUserSwitch': self.devFaultPushUserSwitch,

            'virtualCardOnlyCardSwitch': self.virtualCardOnlyCardSwitch
        }

    def query_feature_by_list(self, queryList):
        # type:(list)->dict

        dealerFeatures = self.feature_boolean_map

        resultFeatures = {}

        agent = Agent.objects.get(id=self.agentId)  # type: Agent
        agent_features = agent.feature_boolean_map

        for query in queryList:
            if query in dealerFeatures:
                resultFeatures[query] = dealerFeatures[query]
            else:
                if query in agent_features:
                    resultFeatures[query] = agent_features[query]
                else:
                    resultFeatures[query] = False

        return resultFeatures

    def get_feature(self, feature_name):
        return self.query_feature_by_list([feature_name])

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

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

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

            'deviceBalance': self.sub_balance(DEALER_INCOME_TYPE.DEVICE_INCOME),
            'adBalance': self.sub_balance(DEALER_INCOME_TYPE.AD_INCOME),
            "ledgerBalance": self.sub_balance(DEALER_INCOME_TYPE.LEDGER_CONSUME),
            'balance': self.total_balance,

            'wechat': self.wechat,
            'description': self.description,

            'managerialOpenId': self.managerialOpenId,
            'managerialAppId': self.managerialAppId,

            'agentId': self.agentId,
            'bottomAd': self.bottomAd,

            'featureList': self.feature_list,

            'withdrawFeeRatio': self.withdrawFeeRatio,
            'annualTrafficCost': self.annualTrafficCost,

            'defaultDiscountConfig': self.defaultDiscountConfig,

            'pushBrokerUrl': self.pushBrokerUrl,

            'isPurePartner': self.isPurePartner,
            'pointDict': pointList,
            'forceFollowGzh': self.forceFollowGzh,
            'smsVendor': self.smsVendor,

            'hasTempPackage': self.hasTempPackage,

            'freeModeDisplayedIdle': self.freeModeDisplayedIdle,

            'offlineNotifySwitch': self.offlineNotifySwitch,
            'offlineNotifyTime': self.offlineNotifyTime,

            'currencyMode': self.currency_mode,

            'supportedConsumptionShow': self.supportedConsumptionShow,
        })

        if hasattr(self, 'displayTempPackage'):
            rv.update({"displayTempPackage": self.displayTempPackage})

        rv.update(self.switches)
        return rv

    @classmethod
    def all(cls):
        return [dealer.to_dict() for dealer in cls.objects.all()]

    @staticmethod
    def cache_key(ownerId):
        # type:(Optional[ObjectId, str])->str
        return Dealer.CACHE_KEY.format(id = str(ownerId))

    @staticmethod
    def get_dealer(ownerId):    # type:(ObjectId)->Optional[DealerDict, None]
        if not ownerId:
            return None

        dealer = cache.get(Dealer.cache_key(ownerId))  # type: dict
        if dealer:
            return DealerDict(dealer)
        else:
            dealer = Dealer.objects(id = ownerId).first()  # type: Dealer
            if dealer:
                value = dealer.to_dict()
                cache.set(Dealer.cache_key(ownerId), value)
                return DealerDict(value)
            else:
                logger.error('can not find dealer from db,id=%s' % ownerId)
                return None

    @staticmethod
    def invalid_cache(ownerId):
        # type:(Optional[ObjectId, str])->None
        cache.delete(Dealer.cache_key(ownerId))

    @staticmethod
    def update_dealer(ownerId, **valueDict):
        # type:(ObjectId, **Any)->bool
        try:
            if 'noRecharge' in valueDict and valueDict['noRecharge']:
                valueDict.update({'beforeCharge': False})
            elif 'beforeCharge' in valueDict and valueDict['beforeCharge']:
                valueDict.update({'noRecharge': False})

            Dealer.objects(id = ObjectId(ownerId)).update(**valueDict)
        except Exception as e:
            logger.exception('update dealer info error=%s' % e)
            return False

        Dealer.invalid_cache(ownerId)

        return True

    @property
    def isManagerialOpenIdBound(self):
        return self.managerialOpenId != ''

    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 bound.openId

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

    def get_bound_pay_info(self, key):
        pay_openid_map = self.payOpenIdMap
        bound = pay_openid_map.get(key, BoundOpenInfo(openId = ''))
        return bound.to_detail_dict()

    def set_bound_pay_info(self, key, **payload):
        bound = BoundOpenInfo(**payload)
        self.payOpenIdMap[key] = bound
        return

    def isAutoWithdrawOpenIdBound(self, key):
        pay_openid_map = self.payOpenIdMap  # type: dict
        bound = pay_openid_map.get(key, BoundOpenInfo(openId = ''))
        return bound.openId != "" and bound.nickname != ""

    @property
    def format_default_discount(self):
        tmp_rule_dict = {}
        for k, v in self.defaultDiscountConfig.items():
            tmp_rule_dict[k.replace('-', '.')] = v
        return tmp_rule_dict

    @property
    def format_card_discount(self):
        card_rule_dict = dict()
        for k, v in self.defaultCardDiscountConfig.items():
            card_rule_dict[k.replace("-", ".")] = v
        return card_rule_dict

    @staticmethod
    def get_dealers(agentId):
        dealers = Dealer.get_collection().find({'agentId': agentId})
        return [str(dealer['_id']) for dealer in dealers]

    @staticmethod
    def get_dealer_from_groupId(groupId):
        group = Group.get_group(groupId)
        if group is not None:
            dealer = Dealer.objects(id = group['ownerId']).first()
        else:
            group = Group.objects(id = groupId).first()
            dealer = Dealer.objects(id = group.ownerId).first()
        return dealer

    def get_own_devices(self):
        groupIds = Group.get_group_ids_of_dealer(str(self.id))
        devNoList = Device.get_devNos_by_group(groupIds)
        return Device.get_dev_by_nos(devNoList).values()

    @property
    def income_aggregate_source(self):
        # type: ()->Dict[str, AnyStr]
        """
        e.g. {'ad': u'广告'}
        :return:
        """
        supports = [item for item in self.supportedIncomeAggregate if item != 'ad']
        g = itemgetter(*supports)
        return dict(zip(supports, g(DEALER_INCOME_SOURCE_TRANSLATION)))

    @property
    def consumption_aggregate_source(self):
        # type: ()->Dict[str, AnyStr]
        """
        e.g. {'elect': u'电量'}
        :return:
        """
        g = itemgetter(*self.supportedConsumptionAggregate)
        return dict(zip(self.supportedConsumptionAggregate, g(DEALER_CONSUMPTION_AGG_KIND_TRANSLATION)))

    def set_agent_profit_share(self, share):
        # type: (Percent)->int
        """
        :param share:
        :return:
        """
        updated = self.update(agentProfitShare = share.mongo_amount)
        return updated

    def set_agent_mer_profit_share(self, share):
        # type: (Percent)->int
        """
        :param share:
        :return:
        """
        updated = self.update(agentMerProfitShare = share.mongo_amount)
        return updated

    @classmethod
    def set_traffic_costs_multiply(cls, dealer_ids, cost):
        # type: (List[ObjectId], RMB)->int
        updated = cls.get_collection().update_many({'_id': {'$in': dealer_ids}},
                                                   {'$set': {'annualTrafficCost': cost.mongo_amount}},
                                                   upsert = False)
        return updated

    @property
    def login_url(self):
        # type:()->str

        return concat_dealer_access_entry_url(agentId = self.agentId)

    @property
    def withdraw_sms_provider(self):
        if self.monitorPhone:
            return dealerMonitorWithdrawSMSProvider
        else:
            return dealerWithdrawSMSProvider

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

    @property
    def auto_withdraw_bound_open_id(self):
        from apps.web.helpers import get_wechat_auth_bridge

        auth_bridge = get_wechat_auth_bridge(source = self, app_type = APP_TYPE.WECHAT_WITHDRAW)
        return self.get_bound_pay_openid(auth_bridge.bound_openid_key)

    @property
    def auto_withdraw_bank_account(self):
        return self.withdrawOptions.get('bankAccount', None)

    def record_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 DEALER_INCOME_SOURCE.choices(), 'not support this source'
        assert source_key, 'source key must not be none'

        income_type = DealerConst.MAP_SOURCE_TO_TYPE[source]

        return self.incr_fund(income_type, source_key, money)

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

    def check_withdraw_min_fee(self, income_type, pay_type, amount):
        if pay_type == WITHDRAW_PAY_TYPE.BANK:
            minimum = RMB(settings.WITHDRAW_MINIMUM)
        else:
            if income_type == DEALER_INCOME_TYPE.AD_INCOME:
                minimum = RMB(settings.AD_WITHDRAW_MINIMUM)
            else:
                minimum = RMB(settings.WITHDRAW_MINIMUM)

        if amount < minimum:
            raise ServiceException(
                {'result': 0, 'description': u"提现实际到账金额不能少于%s元" % (settings.WITHDRAW_MINIMUM,), 'payload': {}})

    def new_withdraw_record(self, withdraw_gateway, pay_entity, source_key, income_type, amount, pay_type, manual, recurrent):
        """
        创建自动提现的单子
        :param withdraw_gateway:  提现网关 WechatPaymentGateway
        :param pay_entity: 银行卡 WithdrawBankCard
        :param income_type: 收益类型  str
        :param amount: 提现金额（非实际到账） RMB
        :param pay_type: 提现的方式 str
        :param manual: 手动提现 Bool
        :param recurrent: 是否自动提现 Bool
        :return: WithdrawRecord
        """

        agent = Agent.objects(id = self.agentId).first()  # type: Agent
        if not agent:
            logger.error('agent is None, agentId={}'.format(self.agentId))
            raise ServiceException(
                {
                    'result': 0,
                    'description': u'系统繁忙，请稍后再试',
                    'payload': {}
                })

        if income_type == DEALER_INCOME_TYPE.AD_INCOME:
            agentFeeRation = Permillage('0')
            managerFeeRatio = Permillage('0')
            dealerFeeRatio = Permillage('0')
        else:
            agentFeeRation = agent.withdrawFeeRatio
            managerFeeRatio = agent.withdrawFeeRatioCost
            dealerFeeRatio = self.withdrawFeeRatio

        # 微信或者支付宝平台服务费
        serviceFee = amount * (dealerFeeRatio.as_ratio)

        # 计算微信转银行卡手续费
        has_bank_fee = False

        if pay_type == WITHDRAW_PAY_TYPE.BANK:
            if recurrent:
                has_bank_fee = self.auto_withdraw_bank_fee
            else:
                withdraw_agent = withdraw_gateway.occupant  # type: Agent
                has_bank_fee = withdraw_agent.dealerBankWithdrawFee and self.bankWithdrawFee

        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')

        actualPay = amount - serviceFee - bank_trans_fee

        self.check_withdraw_min_fee(income_type, pay_type, actualPay)

        if dealerFeeRatio > managerFeeRatio:
            earned = amount * ((dealerFeeRatio - managerFeeRatio).as_ratio)
            if earned < RMB(0):
                earned = RMB(0)
            elif earned > serviceFee:
                earned = serviceFee

            # 分给代理商
            agentEarned = RMB(0)
            if dealerFeeRatio > agentFeeRation:
                agentEarned = amount * ((dealerFeeRatio - agentFeeRation).as_ratio)

            if agentEarned < RMB(0):
                agent_earned = RMB(0)
            elif agentEarned > earned:
                agentEarned = earned

            # 代理商分完如果还有 继续分给厂商
            managerEarned = (earned - agentEarned)
            if managerEarned < RMB(0):
                managerEarned = RMB(0)

        else:
            agentEarned = RMB(0)
            managerEarned = RMB(0)

        partition = [
            {
                'role': ROLE.agent,
                'id': self.agentId,
                'cost': agentFeeRation.mongo_amount,
                'earned': agentEarned.mongo_amount
            },
            {
                'role': ROLE.manager,
                'id': str(agent.primary_agent.id),
                'cost': managerFeeRatio.mongo_amount,
                'earned': managerEarned.mongo_amount
            }
        ]

        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': serviceFee,
                'bankTransFee': bank_trans_fee,
                'actualPay': actualPay,
                'withdrawFeeRatio': dealerFeeRatio,
                'partition': partition
            },
            manual = manual,
            recurrent = recurrent
        )

    def get_addr_list(self):
        result = []
        for obj in self.expressAddrList:
            if not obj.default:
                result.append({'tel':obj.tel,'name':obj.name,'addr':obj.addr,'id':str(obj.id),'default':obj.default})
            else:
                result.insert(0, {'tel':obj.tel,'name':obj.name,'addr':obj.addr,'id':str(obj.id),'default':obj.default})
        return result

    @classmethod
    def get_auto_withdraw_dealers(cls):
        dealers = list()
        for _dealer in cls.objects.filter(withdrawOptions__autoWithdrawSwitch = True):  # type: Dealer
            if "autoWithdraw" in _dealer.features:
                dealers.append(_dealer)

        return dealers

    @classmethod
    def get_income_balance_list(cls, dealer, incomeType, minNum=0):
        """
        计算经销商在 每个收益来源下面的可提现余额 以及 手续费
        :param dealer:
        :param incomeType:
        :param minNum: 过滤金额
        :return:  list --> (sourceKey, amount) 来源 当前账户余额
        """
        assert incomeType in DEALER_INCOME_TYPE.choices(), 'not support this income type'
        balanceDict = getattr(dealer, DealerConst.MAP_TYPE_TO_FIELD[incomeType])

        withdrawInfoList = list()
        for sourceKey, item in balanceDict.items():
            if not WithdrawGateway.is_ledger(sourceKey):
                continue

            balance = item.balance
            if balance <= RMB(0):
                continue

            if balance < RMB(minNum):
                continue

            withdrawInfoList.append((sourceKey, balance))

        return withdrawInfoList

    @classmethod
    def factory(cls, **kwargs):
        factory_type = kwargs.pop('factory_type')
        if factory_type == 'app':
            app_type = kwargs.pop('app_type')
            return lambda dealer: cls.from_dealer(dealer = dealer, app_type = app_type, **kwargs)
        else:
            raise InvalidParameter(u'参数错误')

    @classmethod
    def from_dealer(cls, dealer, app_type, **kwargs):
        appKey = "{}_app".format(app_type)
        attr_or_func = getattr(dealer, appKey)
        if callable(attr_or_func):
            return attr_or_func(**kwargs)
        else:
            return attr_or_func

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

        if self.isManagerialOpenIdBound and self.wechatAuthUserInfo:
            return self.wechatAuthUserInfo.get('avatar', '')

        return ''

    def withdraw_source_key(self, pay_app = None):
        # type:(PayAppBase)->str
        """
        经销商提现资金池KEY是虚拟的, 仅仅对应账号的资金统计值
        :param pay_app:
        :return:
        """

        return APP_KEY_DELIMITER.join(
            [WithdrawGateway.NO_LEDGER_PREFIX, pay_app.pay_app_type, getattr(pay_app, '__gateway_key__')])

    @classmethod
    def get_cooperative_dealer_ids(cls, dealerId):
        rcds = Group.get_collection().find({'partnerList.id': dealerId})

        dealer_ids = set([dealerId])
        for rcd in rcds:  # type: Group
            dealer_ids.add(rcd['ownerId'])

        return list(dealer_ids)

    @classmethod
    def get_cooperative_group_ids(cls, dealerId):
        gs = Group.get_collection().find({'partnerList.id': dealerId})
        groupIds = []
        for _ in gs:
            groupIds.append(str(_['_id']))

        gsMe = Group.get_collection().find({'ownerId': dealerId})
        for _ in gsMe:
            groupIds.append(str(_['_id']))

        return groupIds

    @property
    def is_inhouse_wallet(self):
        app = get_app(source = self, app_type = APP_TYPE.WECHAT_ENV_PAY, role = ROLE.myuser)
        return getattr(app, 'inhouse', False)

    @property
    def defaultMonthlyPackage(self):
        """
        默认的包月套餐的列表
        :return:
        """
        return MonthlyPackageTemp.default_one(ownerId=str(self.id))

    @property
    def monthlyPackage(self):
        """
        所有的包月套餐的列表
        :return:
        """
        return MonthlyPackageTemp.get_template_by_dealer(str(self.id))

    def isJosEnable(self, clientEnv="wechat"):
        """
        判断该经销商是否能够使用京东的拉新活动
        # 由于分账的原因 目前只能够走 使用我们自身资金池的用户
        :param clientEnv: 客户端的环境
        :return:
        """
        return False

    def limit_filter_date(self, startTime, endTime):
        # type:(str, str)->(str, str)

        if not startTime:
            startTime = datetime.datetime.now().strftime('%Y-%m-%d')

        if not endTime:
            endTime = datetime.datetime.now().strftime('%Y-%m-%d')

        if str(self.id) == '5d132407003048494763a51b':
            # 港德的隐藏数据需求
            if startTime < '2021-01-01':
                startTime = '2021-01-01'

            if endTime < '2021-01-01':
                startTime = '2021-01-01'
                endTime = '2020-12-31'
        else:
            if startTime > endTime:
                endTime = startTime

        return startTime, endTime

    @classmethod
    def limit_filter_year(cls, dealerId, year):
        if dealerId == '5d132407003048494763a51b':
            # 港德的隐藏数据需求
            if int(year) < 2021:
                return None
        return year

    def get_currency_group_ids(self, group, groups = None):
        # type: (Optional[Group, GroupDict], List[Optional[Group, GroupDict]])->list

        dealer_dict = DealerDict(self.to_dict())  # type: DealerDict
        return dealer_dict.get_currency_group_ids(group, groups)

    @property
    def ad_show(self):
        if self.adShow:
            return 'show'
        else:
            return self.noAdPolicy

    @property
    def minProfitShare(self):
        """
        经销商 所有的地址中 经销商的最低的分成比例
        :return:
        """
        profitShare = Percent('100')
        for _group in Group.objects.filter(ownerId=str(self.id)):

            if not _group.partnerList:
                continue
            for _partner in _group.partnerList:

                _profitShare = Percent(_partner["percent"])
                if _profitShare < profitShare:
                    profitShare = _profitShare

        return profitShare

    def rent_freeze_balance(self, income_type, money, source_key, transaction_id):      # type:(str, RMB, str, str)->bool
        """
        出租设备的冻结订单
        :param income_type:
        :param money:
        :param source_key:
        :param transaction_id:
        :return:
        """
        assert source_key, 'gateway is null'
        assert transaction_id, 'transaction id is null'

        orderType = "inHandRentOrderList"

        field = self.income_field_name(income_type = income_type)
        fund_key = self.fund_key(income_type = income_type, source_key = source_key)

        query = {'_id': self.id, '{}.transaction_id'.format(orderType): {'$ne': transaction_id}}
        update = {
            '$inc': {
                '{fund_key}.balance'.format(fund_key = fund_key): (-money).mongo_amount
            },
            '$addToSet': {
                '{}'.format(orderType): {
                    'transaction_id': transaction_id,
                    'field': field,
                    'key': source_key,
                    'value': money.mongo_amount
                }
            }
        }
        result = self.get_collection().update_one(query, update, upsert = False)
        return bool(result.modified_count == 1)

    def rent_clear_frozen_balance(self, transaction_id):     # type:(str)->bool
        """
        订单扣款完毕 清除冻结信息
        :param transaction_id:
        :return:
        """
        assert transaction_id, 'transaction id is null'

        orderType = "inHandRentOrderList"
        query = {'_id': self.id, '{}.transaction_id'.format(orderType): transaction_id}
        update = {
            '$pull': {
                '{}'.format(orderType):
                    {
                        'transaction_id': transaction_id
                    }
            }
        }
        result = self.get_collection().update_one(query, update, upsert = False)
        return bool(result.modified_count == 1)

    def rent_recover_frozen_balance(self, income_type, money, source_key, transaction_id):   # type:(str, RMB, str, str)->bool
        """
        回滚冻结的支付订单余额
        :param source_key:
        :param income_type:
        :param money:
        :param transaction_id:
        :return:
        """

        assert source_key, 'gateway is null'
        assert transaction_id, 'transaction id is null'

        orderType = "inHandRentOrderList"

        fund_key = self.fund_key(income_type = income_type, source_key = source_key)

        query = {'_id': self.id, '{}.transaction_id'.format(orderType): transaction_id}

        update = {
            '$inc': {
                '{fund_key}.balance'.format(fund_key = fund_key): money.mongo_amount
            },
            '$pull': {
                '{}'.format(orderType):
                    {
                        'transaction_id': transaction_id
                    }
            }
        }
        result = self.get_collection().update_one(query, update, upsert = False)
        return bool(result.modified_count == 1)

    def get_linkage_switch(self):
        """
        获取经销商的两级联动开关设置
        """
        return {
            LinkageSwitchEnum.CHARGE_INSURANCE: self.linkageSwitch.chargeInsurance
        }

    def turn_on(self, category):
        """
        打开保险的开关
        """
        # 打开保险前的前置检查
        if category == LinkageSwitchEnum.CHARGE_INSURANCE:
            if not self.is_inhouse_wallet:
                return False

            self.linkageSwitch.chargeInsurance = True
            return bool(self.update(linkageSwitch__chargeInsurance=self.linkageSwitch.chargeInsurance))

    def turn_off(self, category):
        """
        关掉保险的开关
        """
        if category == LinkageSwitchEnum.CHARGE_INSURANCE:
            self.linkageSwitch.chargeInsurance = False
            return bool(self.update(linkageSwitch__chargeInsurance=self.linkageSwitch.chargeInsurance))

    def get_all_api_devs(self):
        devList = Device.get_devs_by_ownerId(str(self.id))
        return filter(lambda _: _.isApi == True, devList)

    @property
    def api_quota_info(self):
        totalQuota = self.api_app.apiDeviceMax
        devs = self.get_all_api_devs()
        usedQuota = len(devs)
        remainingQuota = totalQuota - usedQuota if totalQuota - usedQuota > 0 else 0
        return {
            'totalQuota': totalQuota,
            'usedQuota': usedQuota,
            'remainingQuota': remainingQuota
        }

    @property
    def api_app(self):
        if self.apiInfo:
            return self.apiInfo.fetch()
        else:
            return ApiAppInfo()

    def update_api_app(self, **payload):
        if not self.apiInfo:
            self.apiInfo = ApiAppInfo().save()
            self.save()

        apiInfo = self.apiInfo.fetch()  # type: ApiAppInfo
        apiInfo.update(**payload)

    @property
    def disable_ad_plan(self):
        if self.disableAdPlan:  # type: DisableAdPlan
            return self.disableAdPlan.fetch()
        else:
            return DisableAdPlan()

    def update_disable_ad_plan(self, **payload):
        if not self.disableAdPlan:
            self.disableAdPlan = DisableAdPlan().save()
            self.save()

        disable_ad_plan = self.disableAdPlan.fetch()  # type: DisableAdPlan
        disable_ad_plan.update(**payload)

    def disable_ad_quota_info(self):
        all_devs = Device.get_devs_by_ownerId(str(self.id))
        now = datetime.datetime.now()
        configured_filter = lambda _: _.disableADExpireDate and _.disableADExpireDate > now
        expiring_soon_filter = lambda _: _.disableADExpireDate and (_.disableADExpireDate - now).total_seconds() <= 60 * 60 * 24 * 15

        configured = len(filter(configured_filter, all_devs))
        expiringSoon= len(filter(expiring_soon_filter, all_devs))

        notConfigured = len(Device.get_devs_by_ownerId(str(self.id))) - configured

        return {
            'configured': configured,
            'expiringSoon': expiringSoon,
            'notConfigured': notConfigured
        }

    def query_home_page_layout(self):
        menuDict = {}

        menu_features = []
        for menu, default_value in Const.MAIN_MENU_LIST.iteritems():
            if default_value:
                menu_features.append('hide_{}'.format(menu))
            else:
                menu_features.append(menu)

            menuDict[menu] = default_value

        homeDataDict = {}
        home_data_features = ['show_ad_income', 'show_offline_coins']
        for menu, default_value in Const.HOME_PAGE_DATA_LIST.iteritems():
            homeDataDict[menu] = default_value

        queryResult = self.query_feature_by_list(menu_features + home_data_features)

        for k, v in queryResult.items():
            if k in home_data_features:
                if k == 'show_ad_income':
                    homeDataDict['today_ad_income'] = v
                elif k == 'show_offline_coins':
                    homeDataDict['offline_coins'] = v
            else:
                if 'hide_' in k:
                    if v:
                        menu_name = k.replace('hide_', '')
                        menuDict[menu_name] = False
                else:
                    menu_name = k
                    menuDict[menu_name] = v

        return menuDict, homeDataDict

    @property
    def service_phone(self):
        # 如果代理商接管客服电话,就显示代理商的服务电话
        if self.linkageSwitch.agentProxyServicePhone is True:
            agent = Agent.objects(id=self.agentId).first()  # type: Optional[Agent]
            if agent and agent.service_phone:
                return agent.service_phone
        elif self.servicePhone:
            return self.servicePhone
        else:
            return self.username

    @property
    def my_agent(self):
        # type:()->Agent

        if not hasattr(self, '__my_agent__'):
            my_agent = Agent.objects(id=self.agentId).first()
            setattr(self, '__my_agent__', my_agent)

        return getattr(self, '__my_agent__')

    @property
    def force_follow_gzh(self):
        if self.forceFollowGzh == 'agent':
            isNeedFollow = self.my_agent.forceFollowGzh
        else:
            isNeedFollow = True if self.forceFollowGzh == 'yes' else False

        return isNeedFollow

    @property
    def show_auth_window(self):
        showAuth = 0 if self.supports('not_show_auth_window') else 1
        if showAuth == 0:
            return False
        else:
            return True

    @property
    def productAgent(self):
        if not hasattr(self, '__product_agent__'):
            agent = get_user_manager_agent(self)
            setattr(self, '__product_agent__', agent)

        return getattr(self, '__product_agent__')

    def freeze_ledger_balance(self, money, source_key, transaction_id):
        """
        消费订单分账前的冻结
        """
        income_type = DEALER_INCOME_TYPE.DEVICE_INCOME
        self._freeze_balance(income_type, money, source_key, transaction_id, "inhandLedger")

    def clear_ledger_balance(self, transaction_id):
        """
        分账记录收益完成 回滚
        """
        self._clear_frozen_balance(transaction_id, "inhandLedger")


class DealerDict(dict):
    """
    经销商的缓存表示
    """

    def __repr__(self):
        return '<DealerDict id=%s>' % (self.get('groupId', 'unknown'),)

    @property
    def v(self):
        return dict(self)

    @property
    def id(self):
        return self.get('id')

    @property
    def agentId(self):
        return self.get('agentId')

    @property
    def entity(self):
        if hasattr(self, '__entity__'):
            return getattr(self, '__entity__')

        dealer = Dealer.objects(id = self.id).first()
        if not dealer:
            raise Exception('no such dealer<id={}>'.format(self.id))
        else:
            setattr(self, '__entity__', dealer)

        return dealer

    @property
    def currency_mode(self):
        if 'currencyMode' not in self:
            Dealer.invalid_cache(self.id)
            dealer = Dealer.get_dealer(self.id)  # type: DealerDict
            self.update(dealer.v)

        return self.get('currencyMode')

    @property
    def supportedConsumptionShow(self):
        """
        客户端的 消费消息显示开关
        """
        if 'supportedConsumptionShow' not in self:
            Dealer.invalid_cache(self.id)
            dealer = Dealer.get_dealer(self.id)
            self.update(dealer.v)

        return self["supportedConsumptionShow"]

    def is_currency(self, left_group, right_group):
        # type: (GroupDict, GroupDict)->bool

        if left_group.ownerId != right_group.ownerId:
            return False

        if self.currency_mode == 'allNo':
            return False

        if self.currency_mode == 'allYes':
            return True

        if left_group.currencyGroup and right_group.currencyGroup and (
                left_group.currencyGroup == right_group.currencyGroup):
            return True
        else:
            return False

    def get_currency_group_ids(self, group, groups = None):
        # type: (Optional[Group, GroupDict], List[Optional[Group, GroupDict]])->list

        if self.currency_mode == 'allNo':
            return []
        else:
            if not groups:
                groups = Group.get_groups_of_dealer(str(self.id))

            if self.currency_mode == 'allYes':
                rv = []
                for item in groups:  # type: GroupDict
                    if item.groupId == group.groupId:
                        continue

                    rv.append(item.groupId)

                return rv
            else:
                if not group.currencyGroup:
                    return []
                else:
                    rv = []
                    for item in groups:  # type: GroupDict
                        if item.groupId == group.groupId:
                            continue

                        if item.currencyGroup == group.currencyGroup:
                            rv.append(item.groupId)

                    return rv

    @property
    def servicePhone(self):
        return self['servicePhone']

    @property
    def my_agent(self):
        if '__my_agent__' not in self:
            my_agent = Agent.objects(id=self.agentId).first()
            self['__my_agent__'] = my_agent

        return self.get('__my_agent__')

    @property
    def username(self):
        return self['username']

    @property
    def nickname(self):
        return self['nickname']


class UpscoreRecord(Document):
    logicalCode = StringField(verbose_name = "设备逻辑编号", default = '')
    devNo = StringField(verbose_name = "设备编号")
    devType = StringField(verbose_name = "设备类型", default = '')
    ownerId = StringField(verbose_name = "所有者", default = "")
    time = DateTimeField(verbose_name = "时间", default = datetime.datetime.now)
    score = IntField(verbose_name = "上分数量", default = 0)
    groupName = StringField(verbose_name = "上分分组名称", default = "")
    address = StringField(verbose_name = "上分设备地址", default = "")
    remark = StringField(verbose_name = "备注", default = "")
    package = DictField(verbose_name = "套餐", default = {})

    type = StringField(verbose_name = '上分类型(套餐上分,远程充卡),默认为空,表示上分', default = '')

    meta = {"collection": "UpscoreRecord", "db_alias": "logdata",
            # "shard_key":("ownerId",)
            }

    @classmethod
    def get_collection(cls):
        return cls._get_collection()


class AdjustUserVirtualCardRecord(Document):
    cardId = StringField(verbose_name = "虚拟卡ID")
    cardNo = StringField(verbose_name = "虚拟卡卡号")
    adjustDays = IntField(verbose_name = "调整天数", default = 0)
    beforeAdjust = DateTimeField(verbose_name = "调整前的过期时间")
    adjustQuota = ListField(verbose_name = "调整的额度", default = [])
    oldQuota = ListField(verbose_name = "调整之前的额度", default = [])
    operator = StringField(verbose_name = "操作人")
    dealerId = StringField(verbose_name = "卡的管理者")
    adjustType = StringField(verbose_name = "调整方式", choices = TYPE_ADJUST_USER_VIRTUAL_CARD.choices())
    dateTimeAdded = DateTimeField(verbose_name = "添加时间", default = datetime.datetime.now)

    meta = {'collection': 'adjust_user_virtual_card_record', 'db_alias': 'logdata'}

    @classmethod
    def get_collection(cls):
        return cls._get_collection()


class DealerRechargeRecord(Searchable):
    """
    经销商充值记录
    """
    class PayState(IterConstant):
        UnPaid = 'UnPaid'
        Paid = 'Paid'
        Failure = 'Failure'
        Fake = 'Fake'
        Cancel = 'Cancel'
        Quit = 'Quit'
        Close = 'Close'

    class ProductType(IterConstant):
        SimCard = 'WF4801'  # 流量卡充值
        ApiCost = 'WF48A'  # API接入
        DisableAd = 'WF48D'  # 禁用广告接入
        AutoSimCard = 'WF4802'  # 流量卡自动充值
        ManualSimCard = 'WF4803'  # 流量卡线下充值
        SimOrderVerify = 'WF5801' # 流量卡充值对账后重新分账

    ProductDesc = {
        ProductType.SimCard: cn(u'流量卡充值'),
        ProductType.AutoSimCard: cn(u'流量卡自动充值'),
        ProductType.ManualSimCard: cn(u'流量卡线下充值'),
        ProductType.ApiCost: cn(u'API配额充值'),
        ProductType.DisableAd: cn(u'纯净计划'),
        ProductType.SimOrderVerify: cn(u'流量卡充值对账'),
    }

    MapSubType = {
        DealerPaySubType.AUTO_SIM_CARD: ProductType.AutoSimCard,
        DealerPaySubType.MANUAL_SIM_CARD: ProductType.ManualSimCard,
        DealerPaySubType.SIM_CARD: ProductType.SimCard,
        DealerPaySubType.API_COST: ProductType.ApiCost,
        DealerPaySubType.DISABLE_AD: ProductType.DisableAd,
        DealerPaySubType.SIM_ORDER_VERIFY: ProductType.SimOrderVerify
    }

    MY_MAIN_TYPE = OrderMainType.PAY

    orderNo = StringField(verbose_name = u"订单号", unique = True)
    wxOrderNo = StringField(verbose_name = u"渠道订单号. 聚合商户,是聚合平台商户订单号;直连商户则是微信或支付宝订单号", default = "")
    transactionId = StringField(verbose_name = u"微信或者支付宝订单号", default = "")

    product = StringField(verbose_name = u'产品名称', default = ProductType.SimCard)

    items = ListField(verbose_name = u"充值商品", null = False)

    name = StringField(verbose_name = u'商品名称', default = '')

    dealerId = StringField(verbose_name = u"经销商ID", null = False)

    nickname = StringField(verbose_name = u"昵称", default = '')

    totalFee = IntField(verbose_name = u"订单金额(以分为单位)", min_value = 0, null = False)

    settleInfo = DictField(verbose_name = u'结算信息', default = {})

    status = StringField(verbose_name = u"充值结果", default = PayState.UnPaid)

    description = StringField(verbose_name = u"订单错误结果描述,一般为第三方错误码", default = "")

    createdTime = DateTimeField(default = datetime.datetime.now, verbose_name = u"生成时间")

    finishedTime = DateTimeField(default = None, verbose_name = u"支付完成时间")

    attachParas = DictField(verbose_name = u"消费模型信息", default = {})

    gateway = StringField(verbose_name = u'支付网关类型', default = '')
    payAppType = StringField(verbose_name = u'支付应用类型', default = '')
    payGatewayKey = StringField(verbose_name = u'支付网关key', default = '')

    withdrawSourceKey = StringField(verbose_name = u'提现网关source key', default = '')

    subject = StringField(verbose_name = u'商品标题', default = '')

    extraInfo = DictField(verbose_name = u"订单本身模型信息", default = None)

    meta = {'collection': 'dealer_recharge_record', 'db_alias': 'default',
            'indexes': [
                {
                    'fields': ['dealerId', 'product']
                },
                'orderNo', 'wxOrderNo', 'transactionId',
                # 'name'
            ]}

    search_fields = ('wxOrderNo', 'name')

    def __repr__(self):
        return '<DealerRechargeRecord id=%s>' % (str(self.id))

    @classmethod
    def get_product(cls, sub_type):
        return {
            'type': cls.MapSubType[sub_type],
            'desc': cls.ProductDesc[cls.MapSubType[sub_type]]
        }

    @classmethod
    def issue(cls, sub_type, payment_gateway, user, identifier=None, **payload):
        # type: (str, PaymentGateway, CapitalUser, basestring, Dict)->DealerRechargeRecord

        payload.update({
            'product': cls.get_product(sub_type)['type'],
            'gateway': payment_gateway.gateway_type,
            'payAppType': payment_gateway.pay_app_type,
            'payGatewayKey': payment_gateway.gateway_key,
            'subject': cls.get_product(sub_type)['desc'],
            'createdTime': datetime.datetime.now()
        })

        payload.update({
            'withdrawSourceKey': payment_gateway.withdraw_source_key()
        })

        if 'extraInfo' not in payload:
            payload['extraInfo'] = {}

        payload['status'] = cls.PayState.UnPaid

        identifier = identifier if identifier else str(user.id)
        
        if 'orderNo' not in payload:
            payload['orderNo'] = OrderNoMaker.make_order_no_32(
                identifier = identifier, main_type = cls.MY_MAIN_TYPE, sub_type = sub_type)

        record = cls(**payload)
        record.save()

        return record

    @classmethod
    def get_record(cls, order_no, dealer_id = None):
        # type: (str, str)->Optional[DealerRechargeRecord]

        # TODO: 分片需要带上片键
        record = cls.objects(orderNo = order_no).first()
        return record

    def succeed(self, **kwargs):
        payload = {
            'status': self.PayState.Paid
        }

        if kwargs:
            payload.update(kwargs)

        if 'finishedTime' not in payload:
            payload.update({'finishedTime': datetime.datetime.now()})

        result = DealerRechargeRecord.get_collection().update_one(
            filter = {'_id': ObjectId(self.id),
                      'status': {'$nin': [self.PayState.Paid, self.PayState.Close, self.PayState.Cancel]}},
            update = {'$set': payload},
            upsert = False
        )

        return result.matched_count == 1

    def fail(self, **kwargs):
        self.update(status = self.PayState.Failure, finishedTime = datetime.datetime.now(), **kwargs)
        return self.reload()

    def cancel(self):
        result = DealerRechargeRecord.get_collection().update_one(
            filter = {'_id': ObjectId(self.id), 'status': self.PayState.UnPaid},
            update = {'$set': {'status': self.PayState.Cancel,
                               'finishedTime': datetime.datetime.now()}},
            upsert = False
        )

        return result.matched_count == 1

    def quit(self):
        self.status = self.PayState.Quit
        self.save()
        return self

    def close(self, **kwargs):
        payload = {
            'status': self.PayState.Close
        }

        if kwargs:
            payload.update(kwargs)

        result = self.get_collection().update_one(
            filter = {'_id': ObjectId(self.id),
                      'status': {'$nin': [self.PayState.Paid, self.PayState.Close, self.PayState.Cancel]}},
            update = {'$set': kwargs},
            upsert = False
        )

        return result.matched_count == 1

    @property
    def is_success(self):
        return self.status == self.PayState.Paid

    @property
    def is_fail(self):
        return self.status == self.PayState.Failure

    @property
    def is_cancel(self):
        return self.status == self.PayState.Cancel

    @property
    def is_unpay(self):
        return self.status == self.PayState.UnPaid

    @property
    def is_close(self):
        return self.status == self.PayState.Close

    @property
    def sum_of_price(self):
        # type: ()->RMB
        return sum_rmb([RMB(_['price']) for _ in self.items])

    def get_body(self):
        return self.ProductDesc.get(self.product)

    @property
    def fen_total_fee(self):
        return self.totalFee

    @property
    def my_gateway(self):
        return self.gateway

    @property
    def pay_app_type(self):
        return self.payAppType

    @property
    def pay_gateway_key(self):
        return self.payGatewayKey

    @property
    def withdraw_source_key(self):
        return self.withdrawSourceKey

class RefundDealerRechargeRecord(RefundOrderBase):
    meta = {
        'collection': 'refund_dealer_recharge_record',
        'db_alias': 'default'
    }

    @classmethod
    def issue(cls, order, refundCash):
        # type:(DealerRechargeRecord, RMB)->RefundDealerRechargeRecord

        refund_order_no = OrderNoMaker.make_order_no_32(
            identifier = order.orderNo,
            main_type = OrderMainType.REFUND,
            sub_type = RefundSubType.REFUND)

        return cls(
            rechargeObjId = order.id,
            # refundSeq=next_seq,
            orderNo = refund_order_no,
            money = refundCash,
            status = cls.Status.CREATED,
            payAppType = order.pay_app_type).save()
    
    @property 
    def pay_app_type(self):
        return self.payAppType
    
    @property
    def pay_sub_order(self):
        # type: ()->DealerRechargeRecord

        if not hasattr(self, '__pay_sub_order__'):
            pay_order = DealerRechargeRecord.objects(id = str(self.rechargeObjId)).first()
            setattr(self, '__pay_sub_order__', pay_order)

        return getattr(self, '__pay_sub_order__')

    @pay_sub_order.setter
    def pay_sub_order(self, order):
        setattr(self, '__pay_sub_order__', order)

    @classmethod
    def get_record(cls, **kwargs):
        return cls.objects(**kwargs).first()

    @property
    def notify_url(self):
        if self.pay_app_type in [PayAppType.WECHAT, PayAppType.WECHAT_MINI]:
            return REFUND_NOTIFY_URL.WECHAT_REFUND_BACK
        elif self.pay_app_type == PayAppType.JD_AGGR:
            return REFUND_NOTIFY_URL.JD_AGGRE_REFUND_BACK
        elif self.pay_app_type == PayAppType.JD_OPEN:
            return REFUND_NOTIFY_URL.JDOPEN_REFUND_BACK
        else:
            return None

    
class OnSale(DynamicDocument):
    """
    优惠活动管理
    """
    dealerId = StringField(verbose_name = "经销商ID", null = False)

    name = StringField(verbose_name = '活动名称', default = '')

    showSite = IntField(verbose_name = "展现的位置", default = 0)  # 展现的位置0，表示扫码后的套餐页面。

    logicalCodeList = ListField(verbose_name = "需要推送的设备", null = False)

    onsaleType = StringField(verbose_name = '活动类型', default = '')

    detailDict = DictField(verbose_name = '活动详细数据', default = {})

    desc = StringField(verbose_name = '描述', default = '')

    img = StringField(verbose_name = '创意图片', default = '')

    onClickUrl = StringField(verbose_name = '回调地址', default = '')

    startTime = DateTimeField(verbose_name = '起始时间', default = datetime.datetime.now)

    endTime = DateTimeField(verbose_name = '结束时间', default = datetime.datetime.now)

    status = StringField(verbose_name = '状态', default = 'start')

    #: 展现类型，onlyOne，点击后就不弹出。forever，表示即使点击过，还是一直弹出，web:表示是页面，不是图片
    showType = StringField(verbose_name = '状态', default = 'onlyOne')

    onsaleTypeDict = {
        u'首次使用送金币': {
            'onClickUrl': '/user/promotion/getCoins',
            'desc': u'适用所有设备类型，用户首次扫码的时候，允许客户领金币进行体验，本次活动领过金币的用户，下次无法领取金币。',
            'expression': "",
            'img': '/app/img/marketing/first_give_coins.jpg',
            'showType': 'onlyOne'
        },
        u'首次免费按摩': {
            'onClickUrl': '/user/promotion/getDuration',
            'desc': u'适用按摩坐垫以及足疗机，用户首次扫码的时候，允许客户直接启动设备体验，本次活动体验过的用户，下次无法再体验。',
            'expression': "dev['devType']['code'] in ['100110','100111','100112','100113','100119','100900']",
            'img': '/app/img/marketing/first_free.jpg',
            'showType': 'onlyOne'
        },
        u'优惠充值大放送': {
            'onClickUrl': '/user/onsaleRecharge',
            'desc': u'适用所有设备类型，用户优惠充值，活动期间始终弹出此推广创意，即使用户已经充值优惠过了。(用户点击推广活动后，会跳转到对应设备的充值页面。需要提前在优惠设置中，把对应的地址下的充值套餐配置好)',
            'expression': "",
            'img': '/app/img/marketing/recharegeOnsale.jpg',
            'showType': 'forever'
        },
        u'手机实名免费用': {
            'onClickUrl': '/user/inputMobile',
            'desc': u'首次验证手机号后，可以免费使用设备',
            'expression': "",
            'img': '/app/img/marketing/inputMobile.jpg',
            'showType': 'onlyOne-web'
        },

        u'包年包月优惠多多': {
            'onClickUrl': '/user/onsaleTicketList',
            'desc': u'适用所有设备类型，用户优惠充值，活动期间始终弹出此推广创意，即使用户已经充值优惠过了。(用户点击推广活动后，会跳转到对应设备的充值页面。需要提前在优惠设置中，把对应的地址下的充值套餐配置好)',
            'expression': "",
            'img': '/app/img/marketing/recharegeOnsale.jpg',
            'showType': 'forever'
        },
    }

    def to_dict(self):
        return {
                'id': str(self.id),
                'name': self.name,
                'showSite': self.showSite,
                'logicalCodeList': self.logicalCodeList,
                'onsaleType': self.onsaleType,
                'onClickUrl': self.onClickUrl,
                'desc': self.desc,
                'detailDict': self.detailDict,
                'img': self.img,
                'startTime': self.startTime.strftime("%Y-%m-%d"),
                'endTime': self.endTime.strftime("%Y-%m-%d"),
                'status': self.status
            }


# 优惠活动访问记录
class OnSaleRecord(DynamicDocument):
    onsaleId = StringField(verbose_name = "经销商ID", null = False)
    userId = StringField(verbose_name = "用户ID", null = False)
    nickName = StringField(verbose_name = "用户昵称", null = False)
    addedTime = DateTimeField(verbose_name = "访问时间", default = datetime.datetime.now)
    onsaleDetail = DictField(verbose_name = "活动详情数据", default = {})

    meta = {
        "collection": "on_sale_record",
        "db_alias": "logdata",
    }


#: 经销商创建卡卷
class VirtualCard(Searchable):
    cardName = StringField(verbose_name = "卡名称", default = "")
    ownerId = StringField(verbose_name = "卡的发布老板", default = "")
    groupIds = ListField(verbose_name = "可用使用卡的地址", default = [])  # 空表示没有地址，*表示所有地址下可用
    devTypeList = ListField(verbose_name = "支持的设备类型清单", default = [])  # 存放设备类型的ID
    price = MonetaryField(verbose_name = "卡的售价", default = RMB('0.00'))
    createTime = DateTimeField(verbose_name = "创建时间", default = datetime.datetime.now)
    periodDays = FloatField(verbose_name = "卡的可用天数", default = 30.0)
    expiredTime = DateTimeField(verbose_name="卡卷售卖下架时间", default=lambda : datetime.datetime.now() + datetime.timedelta(days=365))
    dayQuota = ListField(verbose_name = "日限额度", default = [])  # {'unit':u'次','count':2}
    quota = ListField(verbose_name = "总的额度", default = [])
    userLimit = IntField(verbose_name = "用户数量限制", default = 10)
    userDesc = StringField(verbose_name = "用户侧的描述", default = "")
    dealerDesc = StringField(verbose_name = "经销商侧描述", default = "")
    status = IntField(verbose_name = "卡状态", default = 1)  # -1:删除；0：停止发售；1：开始发售

    # 作为内置参数，接口不暴露出来 使用的时候后台脚本修改， 修改脚本名称为 adjust_virtual_card_need_renew
    needRenewMax = IntField(verbose_name = "过期后可续卡天数", default=10)
    needRenewMin = IntField(verbose_name = "过期前可续卡天数", default=10)

    # 暂停发售之后是否可以继续续卡
    renewIgnoreStatus = BooleanField(verbose_name = "暂停发售之后是否可以继续续卡", default = False)

    # 坤元虚拟卡特有:功率
    power = IntField(verbose_name = "包月卡功率限制", default = 0)

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

    @property
    def is_expired(self):
        return self.expiredTime <= datetime.datetime.now()

    @property
    def description(self):
        """
        用于描述虚拟卡的信息
        :return:
        """
        try:
            qStr = u"额度："
            for item in self.quota:
                qStr += "{} {}".format(item.get("count"), item.get("unit"))
            tStr = u"时间：{}天".format(self.periodDays)
            return "{}\t{}".format(qStr, tStr)
        except Exception:
            return None

    @classmethod
    def get_last(cls, ownerId):
        return cls.objects.filter(ownerId=ownerId).order_by("-id").first()

    @property
    def groups(self):
        if '*' in self.groupIds:
            groups = '*'
        else:
            groups = [{'groupName': grp['groupName'], 'address': grp['address'], 'groupId': grp['groupId']} for grp in
                      Group.get_groups_by_group_ids(self.groupIds).values()]

        return groups

    @property
    def devTypes(self):
        devs = DeviceType.objects.filter(id__in=self.devTypeList).only("majorDeviceType", "name")
        devTypes = list()

        for _dev in devs:   # type: DeviceType
            devTypes.append({"devTypeName": _dev.name, "majorDeviceType": _dev.majorDeviceType})

        return devTypes

    def to_dict(self):
        return {
            "id": str(self.id),
            "cardId": str(self.id),
            "cardName": self.cardName,
            "price": self.price,
            "periodDays": self.periodDays,
            "createTime": self.createTime,
            "expiredTime": self.expiredTime,
            "userDesc": self.userDesc,
            "dealerDesc": self.dealerDesc,
            "status": self.status,
            'isExpired': self.is_expired,
            'power': self.power,
            "quota": self.quota,
            "dayQuota": self.dayQuota,
            "groups": self.groups,
            "devTypes": self.devTypes,
            'userLimit': self.userLimit
        }


class ElecPriceTemplate(Searchable):
    ownerId = StringField(verbose_name = "卡的发布老板", default = "")
    name = StringField(verbose_name = "卡的发布老板", default = "")
    priceList = ListField(verbose_name = "48分时价格", default = [])


class ItemType(Searchable):
    ownerId = StringField(verbose_name = "ownerId", default = "")
    title = StringField(verbose_name = "标题", default = "")
    desc = StringField(verbose_name = "描述", default = "")
    picUrl = StringField(verbose_name = "图片", default = "")
    price = IntField(verbose_name = "价格，分", default = 1)


class SubAccount(UserSearchable):
    """
    子账号表
    """
    agentId = StringField(verbose_name = "agentId", min_length = 1)

    phone = StringField(verbose_name = "电话", default = "")

    bossId = ObjectIdField(verbose_name = "主账号ID")
    permissionList = ListField(verbose_name = "菜单清单", default = [])

    meta = {
        'collection': 'sub_account',
        'db_alias': 'default',
        'indexes': [
            {
                'fields': ['username', 'bossId'], 'unique': True
            },
            {
                'fields': ['username', 'agentId'], 'unique': True
            }
        ],
    }

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

    @property
    def myBoss(self):
        # type:()->Dealer

        if hasattr(self, '_myBoss'):
            return getattr(self, '_myBoss')
        else:
            myBoss = Dealer.objects.get(id = self.bossId)
            setattr(self, '_myBoss', myBoss)
            return myBoss

    def query_feature_by_list(self, queryList):
        return self.myBoss.query_feature_by_list(queryList)

    def get_feature(self, feature_name):
        return self.myBoss.get_feature(feature_name)

    def query_home_page_layout(self):
        return self.myBoss.query_home_page_layout()

    def to_dict(self):
        value = self.myBoss.to_dict()
        value.update(
            {
                'username': self.username,
                'nickname': self.nickname,
                'role': ROLE.subaccount
            }
        )
        return value

    @property
    def switches(self):
        return self.myBoss.switches

    @property
    def isManagerialOpenIdBound(self):
        return self.myBoss.isManagerialOpenIdBound

    def get_bound_pay_openid(self, key):
        return self.myBoss.get_bound_pay_openid(key)

    @property
    def format_default_discount(self):
        return self.myBoss.format_default_discount

    def get_own_devices(self):
        groupIds = Group.get_group_ids_of_dealer(str(self.bossId))
        devNoList = Device.get_devNos_by_group(groupIds)
        return Device.get_dev_by_nos(devNoList).values()

    @property
    def income_aggregate_source(self):
        # type: ()->Dict[str, AnyStr]
        """
        e.g. {'ad': u'广告'}
        :return:
        """
        g = itemgetter(*self.myBoss.supportedIncomeAggregate)
        return dict(zip(self.myBoss.supportedIncomeAggregate, g(DEALER_INCOME_SOURCE_TRANSLATION)))

    @property
    def consumption_aggregate_source(self):
        # type: ()->Dict[str, AnyStr]
        """
        e.g. {'elect': u'电量'}
        :return:
        """
        g = itemgetter(*self.myBoss.supportedConsumptionAggregate)
        return dict(zip(self.myBoss.supportedConsumptionAggregate, g(DEALER_CONSUMPTION_AGG_KIND_TRANSLATION)))

    def set_agent_profit_share(self, share):
        # type: (Percent)->int
        """

        :param share:
        :return:
        """
        updated = self.myBoss.update(agentProfitShare = share.mongo_amount)
        return updated

    @property
    def adShow(self):
        return self.myBoss.adShow

    @property
    def ad_show(self):
        return self.myBoss.ad_show

    @property
    def pushBrokerUrl(self):
        return self.myBoss.pushBrokerUrl

    @property
    def isPurePartner(self):
        return self.myBoss.isPurePartner

    @property
    def wechat(self):
        return self.myBoss.wechat

    @property
    def description(self):
        return self.myBoss.description

    @property
    def cardNo(self):
        return self.myBoss.cardNo

    @property
    def bankName(self):
        return self.myBoss.bankName

    @property
    def cardHolder(self):
        return self.myBoss.cardHolder

    @property
    def isRead(self):
        return self.myBoss.isRead

    @property
    def noPassReason(self):
        return self.myBoss.noPassReason

    @property
    def cibMerchant(self):
        return self.myBoss.cibMerchant

    @property
    def managerialAppId(self):
        return self.myBoss.managerialAppId

    @property
    def managerialOpenId(self):
        return self.myBoss.managerialOpenId

    @property
    def wechatAuthUserInfo(self):
        return self.myBoss.wechatAuthUserInfo

    @property
    def managerialWechatBoundTime(self):
        return self.myBoss.managerialWechatBoundTime

    @property
    def agentProfitShare(self):
        return self.myBoss.agentProfitShare

    @property
    def beforeCharge(self):
        return self.myBoss.beforeCharge

    @property
    def noRecharge(self):
        return self.myBoss.noRecharge

    @property
    def defaultWashConfig(self):
        return self.myBoss.defaultWashConfig

    @property
    def withdrawlNotify(self):
        return self.myBoss.withdrawlNotify

    @property
    def offlineNotify(self):
        return self.myBoss.offlineNotifySwitch

    @property
    def dailyIncomeReportPushSwitch(self):
        return self.myBoss.dailyIncomeReportPushSwitch

    @property
    def newUserPaymentOrderPushSwitch(self):
        return self.myBoss.newUserPaymentOrderPushSwitch

    @property
    def devFaultPushDealerSwitch(self):
        return self.myBoss.devFaultPushDealerSwitch

    @property
    def devFaultPushUserSwitch(self):
        return self.myBoss.devFaultPushUserSwitch

    @property
    def currency_mode(self):
        return self.myBoss.currency_mode

    @property
    def serviceName(self):
        return self.myBoss.serviceName

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

    @property
    def qrcodeUrl(self):
        return self.myBoss.qrcodeUrl

    @property
    def withdrawFeeRatio(self):
        return self.myBoss.withdrawFeeRatio

    @property
    def annualTrafficCost(self):
        return self.myBoss.annualTrafficCost

    @property
    def limitAdLocation(self):
        return self.myBoss.limitAdLocation

    @property
    def defaultDiscountConfig(self):
        return self.myBoss.defaultDiscountConfig

    @property
    def supportedIncomeAggregate(self):
        return self.myBoss.supportedIncomeAggregate

    @property
    def supportedConsumptionAggregate(self):
        return self.myBoss.supportedConsumptionAggregate

    @property
    def bottomAd(self):
        return self.myBoss.bottomAd

    @property
    def limitDevNum(self):
        return self.myBoss.limitDevNum

    @property
    def feature_boolean_map(self):
        return self.myBoss.feature_boolean_map

    @property
    def features(self):
        return self.myBoss.features

    @property
    def normal(self):
        my_boss = self.myBoss  # type: Dealer
        if not my_boss.normal:
            return my_boss.normal
        else:
            return self.status == self.Status.NORMAL

    @property
    def forbidden(self):
        my_boss = self.myBoss  # type: Dealer
        if my_boss.forbidden:
            return my_boss.forbidden
        else:
            return self.status == self.Status.FORBIDDEN

    @property
    def abnormal(self):
        my_boss = self.myBoss  # type: Dealer
        if my_boss.abnormal:
            return my_boss.abnormal
        else:
            return self.status == self.Status.ABNORMAL

    @property
    def no_withdraw(self):
        return False

    @property
    def is_inhouse_wallet(self):
        return False

    def limit_filter_date(self, startTime, endTime):
        my_boss = self.myBoss  # type: Dealer
        return my_boss.limit_filter_date(startTime, endTime)

    @property
    def service_phone(self):
        my_boss = self.myBoss  # type: Dealer
        return my_boss.service_phone


class Complaint(Searchable):
    openId = StringField(verbose_name = 'openId', default = '')
    logicalCode = StringField(verbose_name = 'logicalCode', default = '')
    groupName = StringField(verbose_name = 'groupName', default = '')
    orderNo = StringField(verbose_name = 'orderNo', default = '')
    handledStatus = StringField(verbose_name = 'status', default = 'init')  # init/notice/warning/forbidden
    reason = StringField(verbose_name = 'reason', default = '')
    dateTimeAdded = DateTimeField(verbose_name = u'添加进来的时间', default = datetime.datetime.now)
    handledTime = DateTimeField(verbose_name = u'添加进来的时间', default = datetime.datetime.now)

    meta = {'collection': 'complaint', 'db_alias': 'logdata'}


# 发送给经销商的消息
class DealerMessage(Searchable):
    ownerId = StringField(verbose_name = 'ownerId', default = '')
    fromUser = StringField(verbose_name = 'fromUser', default = '')
    messageType = StringField(verbose_name = 'messageType', default = '')
    title = StringField(verbose_name = 'title', default = '')
    desc = StringField(verbose_name = 'desc', default = '')
    read = BooleanField(verbose_name = 'status', default = False)
    dateTimeAdded = DateTimeField(verbose_name = u'添加时间', default = datetime.datetime.now)
    readTime = DateTimeField(verbose_name = u'添加时间', default = datetime.datetime.now)
    relatedInfo = DictField(verbose_name = u'关联信息，便于查找，比如投诉信息', default = {})

    meta = {'collection': 'dealer_message', 'db_alias': 'logdata'}


# 经销商的换货单
class ExchangeOrder(Searchable):
    factoryId = ObjectIdField(verbose_name = 'factoryId', default = '')
    agentId = ObjectIdField(verbose_name = 'agentId', default = '')
    ownerId = ObjectIdField(verbose_name = 'ownerId', default = '')
    dateTimeAdded = DateTimeField(verbose_name = 'dateTimeAdded', default = datetime.datetime.now())

    dealerStatus = StringField(verbose_name = u'经销商侧状态', default = '')  # 比如已提单、已发货等
    partIds = ListField(verbose_name = u'换货单', default = [])
    pics = ListField(verbose_name = u'用户上传的图片', default = [])
    dealerWords = StringField(verbose_name = u'经销商留言', default = '')
    factoryAddr = DictField(verbose_name = u'厂家维修地址', default = {})
    dealerOrderNo = StringField(verbose_name = u'经销商的快递单号', default = '')

    factoryStatus = StringField(verbose_name = u'厂家侧的状态', default = '')  # 比如已发货、已同意换货等
    factoryOrderNo = StringField(verbose_name = u'厂家的快递单号', default = '')
    dealerAddr = DictField(verbose_name = u'经销商地址', default = {})
    factoryWords = StringField(verbose_name = u'厂家留言', default = '')

    more = StringField(verbose_name = u'备注', default = '')

    meta = {'collection': 'exchange_order', 'db_alias': 'default'}
    search_fields = ('dealerWords', 'factoryWords', 'factoryOrderNo', 'dealerOrderNo')

    def make_status_info(self):
        dealerInfo = ''
        factoryInfo = ''
        if self.dealerStatus == 'created':
            dealerInfo = u'经销商申请换板'
        elif self.dealerStatus == 'sended':
            dealerInfo = u'经销商已发货'
        if self.factoryStatus == 'agreed':
            factoryInfo = u'厂家同意换货'
        elif self.factoryStatus == 'disagreed':
            factoryInfo = u'厂家不同意'
        elif self.factoryStatus == 'sended':
            factoryInfo = u'售后中心已发货'
        elif self.factoryStatus == 'closed':
            factoryInfo = u'成功关闭'
        elif self.factoryStatus == '':
            factoryInfo = u'待确认'
        if dealerInfo and factoryInfo:
            return '%s，%s' % (dealerInfo, factoryInfo)
        elif dealerInfo:
            return dealerInfo
        else:
            return factoryInfo


# 卡的上分记录表
class UpCardScoreRecord(Searchable):
    cardId = StringField(verbose_name = "cardId", default = '')
    cardNo = StringField(verbose_name = "cardNo", default = '')
    ownerId = StringField(verbose_name = "所有者", default = "")
    dateTimeAdded = DateTimeField(verbose_name = "时间", default = datetime.datetime.now)
    score = FloatField(verbose_name = "上分数量", default = 0.0)
    address = StringField(verbose_name = "绑定地址", default = "")
    remark = StringField(verbose_name = "备注", default = "")

    meta = {"collection": "UpCardScoreRecord", "db_alias": "logdata"}

    @classmethod
    def get_collection(cls):
        return cls._get_collection()


class PermissionInfo(Searchable):  # 权限模版
    pid = StringField(verbose_name = 'pid', null = True, auto_incre = True)
    key = StringField(verbose_name = 'key', unique = True, null = False)
    value = BooleanField(verbose_name = 'value', default = False)
    name = StringField(verbose_name = '权限名称', default = '')
    desc = StringField(verbose_name = '权限详情', default = '')
    dateTimeAdded = DateTimeField(verbose_name = '添加时间', default = datetime.datetime.now)

    meta = {
        'collection': 'permission_info',
        'db_alias': 'default',
    }

    @classmethod
    def get_permissions(cls):

        def run(head):
            if head.count() == 0:
                return

            res = {}
            for item in head:
                value = run(all_Permission.filter(pid = str(item.id)))  # 结果为None or dict
                res[item.key] = value or item.value
            return res

        all_Permission = cls.objects.all()
        head = all_Permission.filter(pid = None)
        return run(head)


class PermissionRole(Searchable):  # 成员表

    AUTHORIZE_TYPE = ['dealer_to_dealer', 'dealer_to_subAccount']

    roleName = StringField(verbose_name='角色名称', null=True, default='')
    operId = StringField(verbose_name='操作人ID')
    dealerId = StringField(verbose_name='经销商ID')
    subAccountId = StringField(verbose_name='子账号ID')
    permissionRuleId = StringField(verbose_name='权限配置')
    dateTimeAdded = DateTimeField(verbose_name='添加时间', default=datetime.datetime.now)
    lastModifiedTime = DateTimeField(verbose_name='最后一次操作时间', default=datetime.datetime.now)
    authorizeType = StringField(verbose_name='授权关系',choices = AUTHORIZE_TYPE)
    isActive = BooleanField(verbose_name='激活', default=False)

    meta = {
        'collection': 'permission_role',
        'db_alias': 'default',
    }

    @staticmethod
    def add_dealer_role(oper_id, dealerId):
        rule = PermissionRule.objects.create(ruleName=str(uuid.uuid4()), dealerId=dealerId,
                                             permissionDict=PermissionInfo.get_permissions())

        models = {
            'roleName': None,
            'operId': oper_id,
            'dealerId': dealerId,
            'permissionRuleId': str(rule.id),
            'authorizeType': 'dealer_to_dealer',
        }

        return PermissionRole.objects.create(**models)

    @staticmethod
    def add_subAccount_role(oper_id, dealerId):
        rule = PermissionRule.objects.create(ruleName=str(uuid.uuid4()), dealerId=dealerId,
                                             permissionDict=PermissionInfo.get_permissions())

        models = {
            'roleName': None,
            'operId': oper_id,
            'dealerId': dealerId,
            'permissionRuleId': str(rule.id),
            'authorizeType': 'dealer_to_subAccount',
        }

        return PermissionRole.objects.create(**models)

    def save(self, **kw):
        self.lastModifiedTime = datetime.datetime.now()
        return super(PermissionRole, self).save(**kw)

    @classmethod
    def get_role_permission(cls, dealerId, operId):
        cacheKey = '{}-{}'.format(dealerId, operId)
        permissionRule = cache.get(cacheKey)
        if not permissionRule:
            role = cls.objects.filter(dealerId=dealerId, operId=operId, isActive=True).first()
            permissionRule = {}
            if role:
                permissionRule = PermissionRule.objects.get(id=role.permissionRuleId).permissionDict
            cache.set(cacheKey, permissionRule)
        return permissionRule

    @classmethod
    def delete_role_permission(cls, operIds, dealerId):
        cls.objects.filter(operId__in=operIds, dealerId=dealerId).update(isActive=False)
        for operId in operIds:
            cacheKey = '{}-{}'.format(dealerId, operId)
            cache.delete(cacheKey)

    @classmethod
    def get_auth_to_dealer(cls, dealerId):
        """
        获取 本账号授权的经销商
        :param dealerId:
        :return:
        """
        return cls.objects.filter(dealerId=dealerId, authorizeType='dealer_to_dealer', isActive=True).values_list(
            'operId')

    @classmethod
    def get_auth_to_sub(cls, dealerId):
        """
        获取 本账号授权的子账号
        :param dealerId:
        :return:
        """
        return cls.objects.filter(dealerId=dealerId, authorizeType='dealer_to_subAccount', isActive=True).values_list(
            'operId')

    @classmethod
    def get_is_auth_list(cls, operId, authorizeType):
        """
        
        :param operId: 操作人ID
        :param authorizeType: 授权类型
        :return:
        """
        return cls.objects.filter(operId=str(operId), authorizeType=authorizeType, isActive=True).values_list('dealerId')


class PermissionRule(Searchable):  # 权限配置表
    ruleName = StringField(verbose_name = '配置名称')
    dealerId = StringField(verbose_name = '经销商')
    permissionDict = DictField(verbose_name = '权限内容')
    dateTimeAdded = DateTimeField(verbose_name = '添加时间', default = datetime.datetime.now)
    lastModifiedTime = DateTimeField(verbose_name = '最后一次操作时间', default = datetime.datetime.now)

    meta = {
        'collection': 'permission_rule',
        'db_alias': 'default',
    }

    def get_permissionDict(self):

        def update_permission(base_dict, update_dict):
            result = {}
            for key, value in base_dict.items():
                if isinstance(value, dict):
                    res = update_permission(value, update_dict[key])
                    result[key] = res
                else:
                    # print key,value,update_dict
                    if not isinstance(update_dict, dict):
                        result[key] = update_dict
                    else:
                        if key in update_dict:
                            result[key] = update_dict[key]
                        else:
                            result[key] = value
            return result

        return update_permission(PermissionInfo.get_permissions(), self.permissionDict)

    def save(self, **kw):
        self.lastModifiedTime = datetime.datetime.now()
        return super(PermissionRule, self).save(**kw)


class TodoMessage(Searchable):
    title = StringField(verbose_name=u"标题")
    content = StringField(verbose_name=u"内容")
    type = IntField(verbose_name=u"待办的种类", choices=TodoTypeEnum.choices())
    link = StringField(verbose_name=u"跳转地址")
    done = IntField(verbose_name=u"待办状态", choices=TodoDone.choices(), default=TodoDone.INIT)
    ownerId = StringField(verbose_name=u"经销商")
    expiredTime = DateTimeField(verbose_name=u"信息过期时间")
    dateTimeAdded = DateTimeField(verbose_name=u"信息添加的时间", default=datetime.datetime.now())

    popOnlyOnce = BooleanField(verbose_name=u"登录后台只显示一次", default=True)

    def to_dict(self):
        return {
            "id": self.id,
            "title": self.title,
            'content': self.content,
            "type": self.type,
            "link": self.link,
            "status": self.done,
            "ownerId": self.ownerId,
            'popOnlyOnce': self.popOnlyOnce
        }

    def has_done(self):
        self.update(done=TodoDone.DONE)

    @property
    def checkModel(self):
        for _ in TodoTypeEnum:
            if _ == self.type:
                return _._load_todoCls()

    @classmethod
    def get_todo_message(cls, user, typeList = TodoTypeEnum.choices()):
        """
        获取所有未完成的任务
        """

        items = cls.objects.filter(
            ownerId = str(user.id),
            done__in = [TodoDone.INIT, TodoDone.ING],
            type__in = typeList,
            expiredTime__gt = datetime.datetime.now())

        dataList = list()
        for _m in items:  # type: TodoMessage
            checkModel = _m.checkModel
            # 进行一次任务检查 查看任务是否完成以及是否需要强制执行
            hasDone, force = checkModel.check_has_done(_m)
            if hasDone:
                _m.has_done()
                continue

            data = _m.to_dict()
            data["force"] = force
            dataList.append(data)

        return dataList

    @classmethod
    def sim_expire_message(cls, ownerId, expireCount):
        params = {
            'type': 'simCard',
            'redirect': concat_front_end_url(uri = '/app/deviceCard.html')
        }

        link = add_query(concat_server_end_url(uri = '/dealer/wechat/entry'), params)

        msg = cls(
            id = "1",
            title = u"流量卡续费提醒",
            content = u"您当前有{}台设备流量卡需要续费，为了避免影响设备的正常运行，请您尽快续费（已经充值请忽略）".format(expireCount),
            type = TodoTypeEnum.SMS_TODO.code,
            link = link,
            ownerId = ownerId,
            done = int(TodoDone.ING),
            expiredTime = datetime.datetime.now() + datetime.timedelta(days = 365),
            popOnlyOnce = False
        ).to_dict()
        return msg