# -*- coding: utf-8 -*-

from __future__ import division

import xlrd
from django.conf import settings
from django.db import transaction
from pysnippets.strsnippets import strip
from TimeConvert import TimeConvert as tc

from integral.models import SaleclerkSubmitLogInfo
from mch.models import (BrandInfo, ConsumeInfoSubmitLogInfo, DistributorInfo, ModelCameraBodyInfo, ModelInfo,
                        SaleclerkInfo)
from statistic.models import (ConsumeDistributorSaleStatisticInfo, ConsumeModelSaleStatisticInfo,
                              ConsumeProvinceSaleStatisticInfo, ConsumeSaleStatisticInfo, ConsumeUserStatisticInfo,
                              DistributorSaleStatisticInfo, ModelSaleStatisticInfo, ProvinceSaleStatisticInfo,
                              SaleclerkSaleStatisticInfo, SaleStatisticInfo)
from utils.redis.connect import r
from utils.redis.rkeys import MINI_PROGRAM_GIS_LIST


PROVINCE_LIST = {
    "110000": "北京市",
    "120000": "天津市",
    "130000": "河北省",
    "140000": "山西省",
    "150000": "内蒙古自治区",
    "210000": "辽宁省",
    "220000": "吉林省",
    "230000": "黑龙江省",
    "310000": "上海市",
    "320000": "江苏省",
    "330000": "浙江省",
    "340000": "安徽省",
    "350000": "福建省",
    "360000": "江西省",
    "370000": "山东省",
    "410000": "河南省",
    "420000": "湖北省",
    "430000": "湖南省",
    "440000": "广东省",
    "450000": "广西壮族自治区",
    "460000": "海南省",
    "500000": "重庆市",
    "510000": "四川省",
    "520000": "贵州省",
    "530000": "云南省",
    "540000": "西藏自治区",
    "610000": "陕西省",
    "620000": "甘肃省",
    "630000": "青海省",
    "640000": "宁夏回族自治区",
    "650000": "新疆维吾尔自治区",
    "710000": "台湾省",
    "810000": "香港特别行政区",
    "820000": "澳门特别行政区"
}


def pre_provinces():
    brands = BrandInfo.objects.filter(status=True)
    for brand in brands:
        for pcode, pname in PROVINCE_LIST.items():
            pssi, created = ProvinceSaleStatisticInfo.objects.get_or_create(brand_id=brand.brand_id, province_code=pcode, ymd=0)
            pssi.province_name = pname
            pssi.save()

            cpssi, created = ConsumeProvinceSaleStatisticInfo.objects.get_or_create(brand_id=brand.brand_id, province_code=pcode, ymd=0)
            cpssi.province_name = pname
            cpssi.save()


def pre_models():
    brands = BrandInfo.objects.filter(status=True)
    for brand in brands:
        models = ModelInfo.objects.filter(brand_id=brand.brand_id, status=True)
        for mdl in models:
            mssi, created = ModelSaleStatisticInfo.objects.get_or_create(brand_id=brand.brand_id, model_id=mdl.model_id, ymd=0)
            mssi.model_name = mdl.model_name
            mssi.save()

            cmssi, created = ConsumeModelSaleStatisticInfo.objects.get_or_create(brand_id=brand.brand_id, model_id=mdl.model_id, ymd=0)
            cmssi.model_name = mdl.model_name
            cmssi.save()


def pre_distributors():
    brands = BrandInfo.objects.filter(status=True)
    for brand in brands:
        distributors = DistributorInfo.objects.filter(brand_id=brand.brand_id, status=True)
        for dtbt in distributors:
            dssi, created = DistributorSaleStatisticInfo.objects.get_or_create(brand_id=brand.brand_id, distributor_id=dtbt.distributor_id, ymd=0)
            dssi.distributor_name = dtbt.distributor_name
            dssi.save()

            cdssi, created = ConsumeDistributorSaleStatisticInfo.objects.get_or_create(brand_id=brand.brand_id, distributor_id=dtbt.distributor_id, ymd=0)
            cdssi.distributor_name = dtbt.distributor_name
            cdssi.save()


def pre_all():
    pre_provinces()
    pre_models()
    pre_distributors()


# In [55]: cv
# Out[55]: 991000295147.0
#
# In [56]: str(cv)
# Out[56]: '9.91000295147e+11'
#
# In [57]: repr(cv)
# Out[57]: '991000295147.0'
def convert_to_str(cv):
    if isinstance(cv, (int, float)):
        cv = repr(cv)[:-2] if repr(cv).endswith('.0') else repr(cv)
    if isinstance(cv, str):
        cv = cv.strip()
    return cv


