Python Multi-person chat tool (multi-threaded)

Source: Internet
Author: User

Program implementation:1, single or multiple clients using Telnet to log on to the server (remote) for session2, service-side implementation login, registration, Exit Function3, messages sent by the client are broadcast to other user interfaces that have been logged in4, after connecting to the server, you can execute the corresponding program instruction code: https:coding.net/u/wangxiaoqiangs/p/pycode/git/tree/master/socket/gservergserver.py#!/usr/bin/env python#Coding:utf-8#Author:xiao GuaishouImportSocket fromDbImportDB fromThreadingImportCurrentThread, ThreadclassHandlerthread (object): Queue= []#sockect Queuedb =DB ()def __init__(self, sock): Self.sock=sockdefrecv (self): Data= Self.sock.recv (1024x768). Strip ()#If you use while to receive data, it causes the user to have to hit the enter key more than once        returnDatadefSend (self, data): Self.sock.sendall ('\n[system]:%s\n'%data)#broadcast messages to the queue    defBroadcast (self, user, data): forSockinchSelf.queue:sock.sendall ('\n[%s]:%s\n'%(user, data))#Close Client Connections    defStop (self): Self.send ('byebye!') Self.sock.close () self.queue.remove (Self.sock)#after you close the connection, remember to delete it from the queue    #Program Entry    defhandler (self): funcdict= {                    'Login': Self.login,'Register': Self.register}Try: Thname=CurrentThread (). GetName ()Print('[%s] Got connection from%s'% (Thname, Self.sock.getpeername ()))#all print data in the program will be replaced with the Loging module .Self.send ('Please select function: Login/register/exit') Data=self.recv ()ifdata = ='Exit': Self.stop ()#In fact, this should be done separately using self.sock.close () to close the connection, because there is no such connection in the queue, but with the following capture there is no problem ^_^            elifDatainchfuncdict:returnfuncdict.get (data) ()Else: Self.handler ()except:#If you don't capture it here, you won't be able to disconnect the client properly.            Pass    #Handling User Login    defLogin (self): Self.send ('Login ... Please enter username password, format: User Password, enter Server: Execute program Instruction! ') User_data=self.recv ()#Program Internal Directives        ifUser_data = ='Server:': Self.send ('\n\tserver:use reged\t Switch to the registration page \n\tserver:exit\t\t exit the system') User_data=self.recv ()ifUser_data = ='Server:use reged': Self.register ()elifUser_data = ='Server:exit': Self.stop ()Else: Self.send ('Input Error ...') DataList=User_data.split ()#determine user input, format is correct        ifLen (datalist) = = 2: User=datalist[0] Password= Datalist[1] Db_data= Self.db.get_data ()or {}            ifUserinchDb_data andPassword = =db_data.get (user): Self.queue.append (Self.sock)#users with permission to log on to the system, the connection is added to the queueSelf.send ('Welcome to the chat room, enter the Server: Get the Function Method! ') Self.broadcast ('System','[%s] Join the chat room! '%user) Self.chat_room (user)Else: Self.send ('User name, password error! ') Self.login () Self.login ()defRegister (self): Self.send ('Register ... Please enter username password, format: User Password, enter Server: Execute program Instruction! ') User_data=self.recv ()ifUser_data = ='Server:': Self.send ('\n\tserver:use login\t Switch to the registration page \n\tserver:exit\t\t exit the system') User_data=self.recv ()ifUser_data = ='Server:login': Self.login ()elifUser_data = ='Server:exit': Self.stop ()Else: Self.send ('Input Error ...') DataList=User_data.split ()ifLen (datalist) = = 2: User=datalist[0] Password= Datalist[1] Db_data= Self.db.get_data ()or {}            ifUserinchDb_data:self.send ('the user name has been registered! ') Self.register ()Else: Db_data[user]=password Self.db.put_data (db_data) self.queue.append (self.sock) Self.bro Adcast ('System','new User [%s] join the chat room! '%user) Self.chat_room (user) Self.register ()defchat_room (self, user): User_data=self.recv ()ifUser_data = ='Server:': Self.send ('\n\tserver:logout\t exit chat room') User_data=self.recv ()ifUser_data = ='Server:logout': Self.stop ()return #If you do not add a return here, the Server: command executed by the client will also be broadcast            Else: Self.send ('Input Error ...') self.chat_room (user)Else: Self.broadcast (User, User_data) self.chat_room (user)#create a thread for each connectiondefStartthread (sock, addr):Print('Received New Client connection.%s:%s'% (Addr[0], addr[1]) th=Handlerthread (sock) T= Thread (target=th.handler) T.setdaemon (True) T.start ()#Start the servicedefServer (): s=Socket.socket (socket.af_inet, socket. SOCK_STREAM) s.setsockopt (socket. Sol_socket, SOCKET. SO_REUSEADDR,1) S.bind ('0.0.0.0', 17170)) S.listen (1)     whileTrue:Try: sock, addr=s.accept ()exceptKeyboardinterrupt:exit ('\nbyebye!') Startthread (sock, addr) s.close ()if __name__=='__main__': Server () db.py#Coding:utf-8ImportJSON#create a class instead of a databaseclassDB (object):def __init__(Self, path='storage.db'): Self.path=PathdefGet_data (Self, data=None):Try: With open (Self.path) as F:data=json.load (f)exceptIOError as E:returnData#when data is first taken, the default value of None is returned because the file does not exist or no data        finally:            returnDatadefPut_data (self, data): With open (Self.path,'W') as F:json.dump (data, F)

Python Multi-person chat tool (multi-threaded)

Related Article

Contact Us

The content source of this page is from Internet, which doesn't represent Alibaba Cloud's opinion; products and services mentioned on that page don't have any relationship with Alibaba Cloud. If the content of the page makes you feel confusing, please write us an email, we will handle the problem within 5 days after receiving your email.

If you find any instances of plagiarism from the community, please send an email to: info-contact@alibabacloud.com and provide relevant evidence. A staff member will contact you within 5 working days.

A Free Trial That Lets You Build Big!

Start building with 50+ products and up to 12 months usage for Elastic Compute Service

  • Sales Support

    1 on 1 presale consultation

  • After-Sales Support

    24/7 Technical Support 6 Free Tickets per Quarter Faster Response

  • Alibaba Cloud offers highly flexible support services tailored to meet your exact needs.