利用socketserver实现异步多线程服务端简单聊天功能

废话不多说，直接上代码

程序组成

数据访问层：model-------chat_log.py(用于保存聊天记录及展示)

　　　　　　　　　　　---user_info.py(用户登录信息表，验证通过才能聊天)

公共功能层：utility-------sqlhelper.py(用于处理查询及插入数据的操作)

　　　　　　　　　　　---conf.py(用于保存数据库连接字符串)

主程序:server.py--------Socket服务端

　　　client.py----------Socket客户端

server.py

 1 #!/usr/bin/env python
 2 #_*_ coding:utf-8 _*_
 3 import SocketServer
 4 import json
 5 import time
 6 from model.userinfo import UserInfo
 7 from model.chat_log import ChatLog
 8
 9 class  MyServer(SocketServer.BaseRequestHandler):
10
11     def setup(self):
12         pass
13
14     def handle(self):
15         container = {‘key‘:‘‘,‘data‘:‘‘}
16         container[‘data‘] = ‘ok...‘
17         conn = self.request
18         conn.sendall(json.dumps(container))
19
20         Flag =True
21         while Flag:
22             try:
23                 datetime = time.strftime(‘%Y-%m-%d %H:%M:%S‘)
24                 data = conn.recv(1024)
25                 print data
26                 recv_data = json.loads(data)
27                 if recv_data[‘data‘] == ‘exit‘:
28                     conn.close()
29                     break
30                 #key为空，表示用户没有登录或登录失败
31                 if not recv_data[‘key‘]:
32                     name,pwd = recv_data[‘data‘]
33                     re = UserInfo().ChechkLogin(name, pwd)
34                     #re = 1
35                     if re:
36                         recv_data[‘key‘] = re
37                         recv_data[‘data‘] = ‘约吗‘
38                     else:
39                         recv_data[‘data‘] = ‘failed‘
40                     conn.sendall(json.dumps(recv_data))
41                 #用户已经登陆
42                 else:
43                     if recv_data[‘data‘] == ‘list‘:
44                         ChatLog().ShowLog()
45                     elif recv_data[‘data‘].__contains__(‘yes‘):
46                         ChatLog().SaveLog(recv_data[‘data‘], datetime)
47                         recv_data[‘data‘] = ‘i am gay!‘
48                         ChatLog().SaveLog(recv_data[‘data‘], datetime)
49                     else:
50                         ChatLog().SaveLog(recv_data[‘data‘], datetime)
51                         recv_data[‘data‘] = ‘what?‘
52                         ChatLog().SaveLog(recv_data[‘data‘], datetime)
53                     conn.sendall(json.dumps(recv_data))
54             except Exception,e:
55                 print e
56                 Flag = False
57
58     def finish(self):
59         pass
60
61 if __name__ == ‘__main__‘:
62     server = SocketServer.ThreadingTCPServer((‘127.0.0.1‘,9999),MyServer)
63     server.serve_forever()

client.py

 1 #!/usr/bin/env python
 2 #_*_ coding:utf-8 _*_
 3 import socket
 4 import json
 5
 6 #创建客户端socket对象
 7 client = socket.socket()
 8 ip_port = (‘127.0.0.1‘,9999)
 9 #客户端连接服务端
10 client.connect(ip_port)
11 #客户端接收数据
12 while True:
13     data = client.recv(1024)
14     #print data
15     client_recv_data = json.loads(data)
16     print client_recv_data[‘data‘]
17     if not client_recv_data[‘key‘]:
18         name = raw_input(‘username:‘)
19         pwd = raw_input(‘pasword:‘)
20         client_recv_data[‘data‘] = (name,pwd)
21         client.send(json.dumps(client_recv_data))
22     else:
23         inp = raw_input(‘client:‘)
24         client_recv_data[‘data‘] = inp
25         client.send(json.dumps(client_recv_data))
26         if inp == ‘exit‘:
27             break

sqlhelper.py

 1 #!/usr/bin/env python
 2 #_*_ coding:utf-8 _*_
 3 import MySQLdb
 4 import conf
 5
 6 class MysqlHelper(object):
 7     def __init__(self):
 8         self.__conn_dict = conf.conn_dict
 9
10     def GetDict(self,sql,params):
11         conn = MySQLdb.connect(**self.__conn_dict)
12         cur = conn.cursor(cursorclass = MySQLdb.cursors.DictCursor)
13
14         #cur = conn.cursor()
15         reCount = cur.execute(sql,params)
16         nRet = cur.fetchall()
17
18         cur.close()
19         conn.close()
20         return nRet
21     def GetOne(self,sql,params):
22         conn = MySQLdb.connect(**self.__conn_dict)
23         cur = conn.cursor(cursorclass = MySQLdb.cursors.DictCursor)
24
25         #cur = conn.cursor()
26         reCount = cur.execute(sql,params)
27         nRet = cur.fetchone()
28
29         cur.close()
30         conn.close()
31         return nRet
32
33     def InsertOne(self,sql,params):
34         conn = MySQLdb.connect(**self.__conn_dict)
35         cur = conn.cursor(cursorclass = MySQLdb.cursors.DictCursor)
36
37         reCount = cur.execute(sql,params)
38         conn.commit()
39
40         cur.close()
41         conn.close()

conf.py

1 #!/usr/bin/env python
2 #_*_ coding:utf-8 _*_
3 conn_dict = dict(host=‘127.0.0.1‘,user=‘root‘,passwd=‘123456‘,db=‘08day05‘)

chat_log.py

 1 #!/usr/bin/env python
 2 #_*_ coding:utf-8 _*_
 3 from utility.sqlhelper import MysqlHelper
 4 class ChatLog():
 5     def __init__(self):
 6         self.__helper = MysqlHelper()
 7     def SaveLog(self,msg,insert_day):
 8         sql = ‘insert into chat_log(msg,time) values(%s,%s)‘
 9         params = (msg,insert_day)
10         self.__helper.InsertOne(sql, params)
11     def ShowLog(self):
12         sql = ‘select * from chat_log‘
13         params = ()
14         result = str(self.__helper.GetDict(sql,params))
15         return result
16         #print result
17         ‘‘‘
18         for msg_dict in result:
19             for item in msg_dict:
20                 print item+‘:‘+str(msg_dict[item])
21         ‘‘‘
22
23 #log1 = ChatLog()
24 #log1.ShowLog()

userinfo.py

 1 #!/usr/bin/env python
 2 #_*_ coding:utf-8 _*_
 3 from utility.sqlhelper import MysqlHelper
 4
 5 class UserInfo(object):
 6     def __init__(self):
 7         self.__helper = MysqlHelper()
 8
 9     def ChechkLogin(self,username,password):
10         sql = ‘select id from userinfo where name = %s and password = %s‘
11         params = (username,password)
12         result = self.__helper.GetOne(sql, params)
13         if result:
14             return result
15         else:
16             return False
17