def pre_new_models(fpath='./pre/static/models_20180816.xls'):
    workbook = xlrd.open_workbook(fpath)
    # sheet = workbook.sheet_by_name('SMR')
    sheets = workbook.sheets()
    sheet = sheets[0]
    nrows = sheet.nrows
    for idx in range(1, nrows):
        rvals = sheet.row_values(idx)
        print(rvals)

        jancode = strip(rvals[0])
        if not jancode:
            continue

        mdl, _ = ModelInfo.objects.get_or_create(jancode=jancode)
        mdl.brand_id = settings.KODO_DEFAULT_BRAND_ID
        mdl.brand_name = settings.KODO_DEFAULT_BRAND_NAME
        mdl.model_name = strip(rvals[1])
        mdl.model_uni_name = strip(rvals[2])
        mdl.category = strip(rvals[3])
        mdl.model_full_name = strip(rvals[4])
        mdl.warehouse = strip(rvals[5])
        mdl.save()


def pre_adaptive_cameras(fpath=u'./pre/static/腾龙镜头适用机型V1.xlsx'):
    ModelCameraBodyInfo.objects.all().delete()

    workbook = xlrd.open_workbook(fpath)
    # sheet = workbook.sheet_by_name('SMR')
    # import ipdb;ipdb.set_trace()
    sheets = workbook.sheets()
    # Deal sheet0
    sheet0 = sheets[0]
    # Deal sheet1-n
    sheets = sheets[1:]
    for sheet in sheets:
        model_name = sheet.name
        model_full_name = sheet.row_values(2)[0]
        print(model_name, model_full_name)
        nrows = sheet.nrows
        for idx in range(4, nrows):
            rvals = sheet.row_values(idx)
            print(rvals)
            for val in rvals:
                val = strip(val)
                if not val:
                    continue
                if u'卡口' in val:
                    continue
                print(val)
                ModelCameraBodyInfo.objects.get_or_create(
                    brand_id=settings.KODO_DEFAULT_BRAND_ID,
                    brand_name=settings.KODO_DEFAULT_BRAND_NAME,
                    model_name=model_name,
                    model_full_name=model_full_name,
                    camera_name=val,
                )


def fill_ym():
    for o in ConsumeSaleStatisticInfo.objects.all():
        ymd = str(o.ymd)
        if len(ymd) == 8:
            # 月销量统计
            ssi, _ = ConsumeSaleStatisticInfo.objects.get_or_create(
                brand_id=o.brand_id,
                ymd=ymd[:6],
            )
            ssi.num += o.num
            ssi.save()
            # 年销量统计
            ssi, _ = ConsumeSaleStatisticInfo.objects.get_or_create(
                brand_id=o.brand_id,
                ymd=ymd[:4],
            )
            ssi.num += o.num
            ssi.save()

    for o in ConsumeModelSaleStatisticInfo.objects.all():
        ymd = str(o.ymd)
        if len(ymd) == 8:
            # 月型号销量统计
            mssi, _ = ConsumeModelSaleStatisticInfo.objects.get_or_create(
                brand_id=o.brand_id,
                model_id=o.model_id,
                ymd=ymd[:6],
            )
            mssi.model_name = o.model_name
            mssi.num += o.num
            mssi.save()
            # 年型号销量统计
            mssi, _ = ConsumeModelSaleStatisticInfo.objects.get_or_create(
                brand_id=o.brand_id,
                model_id=o.model_id,
                ymd=ymd[:4],
            )
            mssi.model_name = o.model_name
            mssi.num += o.num
            mssi.save()

    for o in SaleStatisticInfo.objects.all():
        ymd = str(o.ymd)
        if len(ymd) == 8:
            # 月销量统计
            ssi, _ = SaleStatisticInfo.objects.get_or_create(
                brand_id=o.brand_id,
                ymd=ymd[:6],
            )
            ssi.num += o.num
            ssi.save()
            # 年销量统计
            ssi, _ = SaleStatisticInfo.objects.get_or_create(
                brand_id=o.brand_id,
                ymd=ymd[:4],
            )
            ssi.num += o.num
            ssi.save()


