operator_queue.py

import heapq
import re
import time

from danmu import DanMuClient


class Operator:
    OP_NAME_DICT = []

    def __init__(self, input_name, user):
        input_name.lower()
        success = False
        for operator in self.OP_NAME_DICT:
            for nickname in self.OP_NAME_DICT[operator]:
                if input_name == nickname:
                    self.name = operator
                    success = True
        if not success:
            raise ValueError('Illegal operator name: ' + input_name)
        self.users = {user}
        self.time = time.time()

    def user_count(self):
        return len(self.users)

    def update(self, username):
        self.users.update({username})

    def match_name(self, input_name):
        return input_name.lower() in self.OP_NAME_DICT[self.name]

    def __eq__(self, other):
        return self.user_count() == other.user_count() and self.time == other.time

    def __lt__(self, other):
        return self.user_count() > other.user_count() \
               or self.user_count() == other.user_count() and self.time < other.time

    def __gt__(self, other):
        return not self == other and not self < other

    def __str__(self):
        return '[{0}: {1}]'.format(self.name, self.user_count())

    def __repr__(self):
        return self.__str__()


class Attacker(Operator):
    OP_NAME_DICT = {
        'sledge': ['sledge', '大锤', '锤哥', '锤子'],
        'thatcher': ['thatcher', 'emp', 'thatcher', '撒切尔'],
        'ash': ['ash'],
        'thermite': ['thermite', '热切', '铝热'],
        'montagne': ['montagne', 'monty', '全盾', '大盾'],
        'twitch': ['twitch', '电车', '电车妹'],
        'blitz': ['blitz', '闪盾', '闪光盾'],
        'iq': ['iq'],
        'fuze': ['fuze', '福泽', '咚咚咚', '咚子'],
        'glaz': ['glaz', '狙', '大狙'],
        'buck': ['buck', '加拿大自爆兵'],
        'blackbeard': ['blackbeard', 'bb', '胡哥', '黑胡子', '大胡子', '枪盾', '步枪盾'],
        'capitao': ['capitao', '开皮条', '队长', '巴西队长', '烧烤哥', '巴西烤肉', '烤肉队长'],
        'jackal': ['jackal', '足控', '脚气', '豺狼'],
        'ying': ['ying', '莹', '烛光'],
        'zofia': ['zofia', '佐菲亚', '人妻', '波兰大姐姐'],
        'dokkaebi': ['dokkaebi', '狗逼', '狗逼小姐姐', '美羊羊', '韩国小姐姐', '狗妹'],
        'hibana': ['hibana', '火花', '花火'],
        'lion': ['lion', '狮子', '狮子头', '河东狮吼', '里昂'],
        'finka': ['finka', '毛妹', '拳头', '无影拳']
    }


class Defender(Operator):
    OP_NAME_DICT = {
        'mute': ['mute', 'wifi', 'wi-fi'],
        'smoke': ['smoke', '毒气', '臭屁'],
        'castle': ['castle', '龙鳞板'],
        'pulse': ['pulse', '心跳'],
        'doc': ['doc', '医生', '庸医', ],
        'rook': ['rook'],
        'jager': ['jager', 'Jäger', '杰哥', '耶格', 'ADS', '耶哥', '耶大头', ],
        'bandit': ['bandit', '皮卡丘', '班迪', '电兵'],
        'tachanka': ['tachanka', 'lord', '机枪', '机枪哥'],
        'kapkan': ['kapkan', '绊雷', 'edd', '卡胖'],
        'frost': ['frost', '夹子', '夹子妹'],
        'valkrie': ['valkrie', '瓦基', '女武神', '瓦尔基里', '黑眼'],
        'caveira': ['caveira', 'cav', '女鬼', '审问'],
        'echo': ['echo', '死宅', '肥宅', '宅'],
        'mira': ['mira', '黑镜'],
        'lesion': ['lesion', '刘醒', '老中医'],
        'ela': ['ela', '波兰小姐姐'],
        'vigil': ['vigil', '伟哥', '白裤裆', '男鬼', '寒冬一鸡']
    }


class OperatorQueue:
    REQUEST_PATTERN = '(?:进攻|防守) *(.+)'

    def __init__(self, url, keyword):
        OperatorQueue.REQUEST_PATTERN = keyword + OperatorQueue.REQUEST_PATTERN
        print(OperatorQueue.REQUEST_PATTERN)
        self.changed = False
        self._dmc = DanMuClient(url)
        if not self._dmc.isValid():
            raise ValueError('Invalid url')

        self._attacker_nick_name = []
        for o in Attacker.OP_NAME_DICT.values():
            for nn in o:
                self._attacker_nick_name.append(nn)

        self._defender_nick_name = []
        for o in Defender.OP_NAME_DICT.values():
            for nn in o:
                self._defender_nick_name.append(nn)

        self._attacker_queue = []
        self._defender_queue = []

        @self._dmc.danmu
        def danmu_fn(msg):
            self.process(msg)

        self._dmc.start(blockThread=False)

    def find_operator_queue(self, nickname):
        if nickname.lower() in self._attacker_nick_name:
            return self._attacker_queue
        elif nickname.lower() in self._defender_nick_name:
            return self._defender_queue
        else:
            return None

    def process(self, msg):
        print('Processing: ')
        print('[{0}] \"{1}\"'.format(msg['NickName'], msg['Content']))
        m = re.match(OperatorQueue.REQUEST_PATTERN, msg['Content'].lower())
        if m:
            self.changed = True

            q = self.find_operator_queue(m.group(1))
            if q is None:
                print('Illegal operator name: ' + m.group(1))
                return

            find = False
            for operator in q:
                if operator.match_name(m.group(1)):
                    operator.update(msg['NickName'])
                    find = True
                    break

            if not find:
                try:
                    new_operator = Attacker(m.group(1), msg['NickName']) if q == self._attacker_queue \
                        else Defender(m.group(1), msg['NickName'])
                    q.append(new_operator)
                except ValueError as e:
                    print(e)
                    return
            heapq.heapify(q)
            print('    ' + repr(self._attacker_queue))
            print('    ' + repr(self._defender_queue))

    def peek_3(self, side):
        q = self._attacker_queue if side == 'attacker' \
            else self._defender_queue if side == 'defender' else None
        if len(q) < 3:
            return q
        lst = q[0:3]
        if q[1] > q[2]:
            lst[1], lst[2] = q[2], q[1]
        return lst

    def next_attacker(self):
        if len(self._attacker_queue) > 0:
            self._attacker_queue.pop(0)
            self.changed = True
        else:
            print('Attacker queue is empty!')

    def next_defender(self):
        if len(self._defender_queue) > 0:
            self._defender_queue.pop(0)
            self.changed = True
        else:
            print('Defender queue is empty!')

    def get_text(self):
        if not self.changed:
            return None

        self.changed = False
        text = 'ATTACKERS:\n'
        for op in self.peek_3('attacker'):
            text += '    ' + op.name.upper() + '\n'
        for _ in range(3 - len(self.peek_3('attacker'))):
            text += '    --\n'

        text += 'DEFENDERS:\n'
        for op in self.peek_3('defender'):
            text += '    ' + op.name.upper() + '\n'
        for _ in range(3 - len(self.peek_3('defender'))):
            text += '    --\n'

        return text.rstrip()