import traceback

import pymongo
from pymongo import UpdateOne
from pydantic import Field

from model import BaseModel
from .task import Task
from utils import *


class AddUserFlag(Task):
    """
    添加新设备标记
    """

    class Model(BaseModel):
        game_role_id: str = Field(..., min_length=1, title="角色id", alias='_game_role_id')
        device_id: str = Field(..., min_length=1, title='设备id', alias='_device_id')
        channel_uid: str = Field(..., min_length=1, title="channel_uid", alias='_channel_uid')
        role_create_time: int = Field(..., min_length=1, title="注册时间戳")

    def cleaning(self, cursor_list):
        for cursor in cursor_list:  # type:dict
            for source_coll, ts in cursor.items():  # type:str,dict
                if ts['cursor_st'] == ts['cursor_et']:
                    continue
                logger.info(f'开始处理{self.game_name} 处理 {source_coll} 游标 {ts}')
                projection = self.Model.get_fields()
                bulk_data = []
                # 处理新账号
                where = {
                    'role_create_time': {
                        '$gte': ts['cursor_st'],
                        '$lt': ts['cursor_et'],
                    },
                    'is_new_channel_uid': {'$exists': False}
                }

                for item in self.local_db[source_coll].find(where, projection).sort('role_create_time',
                                                                                    pymongo.ASCENDING):
                    try:
                        # 新账号
                        model = self.Model(**item)
                        # 查找该channel_uid在之前有么
                        role_cnt = self.local_db[source_coll].count(
                            {'_channel_uid': model.channel_uid, 'role_create_time': {'$lt': model.role_create_time}})
                        if not role_cnt:
                            bulk_data.append(
                                UpdateOne({'_game_role_id': model.game_role_id},
                                          {'$set': {'is_new_channel_uid': 1}}))
                    except Exception as e:
                        logger.error(f'msg:{e}')
                        # pass


                # 处理设备
                where = {
                    'role_create_time': {
                        '$gte': ts['cursor_st'],
                        '$lt': ts['cursor_et'],
                    },
                    'is_new_device': {'$exists': False}
                }
                for item in self.local_db[source_coll].find(where, projection).sort('role_create_time',
                                                                                    pymongo.ASCENDING):
                    try:
                        # 新设备
                        model = self.Model(**item)
                        # 查找该_device_id在之前有么
                        role_cnt = self.local_db[source_coll].count(
                            {'_device_id': model.device_id, 'role_create_time': {'$lt': model.role_create_time}})
                        if not role_cnt:
                            bulk_data.append(
                                UpdateOne({'_game_role_id': model.game_role_id},
                                          {'$set': {'is_new_device': 1}}))
                    except Exception as e:
                        logger.error(f'msg:{e}')
                        pass

                # 记录第一次登录设备id
                where = {
                    'role_create_time': {
                        '$gte': ts['cursor_st'],
                        '$lt': ts['cursor_et'],
                    },
                    '_first_device_id': ''
                }
                for item in self.local_db[source_coll].find(where, projection):
                    try:
                        # 新设备
                        model = self.Model(**item)
                        bulk_data.append(
                            UpdateOne({'_game_role_id': model.game_role_id}, {'$set': {'_first_device_id': model.device_id}}))
                    except Exception as e:
                        msg = traceback.format_exc()
                        ddsend_msg(f'{self.game_name}.{source_coll}字段异常 {msg}')
                        logger.error(repr(e))

                if bulk_data:
                    self.local_db[self.dest_coll].bulk_write(bulk_data, ordered=False)
                    self.remote_db[self.dest_coll].bulk_write(bulk_data, ordered=False)
                self.set_cursor(cursor_st=ts['cursor_st'], cursor_et=ts['cursor_et'])