def refreshs():
    ConsumeUserStatisticInfo.objects.all().delete()
    ConsumeSaleStatisticInfo.objects.all().delete()
    ConsumeProvinceSaleStatisticInfo.objects.all().delete()
    ConsumeModelSaleStatisticInfo.objects.all().delete()

    logs = ConsumeInfoSubmitLogInfo.objects.filter(verifyResult=1, dupload=False, test_user=False, status=True)

    for log in logs:
        ymd = tc.local_string(utc_dt=log.created_at, format='%Y%m%d')

        try:
            mdl = ModelInfo.objects.get(model_id=log.model_id)
        except ModelInfo.DoesNotExist:
            continue

        with transaction.atomic():
            cusi, _ = ConsumeUserStatisticInfo.objects.select_for_update().get_or_create(
                brand_id=mdl.brand_id,
                ymd=ymd,
            )
            cusi.users = list(set(cusi.users + [log.user_id]))
            cusi.num = len(cusi.users)
            cusi.save()
            cusi, _ = ConsumeUserStatisticInfo.objects.select_for_update().get_or_create(
                brand_id=mdl.brand_id,
                ymd=ymd[:6],
            )
            cusi.users = list(set(cusi.users + [log.user_id]))
            cusi.num = len(cusi.users)
            cusi.save()
            cusi, _ = ConsumeUserStatisticInfo.objects.select_for_update().get_or_create(
                brand_id=mdl.brand_id,
                ymd=ymd[:4],
            )
            cusi.users = list(set(cusi.users + [log.user_id]))
            cusi.num = len(cusi.users)
            cusi.save()

            cssi, _ = ConsumeSaleStatisticInfo.objects.select_for_update().get_or_create(
                brand_id=mdl.brand_id,
                ymd=ymd,
            )
            cssi.num += 1
            cssi.save()
            cssi, _ = ConsumeSaleStatisticInfo.objects.select_for_update().get_or_create(
                brand_id=mdl.brand_id,
                ymd=ymd[:6],
            )
            cssi.num += 1
            cssi.save()
            cssi, _ = ConsumeSaleStatisticInfo.objects.select_for_update().get_or_create(
                brand_id=mdl.brand_id,
                ymd=ymd[:4],
            )
            cssi.num += 1
            cssi.save()

            # 日型号销量统计
            cmssi, _ = ConsumeModelSaleStatisticInfo.objects.select_for_update().get_or_create(
                brand_id=mdl.brand_id,
                model_name=mdl.model_uni_name,
                ymd=ymd,
            )
            cmssi.num += 1
            cmssi.save()
            # 月型号销量统计
            cmssi, _ = ConsumeModelSaleStatisticInfo.objects.select_for_update().get_or_create(
                brand_id=mdl.brand_id,
                model_name=mdl.model_uni_name,
                ymd=ymd[:6],
            )
            cmssi.num += 1
            cmssi.save()
            # 年型号销量统计
            cmssi, _ = ConsumeModelSaleStatisticInfo.objects.select_for_update().get_or_create(
                brand_id=mdl.brand_id,
                model_name=mdl.model_uni_name,
                ymd=ymd[:4],
            )
            cmssi.num += 1
            cmssi.save()

            r.rpushjson(MINI_PROGRAM_GIS_LIST, {
                'brand_id': log.brand_id,
                'user_id': log.user_id,
                'lat': log.lat,
                'lon': log.lon,
                'phone': log.phone,
                'ymd': tc.local_string(utc_dt=log.created_at, format='%Y%m%d'),
                'model_id': log.model_id,
                'serialNo': log.serialNo,
                'pk': log.pk,
            })


