qq can send and receive msg

src/qq/UI.py

@@ -0,0 +1,200 @@
+'''
+Created on Aug 2, 2015
+
+@author: ivan
+'''
+import urwid
+from collections import deque
+from threading import Thread
+import threading
+from time import gmtime, strftime
+
+class UnknownCommand(Exception):
+    def __init__(self,cmd):
+        Exception.__init__(self,'Uknown command: %s'%cmd)
+
+class Command(object):
+    """ Base class to manage commands in commander
+similar to cmd.Cmd in standard library
+just extend with do_something  method to handle your commands"""
+
+    def __init__(self,quit_commands=['q','quit','exit'], help_commands=['help','?', 'h']):
+        self._quit_cmd=quit_commands
+        self._help_cmd=help_commands
+
+    def __call__(self,line):
+        tokens=line.split()
+        cmd=tokens[0].lower()
+        args=tokens[1:]
+        if cmd in self._quit_cmd:
+            return Commander.Exit
+        elif cmd in self._help_cmd:
+            return self.help(args[0] if args else None)
+        elif hasattr(self, 'do_'+cmd):
+            return getattr(self, 'do_'+cmd)(*args)
+        else:
+            raise UnknownCommand(cmd)
+
+    def help(self,cmd=None):
+        def std_help():
+            qc='|'.join(self._quit_cmd)
+            hc ='|'.join(self._help_cmd)
+            res='Type [%s] command_name to get more help about particular command\n' % hc
+            res+='Type [%s] to quit program\n' % qc
+            cl=[name[3:] for name in dir(self) if name.startswith('do_') and len(name)>3]
+            res += 'Available commands: %s' %(' '.join(sorted(cl)))
+            return res
+        if not cmd:
+            return std_help()
+        else:
+            try:
+                fn=getattr(self,'do_'+cmd)
+                doc=fn.__doc__
+                return doc or 'No documentation available for %s'%cmd
+            except AttributeError:
+                return std_help()
+
+class FocusMixin(object):
+    def mouse_event(self, size, event, button, x, y, focus):
+        if focus and hasattr(self, '_got_focus') and self._got_focus:
+            self._got_focus()
+        return super(FocusMixin,self).mouse_event(size, event, button, x, y, focus)
+
+class ListView(FocusMixin, urwid.ListBox):
+    def __init__(self, model, got_focus, max_size=None):
+        urwid.ListBox.__init__(self,model)
+        self._got_focus=got_focus
+        self.max_size=max_size
+        self._lock=threading.Lock()
+
+    def add(self,line, align='left'):
+        with self._lock:
+            was_on_end=self.get_focus()[1] == len(self.body)-1
+            if self.max_size and len(self.body)>self.max_size:
+                del self.body[0]
+
+            self.body.append(urwid.Text(line, align=align))
+            last=len(self.body)-1
+            if was_on_end:
+                self.set_focus(last,'above')
+
+
+
+class Input(FocusMixin, urwid.Edit):
+    signals=['line_entered']
+    def __init__(self, got_focus=None):
+        urwid.Edit.__init__(self)
+        self.history=deque(maxlen=1000)
+        self._history_index=-1
+        self._got_focus=got_focus
+
+    def keypress(self, size, key):
+        if key=='enter':
+            line=self.edit_text.strip()
+            if line:
+                urwid.emit_signal(self,'line_entered', line)
+                self.history.append(line)
+            self._history_index=len(self.history)
+            self.edit_text=u''
+        if key=='up':
+
+            self._history_index-=1
+            if self._history_index< 0:
+                self._history_index= 0
+            else:
+                self.edit_text=self.history[self._history_index]
+        if key=='down':
+            self._history_index+=1
+            if self._history_index>=len(self.history):
+                self._history_index=len(self.history)
+                self.edit_text=u''
+            else:
+                self.edit_text=self.history[self._history_index]
+        else:
+            urwid.Edit.keypress(self, size, key)
+
+
+
+class Commander(urwid.Frame):
+    """ Simple terminal UI with command input on bottom line and display frame above
+similar to chat client etc.
+Initialize with your Command instance to execute commands
+and the start main loop Commander.loop().
+You can also asynchronously output messages with Commander.output('message') """
+
+    class Exit(object):
+        pass
+
+    PALLETE=[('reversed', urwid.BLACK, urwid.LIGHT_GRAY),
+              ('normal', urwid.LIGHT_GRAY, urwid.BLACK),
+              ('error', urwid.LIGHT_RED, urwid.BLACK),
+              ('green', urwid.DARK_GREEN, urwid.BLACK),
+              ('blue', urwid.LIGHT_BLUE, urwid.BLACK),
+              ('magenta', urwid.DARK_MAGENTA, urwid.BLACK), ]
+
+
+    def __init__(self, title, command_caption='Command:  (Tab to switch focus to upper frame, where you can scroll text)', cmd_cb=None, max_size=1000):
+        self.header=urwid.Text(title)
+        self.model=urwid.SimpleListWalker([])
+        self.body=ListView(self.model, lambda: self._update_focus(False), max_size=max_size )
+        self.input=Input(lambda: self._update_focus(True))
+        foot=urwid.Pile([urwid.AttrMap(urwid.Text(command_caption), 'reversed'),
+                        urwid.AttrMap(self.input,'normal')])
+        urwid.Frame.__init__(self,
+                             urwid.AttrWrap(self.body, 'normal'),
+                             urwid.AttrWrap(self.header, 'reversed'),
+                             foot)
+        self.set_focus_path(['footer',1])
+        self._focus=True
+        urwid.connect_signal(self.input,'line_entered',self.on_line_entered)
+        self._cmd=cmd_cb
+        self._output_styles=[s[0] for s in self.PALLETE]
+        self.eloop=None
+
+    def loop(self, handle_mouse=False):
+        self.eloop=urwid.MainLoop(self, self.PALLETE, handle_mouse=handle_mouse)
+        self._eloop_thread=threading.current_thread()
+        self.eloop.run()
+
+    def on_line_entered(self,line):
+        if self._cmd:
+            try:
+                res = self._cmd(line)
+            except Exception,e:
+                self.output('Error: %s'%e, 'error')
+                return
+            if res==Commander.Exit:
+                raise urwid.ExitMainLoop()
+            elif res:
+                # self.output(str(res))
+                self.output(res)
+        else:
+            if line in ('q','quit','exit'):
+                raise urwid.ExitMainLoop()
+            else:
+                self.output(line)
+
+    def output(self, line, style=None, align='left'):
+        if style and style in self._output_styles:
+                line=(style,line)
+        self.body.add(line, align=align)
+        #since output could be called asynchronously form other threads we need to refresh screen in these cases
+        if self.eloop and self._eloop_thread != threading.current_thread():
+            self.eloop.draw_screen()
+
+
+    def _update_focus(self, focus):
+        self._focus=focus
+
+    def switch_focus(self):
+        if self._focus:
+            self.set_focus('body')
+            self._focus=False
+        else:
+            self.set_focus_path(['footer',1])
+            self._focus=True
+
+    def keypress(self, size, key):
+        if key=='tab':
+            self.switch_focus()
+        return urwid.Frame.keypress(self, size, key)

