コード例 #1
0
ファイル: qcontactdb.py プロジェクト: ellenbboe/BasicPython
 def __init__(self, session):
     self.session = session.Copy()
     dbname = SYSTEMSTR2STR(session.dbname)
     self.db = ContactDB(dbname)
     INFO('联系人数据库文件:%s', dbname)
コード例 #2
0
ファイル: qcontactdb.py プロジェクト: zhuzhuchen/qqbot
 def __init__(self, session):
     self.session = session.Copy()
     self.db = ContactDB(session.dbname)
コード例 #3
0
ファイル: qcontactdb.py プロジェクト: ellenbboe/BasicPython
class QContactDB(DBDisplayer):
    def __init__(self, session):
        self.session = session.Copy()
        dbname = SYSTEMSTR2STR(session.dbname)
        self.db = ContactDB(dbname)
        INFO('联系人数据库文件:%s', dbname)

    def List(self, tinfo, cinfo=None):
        result = self.db.List(tinfo, cinfo)
        if result is None:
            if not self.Update(tinfo):
                return None
            else:
                return self.db.List(tinfo, cinfo)
        else:
            return result

    def Update(self, tinfo):
        contacts = Fetch(self.session, tinfo)
        if contacts is None:
            return False
        else:
            rname = self.db.Update(tinfo, contacts)
            if rname is None:
                return False
            else:
                from qqbot import _bot
                _bot.onUpdate(tinfo)
                INFO('已获取并更新 %s', rname)
                return True

    def FirstFetch(self):
        q = collections.deque(['buddy', 'group', 'discuss'])
        while q:
            tinfo = q.popleft()
            if self.Update(tinfo) and tinfo in ('group', 'discuss'):
                cl = self.List(tinfo)
                if cl:
                    q.extend(cl)
            time.sleep(1.0)

    sysRegex = re.compile('^(' + ')|('.join([
        r'.+\(\d+\) 被管理员禁言.+', r'.+\(\d+\) 被管理员解除禁言',
        r'管理员开启了全员禁言,只有群主和管理员才能发言', r'管理员关闭了全员禁言',
        r'http://buluo\.qq\.com/mobile/detail\.html.+'
    ]) + ')$')

    def find(self, tinfo, uin, thisQQ, content):
        cl = self.List(tinfo, 'uin=' + uin)
        if cl is None:
            return None
        elif not cl:
            if getattr(tinfo, 'ctype', None) == 'group':
                if self.sysRegex.match(content):
                    return 'SYSTEM-MESSAGE'

            if not isinstance(tinfo, str):
                if getattr(self, 'selfUin', None) == uin:
                    cl2 = self.List(tinfo, 'uin=' + thisQQ)
                    return cl2[0] if cl2 else None

            if tinfo == 'buddy':
                if getattr(self, 'selfBuddyUin', None) == uin:
                    cl2 = self.List(tinfo, 'uin=' + thisQQ)
                    return cl2[0] if cl2 else None

            if self.Update(tinfo):
                cl = self.List(tinfo, 'uin=' + uin)
                if not cl:
                    if not isinstance(tinfo, str):
                        self.selfUin = uin
                        cl2 = self.List(tinfo, 'uin=' + thisQQ)
                        return cl2[0] if cl2 else None
                    elif tinfo == 'buddy':
                        self.selfBuddyUin = uin
                        cl2 = self.List(tinfo, 'uin=' + thisQQ)
                        return cl2[0] if cl2 else None
                    else:
                        return None
                else:
                    return cl[0]
            else:
                return None
        else:
            return cl[0]

    def FindSender(self, ctype, fromUin, membUin, thisQQ, content):
        contact = self.find(ctype, fromUin, thisQQ, content)
        member = None
        nameInGroup = None

        if contact is None:
            contact = self.db.NullContact(ctype, fromUin)
            if ctype in ('group', 'discuss'):
                member = self.db.NullContact(contact, membUin)
        elif ctype in ('group', 'discuss'):
            member = self.find(contact, membUin, thisQQ, content)
            if member is None:
                member = self.db.NullContact(contact, membUin)
            if ctype == 'group':
                cl = self.List(contact, 'uin=' + thisQQ)
                if cl:
                    nameInGroup = cl[0].name

        return contact, member, nameInGroup
コード例 #4
0
ファイル: qcontactdb.py プロジェクト: zhuzhuchen/qqbot
class QContactDB(DBDisplayer):
    def __init__(self, session):
        self.session = session.Copy()
        self.db = ContactDB(session.dbname)

    def List(self, tinfo, cinfo=None):
        result = self.db.List(tinfo, cinfo)
        if result is None:
            if not self.Update(tinfo):
                return None
            else:
                return self.db.List(tinfo, cinfo)
        else:
            return result

    def Update(self, tinfo):
        contacts = Fetch(self.session, tinfo)
        if contacts is None:
            return False
        else:
            rname = self.db.Update(tinfo, contacts)
            if rname is None:
                return False
            else:
                INFO('已获取并更新 %s', rname)
                return True

    def FirstFetch(self):
        q = collections.deque(['buddy', 'group', 'discuss'])
        while q:
            tinfo = q.popleft()
            if self.Update(tinfo) and tinfo in ('group', 'discuss'):
                cl = self.List(tinfo)
                if cl:
                    q.extend(cl)
            time.sleep(1.0)

    def find(self, tinfo, uin):
        cl = self.List(tinfo, 'uin=' + uin)
        if cl is None:
            return None
        elif not cl:
            self.Update(tinfo)
            cl = self.List(tinfo, 'uin=' + uin)
            if not cl:
                return None
            else:
                return cl[0]
        else:
            return cl[0]

    def FindSender(self, ctype, fromUin, membUin, thisQQ):
        contact = self.find(ctype, fromUin)
        member = None
        nameInGroup = None

        if contact is None:
            contact = self.db.NullContact(ctype, fromUin)
            if ctype in ('group', 'discuss'):
                member = self.db.NullContact(contact, membUin)
        elif ctype in ('group', 'discuss'):
            member = self.find(contact, membUin)
            if member is None:
                member = self.db.NullContact(contact, membUin)
            if ctype == 'group':
                cl = self.List(contact, 'uin=' + thisQQ)
                if cl:
                    nameInGroup = cl[0].name

        return contact, member, nameInGroup