我打算將WebQQ單獨出來運行, 一開始直接拷貝了pyxmpp2的mainloop, 但是跑起來問題多多, 所以我又研究了利用Tornado進行網絡編程(這里), 所以我放棄了Pyxmpp2的mainloop,使用Tornado進行重寫

首先放出項目代碼

引子

WebQQ協議是一套基于HTTP的QQ協議, 而用Python的urllib2庫進行請求太慢, 因為HTTP本身就使用socket請求, 所以改用多路復用I/O模型, 而Tornado簡單高效, 看過代碼后可以輕松上手.平臺兼容性很好, 所以選擇Tornado作為網絡框架.

原理

首先實現了一個 HTTPStream類, 其主要接口是add_request方法, 它接受一個必選參數:request 是一個 urllib2.Request的實例, 和一個可選參數:readback是一個接受一個urllib2.urlopen(request)返回的Response參數的讀取函數, 代碼如下:

class HTTPStream(object):

# 省略若干代碼

def add_request(self, request, readback = None):

if not isinstance(request, urllib2.Request):

raise ValueError, "Not a invaid requset"

# 此處易觸發timeout異常, 省略處理異常代碼

sock, data = self.http_sock.make_http_sock_data(request)

fd = sock.fileno()

self.fd_map[fd] = sock

self.fd_request_map[fd] = request

callback = partial(self._handle_events, request, data, readback)

self.ioloop.add_handler(fd, callback, IOLoop.WRITE)

HTTPStream.add_request將urllib2.Request的實例解析出一個socket和一個用于socket發送的數據.前面文章介紹過了, tornado.ioloop.IOLoop.add_handler用于將注冊socket, 其需要三個參數: socket的文件描述符, 接受文件描述符和事件參數的回調, 和注冊的事件.

我們用到的回調是HTTPStream._handle_events:

class HTTPStream(object):

# 省略若干代碼

def _handle_events(self, request, data, readback, fd, event):

""" 用于處理Tornado事件

Arguments:

`request` - urllib.Request

`data` - socket要寫入的數據

`readback` - 讀取函數

以上參數應當使用partial封裝然后將此方法作為IOLoop.add_handler的callback

`fd` - IOLoop傳遞 文件描述符

`event` - IOLoop傳遞 tornado

"""

s = self.fd_map[fd]

if event & IOLoop.READ:

# 省略錯誤處理

resp = self.http_sock.make_response(s, request)

args = readback(resp)

s.setblocking(False)

if args and len(args) == 3:

t = threading.Thread(target = self.add_delay_request, args = args)

t.setDaemon(True)

t.start()

if args and len(args) == 2:

self.add_request(*args)

self.ioloop.remove_handler(fd)

if event & IOLoop.WRITE:

s.sendall(data)

if readback:

self.ioloop.update_handler(fd, IOLoop.READ)

else:

self.ioloop.remove_handler(fd)

if event & IOLoop.ERROR:

pass

它接受的參數上面注釋寫的很清楚, 不做解釋, 所以將此方法通過functools.partial封裝做為callback傳遞給tornado.ioloop.IOLoop.add_handler, 并注冊為寫事件, 以便發送HTTP請求.

HTTPStream._handle_events用于處理事件, 當事件為寫時就發送HTTP請求(根據urllib2.Request生成的用于發送的數據), 并判斷是否有讀取函數, 有則注冊讀事件, 當事件為讀時就從socket中構建一個Response并傳遞給讀取函數, 讀取函數會返回3個值, 分別為: 下一個請求, 請求的讀取函數(可為None, 為None則只請求不讀取), 下一個請求的延遲(多長事件后添加此請求, 可選, 單位為秒)

依據讀取函數返回的三個值來確定下一個請求, 并完成一系列的請求. 更加完整的代碼請參見文章開頭給出的項目代碼

HTTPStream.http_sock.make_response執行時會將socket設為阻塞, 因為不設置阻塞會出現httplib.BadStatusLine異常.讀取函數執行完畢,重新將socket設置為非阻塞, 并移除此socket(雖然做了這樣的處理但是QQ連接時間稍長還是會觸發httplib.BadStatusLine異常)

總結

以上是生活随笔為你收集整理的python qq机器人框架有哪些_使用Python的Tornado框架实现一个简单的WebQQ机器人的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。