用tornado写websocket聊天室

在tornado官方的websocket demo 基础上改写的，增加了session（用mongo来存储）；增加了用户列表；有用户注册功能/register 登录/login 登录图片验证；

1. #-*- coding: utf-8 -*-
   
2. 
   
3. import logging,random,uuid,string,time
   
4. import tornado.escape
   
5. import tornado.ioloop
   
6. import tornado.options
   
7. import tornado.web
   
8. import tornado.websocket
   
9. import os.path
   
10. 
    
11. from tornado.web import HTTPError
    
12. from session.auth import MongoAuthentication
    
13. from session.session import MongoSessions
    
14. from BaseHandler import BaseHandler
    
15. from tornado.options import define, options
    
16. from verify_code import make as make_verify_code
    
17. 
    
18. define("port", default=8888, help="run on the given port", type=int)
    
19. 
    
20. '''
    
21.     开发人：
    
22.         ROY (火炎神)
    
23.     时间：
    
24.         2014/6/14
    
25. 
    
26.     服务端的架构体系：
    
27.         db  : newjh (mongodb)
    
28.         b/s : websocket
    
29.         session : 基于mongodb
    
30.         verify_code : 校验码验证
    
31. '''
    
32. 
    
33. class Application(tornado.web.Application):
    
34.     def __init__(self):
    
35.         handlers = [
    
36.             (r"/register",RegisterHandler),
    
37.             (r"/login",LoginHandler),
    
38.             (r"/logout",LogoutHandler),
    
39.             (r"/", MainHandler),
    
40.             (r"/chatsocket", ChatSocketHandler),
    
41.             (r"/verifycode",VerifyCode),
    
42.         ]
    
43.         settings = dict(
    
44.             cookie_secret="__TODO:_GENERATE_YOUR_OWN_RANDOM_VALUE_HERE__",
    
45.             template_path=os.path.join(os.path.dirname(__file__), "templates"),
    
46.             static_path=os.path.join(os.path.dirname(__file__), "static"),
    
47.             xsrf_cookies=False,
    
48.         )
    
49. 
    
50.         self.sessions = MongoSessions("newjh", "sessions", timeout=10)
    
51.         self.auth = MongoAuthentication("newjh", "user")
    
52.         self.sessions.clear_all_sessions()
    
53.         tornado.web.Application.__init__(self, handlers, **settings)
    
54. 
    
55. class RegisterHandler(BaseHandler):
    
56.     def get(self):
    
57.         self.render("register.html")
    
58.     def post(self):
    
59.         username = self.get_argument('username',None)
    
60.         _username = self.get_argument('_username',None)
    
61.         pwd = self.get_argument('pwd',None)
    
62.         #pwd = md5(pwd).hexdigest()
    
63.         print 'argument %s'%self.request.arguments
    
64.         user = self.db.user.find_one({'uid':username})
    
65.         if user :
    
66.             self.write("帐号已经注册了")
    
67.         else:
    
68.             code = self.get_argument('code',None)
    
69.             cookiecode = self.get_secure_cookie('verify_code')
    
70.             print 'register code / server code',code,cookiecode
    
71.             if code != cookiecode : raise HTTPError(403,'验证不对')
    
72.             user = {
    
73.                 'uid':username,
    
74.                 'pwd':pwd,
    
75.                 'name':_username,
    
76.                 'gold':100000,
    
77.                 'gamegold':0,
    
78.                 'jewel':0,
    
79.                 'regtime':time.strftime('%Y-%m-%d',time.localtime(time.time())),
    
80.                 'bind':{'email':0,'mobile':0,'sec':0,'idcard':0},
    
81.                 'sign':{},
    
82.                 'signnum':{},
    
83.             }
    
84.             print 'user%s'%user
    
85.             res = self.application.auth.register(user)
    
86.             if not res : raise HTTPError(403)
    
87.             #user = db.user.find_one({'uid':username})
    
88.             res = self.begin_session(username,pwd)
    
89.             if not res : raise HTTPError(403)
    
90.             self.redirect('/login')
    
91. 
    
92. class LoginHandler(BaseHandler):
    
93.     def post(self):
    
94.         print self.request
    
95.         name = self.get_argument('name',None)
    
96.         pwd = self.get_argument('pwd',None)
    
97.         logging.info(('loging name pwd %s %s'%(name,pwd)))
    
98.         code = self.get_argument('code',None)
    
99.         cookiecode = self.get_secure_cookie('verify_code')
    
100.         print 'verify code cookie',cookiecode
     
101.         if code != cookiecode : raise HTTPError(403)
     
102.         res = self.begin_session(name, pwd)
     
103.         if not res : raise HTTPError(403)
     
104.         self.redirect('/')
     
105.     def get(self):
     
106.         self.render("login.html")
     
107. 
     
108. class LogoutHandler(BaseHandler):
     
109.     def get(self):
     