src/qq/main.py

@@ -0,0 +1,87 @@
+#!/usr/bin/env python
+# coding=utf-8
+#
+# Author: Archer Reilly
+# File: main.py
+# Desc: QQ 命令行版本
+# Date: 20/Aug/2016
+from qqbot import QQBot
+from UI import *
+
+def GetNickName(Contacts, number):
+    for contact in Contacts:
+        if contact[0] == number:
+            return contact[1]
+
+    return None
+
+def GetNumber(Contacts, nickname):
+    for contact in Contacts:
+        if contact[1].encode == nickname:
+            return contact[0]
+
+    return None
+
+if __name__=='__main__':
+    # start and login qq first
+    bot = QQBot()
+    bot.Login()
+    Buddies = bot.buddy
+    Groupes = bot.group
+    Discusses = bot.discuss
+    Contacts = []
+    for buddy in Buddies:
+        Contacts.append(buddy)
+    for group in Groupes:
+        Contacts.append(group)
+    for discuss in Discusses:
+        Contacts.append(discuss)
+
+    class TestCmd(Command):
+        def do_echo(self, *args):
+            '''echo - Just echos all arguments'''
+            return ' '.join(args)
+        def do_raise(self, *args):
+            raise Exception('Some Error')
+
+        def do_contact(self, *args):
+            if args[0] == 'buddy':
+                return bot.buddyStr.replace('?', '')
+            elif args[0] == 'group':
+                return bot.groupStr.replace('?', '')
+            elif args[0] == 'discuss':
+                return bot.discussStr.replace('?', '')
+
+        def do_number(self, *args):
+            nickname = args[0]
+            number = GetNumber(Contacts, args[0].encode('utf-8'))
+            return number
+
+        def do_sendmsg(self, *args):
+            msgType = args[0]
+            number = int(args[1])
+            nickname = GetNickName(Contacts, number)
+
+            bot.send(msgType, number, ' '.join(args[2:]))
+            return '->' + nickname.decode('utf-8') + '(' + str(number) + '): ' + ' '.join(args[2:])
+
+    c=Commander('Drogo', cmd_cb=TestCmd())
+
+    #Test asynch output -  e.g. comming from different thread
+    import time
+    def run():
+        while True:
+            time.sleep(3)
+            msg = bot.poll()
+            if msg[0] == '':
+                continue
+
+            nickname = GetNickName(Contacts, msg[1])
+            c.output(nickname + '(' + str(msg[1]) + '): ' + msg[3], 'green')
+
+    t=Thread(target=run)
+    t.daemon=True
+    t.start()
+
+    #start main loop
+    c.loop()

src/qq/qq.py

@@ -0,0 +1,6 @@
+from qqbot import QQBot
+
+bot = QQBot()
+bot.Login()
+
+bot.send('group', 3999311269, 'ha?')