def refreshs2():
    SaleStatisticInfo.objects.all().delete()
    ModelSaleStatisticInfo.objects.all().delete()
    DistributorSaleStatisticInfo.objects.all().delete()
    ProvinceSaleStatisticInfo.objects.all().delete()
    SaleclerkSaleStatisticInfo.objects.all().delete()

    logs = SaleclerkSubmitLogInfo.objects.filter(dupload=False, test_user=False, test_sn=False, status=True)

    for log in logs:
        ymd = tc.local_string(utc_dt=log.created_at, format='%Y%m%d')

        try:
            brand = BrandInfo.objects.get(pk=log.brand_pk)
        except BrandInfo.DoesNotExist:
            continue

        try:
            model = ModelInfo.objects.get(pk=log.model_pk)
        except ModelInfo.DoesNotExist:
            continue

        try:
            clerk = SaleclerkInfo.objects.get(clerk_id=log.clerk_id, status=True)
        except SaleclerkInfo.DoesNotExist:
            continue

        try:
            distributor = DistributorInfo.objects.get(distributor_id=clerk.distributor_id)
        except DistributorInfo.DoesNotExist:
            continue

        with transaction.atomic():
            # 日销量统计
            ssi, _ = SaleStatisticInfo.objects.select_for_update().get_or_create(
                brand_id=brand.brand_id,
                ymd=ymd,
            )
            ssi.num += 1
            ssi.save()
            # 月销量统计
            ssi, _ = SaleStatisticInfo.objects.select_for_update().get_or_create(
                brand_id=brand.brand_id,
                ymd=ymd[:6],
            )
            ssi.num += 1
            ssi.save()
            # 年销量统计
            ssi, _ = SaleStatisticInfo.objects.select_for_update().get_or_create(
                brand_id=brand.brand_id,
                ymd=ymd[:4],
            )
            ssi.num += 1
            ssi.save()

            # 型号销量统计
            mssi, _ = ModelSaleStatisticInfo.objects.select_for_update().get_or_create(
                brand_id=brand.brand_id,
                model_name=model.model_uni_name,
                ymd=ymd,
            )
            mssi.saleclerks += [clerk.clerk_id]
            mssi.num = len(mssi.saleclerks)
            mssi.save()

            mssi, _ = ModelSaleStatisticInfo.objects.select_for_update().get_or_create(
                brand_id=brand.brand_id,
                model_name=model.model_uni_name,
                ymd=ymd[:6],
            )
            mssi.saleclerks += [clerk.clerk_id]
            mssi.num = len(mssi.saleclerks)
            mssi.save()

            mssi, _ = ModelSaleStatisticInfo.objects.select_for_update().get_or_create(
                brand_id=brand.brand_id,
                model_name=model.model_uni_name,
                ymd=ymd[:4],
            )
            mssi.saleclerks += [clerk.clerk_id]
            mssi.num = len(mssi.saleclerks)
            mssi.save()

            # 经销商销量统计
            dssi, _ = DistributorSaleStatisticInfo.objects.select_for_update().get_or_create(
                brand_id=brand.brand_id,
                distributor_id=distributor.distributor_id,
                ymd=ymd,
            )
            dssi.distributor_name = distributor.distributor_name
            dssi.num += 1
            dssi.save()

            dssi2, _ = DistributorSaleStatisticInfo.objects.select_for_update().get_or_create(
                brand_id=brand.brand_id,
                distributor_id=distributor.distributor_id,
                ymd=0,
            )
            dssi2.distributor_name = distributor.distributor_name
            dssi2.num += 1
            dssi2.save()

            # 日省份销量统计
            pssi, _ = ProvinceSaleStatisticInfo.objects.select_for_update().get_or_create(
                brand_id=brand.brand_id,
                province_code=distributor.distributor_province_code,
                ymd=ymd,
            )
            pssi.province_name = distributor.distributor_province_name
            pssi.num += 1
            pssi.save()
            # 月省份销量统计
            pssi, _ = ProvinceSaleStatisticInfo.objects.select_for_update().get_or_create(
                brand_id=brand.brand_id,
                province_code=distributor.distributor_province_code,
                ymd=ymd[:6],
            )
            pssi.province_name = distributor.distributor_province_name
            pssi.num += 1
            pssi.save()
            # 年省份销量统计
            pssi, _ = ProvinceSaleStatisticInfo.objects.select_for_update().get_or_create(
                brand_id=brand.brand_id,
                province_code=distributor.distributor_province_code,
                ymd=ymd[:4],
            )
            pssi.province_name = distributor.distributor_province_name
            pssi.num += 1
            pssi.save()

            # 日销售员销量统计
            sssi, _ = SaleclerkSaleStatisticInfo.objects.select_for_update().get_or_create(
                brand_id=brand.brand_id,
                clerk_id=clerk.clerk_id,
                ymd=ymd,
            )
            sssi.distributor_id = distributor.distributor_id
            sssi.distributor_name = distributor.distributor_name
            sssi.distributor_short_name = distributor.distributor_short_name
            sssi.clerk_name = clerk.clerk_name
            sssi.num += 1
            sssi.save()
            # 月销售员销量统计
            sssi, _ = SaleclerkSaleStatisticInfo.objects.select_for_update().get_or_create(
                brand_id=brand.brand_id,
                clerk_id=clerk.clerk_id,
                ymd=ymd[:6],
            )
            sssi.distributor_id = distributor.distributor_id
            sssi.distributor_name = distributor.distributor_name
            sssi.distributor_short_name = distributor.distributor_short_name
            sssi.clerk_name = clerk.clerk_name
            sssi.num += 1
            sssi.save()
            # 年销售员销量统计
            sssi, _ = SaleclerkSaleStatisticInfo.objects.select_for_update().get_or_create(
                brand_id=brand.brand_id,
                clerk_id=clerk.clerk_id,
                ymd=ymd[:4],
            )
            sssi.distributor_id = distributor.distributor_id
            sssi.distributor_name = distributor.distributor_name
            sssi.distributor_short_name = distributor.distributor_short_name
            sssi.clerk_name = clerk.clerk_name
            sssi.num += 1
            sssi.save()