110.         self.post()
     
111. 
     
112.     def post(self):
     
113.         self.end_session()
     
114.         self.redirect('/login')
     
115. 
     
116. 
     
117. class VerifyCode(BaseHandler):
     
118.     def get(self):
     
119.         verify_code = ''.join(random.sample(string.ascii_lowercase\
     
120.                       +string.digits,4))
     
121.         self.set_secure_cookie("verify_code",verify_code)
     
122.         content = make_verify_code(verify_code)
     
123.         self.set_header('Content-Type','image/jpeg;charset=utf-8')
     
124.         self.write(content)
     
125. 
     
126. class MainHandler(BaseHandler):
     
127.     def get(self):
     
128. 
     
129.         if self.session :
     
130.             _user = self.session['data']
     
131.             self.render("index.html",
     
132.                         messages=ChatSocketHandler.cache,
     
133.                         userlist=ChatSocketHandler.userlist,
     
134.                         myuser = _user)
     
135.         else:
     
136.             self.redirect('/login')
     
137. 
     
138. class ChatSocketHandler(tornado.websocket.WebSocketHandler):
     
139.     waiters = set()
     
140.     cache = []
     
141.     cache_size = 200
     
142. 
     
143.     userlist = []
     
144.     user_size = 100
     
145.     user = None
     
146. 
     
147. 
     
148.     def allow_draft76(self):
     
149.         # for iOS 5.0 Safari
     
150.         return True
     
151. 
     
152.     def open(self):
     
153.         print 'open... is start '
     
154.         #self.on_message({"name":"roy","body":"welcome roy !","type":1})
     
155.         ChatSocketHandler.waiters.add(self)
     
156.         #self.write_message("ok!")
     
157. 
     
158.     def on_close(self):
     
159.         for i in ChatSocketHandler.userlist:
     
160.             if (self.user["content"] == i["content"]) :
     
161.                 ChatSocketHandler.userlist.remove(i)
     
162.         print self.user
     
163.         self.user["type"] = 3
     
164.         user = self.user
     
165.         self.user = None
     
166.         ChatSocketHandler.waiters.remove(self)
     
167.         ChatSocketHandler.send_updates(user)
     
168. 
     
169.     @classmethod
     
170.     def update_cache(cls, chat):
     
171.         cls.cache.append(chat)
     
172.         if len(cls.cache) > cls.cache_size:
     
173.             cls.cache = cls.cache[-cls.cache_size:]
     
174. 
     
175.     @classmethod
     
176.     def update_user(cls, user):
     
177.         for i in cls.userlist:
     
178.             if (user["content"] == i["content"]) : return
     
179.         #print cls.userlist
     
180.         #self.user = user
     
181.         cls.userlist.append(user)
     
182.         if len(cls.userlist) > cls.user_size:
     
183.             cls.userlist = cls.userlist[-cls.user_size:]
     
184. 
     
185.     @classmethod
     
186.     def send_updates(cls, chat):
     
187.         logging.info("sending message to %d waiters", len(cls.waiters))
     
188.         for waiter in cls.waiters:
     
189.             try:
     
190.                 waiter.write_message(chat)
     
191.             except:
     
192.                 logging.error("Error sending message", exc_info=True)
     
193. 
     
194. 
     
195.     def on_message(self, message):
     
196.         logging.info("got message %r", message)
     
197.         parsed = tornado.escape.json_decode(message)
     
198.         if(parsed["type"] == 1):
     
199.             chat = {
     
200.                 "id": str(uuid.uuid4()),
     
201.                 "body": parsed["body"],
     
202.                 "name": parsed["name"],
     
203.                 "type": parsed["type"]
     
204.                 }
     
205.             chat["html"] = tornado.escape.to_basestring(
     
206.                 self.render_string("message.html", message=chat))
     
207. 
     
208.             ChatSocketHandler.update_cache(chat)
     
209.             ChatSocketHandler.send_updates(chat)
     
210.         if(parsed["type"] == 2):
     
211.             print 'adduser start ...'
     
212.             user = {
     
213.                 "id": str(uuid.uuid4()),
     
214.                 "content": parsed["content"],
     
215.                 "type": parsed["type"]
     
216.                 }
     
217.             user["html"] = tornado.escape.to_basestring(
     
218.                 self.render_string("user.html", user=user))
     
219.             self.user = user
     
220.             ChatSocketHandler.update_user(user)
     
221.             ChatSocketHandler.send_updates(user)
     
222. 
     
223. 
     
224. 
     
225. def main():
     
226.     tornado.options.parse_command_line()
     
227.     app = Application()
     
228.     app.listen(options.port)
     
229.     tornado.ioloop.IOLoop.instance().start()
     
230. 
     
231. 
     
232. if __name__ == "__main__":
     
233.     main()

复制代码

5. [图片] QQ图片20140616195325.jpg   

6. [图片] QQ图片20140616195410.jpg