python实现简单聊天应用 python群聊和点对点均实现

后续代码更新和功能添加会提交到 个人github主页 ，有兴趣可以一起来完善！

如果只是拿过去运行看结果，请注意平台相关性以及python版本号，本示例开发运行平台为win7x86_64 pycharm community，python版本号为3.5!!!

TALK IS CHEAP, SHOW YOU MY CODE:

客户端

​
    #coding:utf-8
    '''
    file:client.py.py
    date:2017/9/11 11:01
    author:lockey
    email:lockey@123.com
    platform:win7.x86_64 pycharm python3
    desc:p2p communication clientside
    '''
    from socket import *
    import threading,sys,json,re
    #引入json模块主要是为了数据的封装传输，re的话是做一些合法性的验证
    HOST = '192.168.1.7'
    PORT=8022
    BUFSIZE = 1024 ##缓冲区大小 1K
    ADDR = (HOST,PORT)
    myre = r"^[_a-zA-Z]\w{0,}"
    tcpCliSock = socket(AF_INET,SOCK_STREAM)
    #创建一个socket连接
    userAccount = None
    #用户登录标志，也用来记录登录的用户名称
    def register():
    #用户注册函数
     print("""
     Glad to have you a member of us!
     """)
     accout = input('Please input your account: ')
     if not re.findall(myre, accout):
      print('Account illegal!')
      return None
     password1 = input('Please input your password: ')
     password2 = input('Please confirm your password: ')
     if not (password1 and password1 == password2):
      print('Password not illegal!')
      return None
     global userAccount
     userAccount = accout
     regInfo = [accout,password1,'register']
     datastr = json.dumps(regInfo)
     tcpCliSock.send(datastr.encode('utf-8'))
     data = tcpCliSock.recv(BUFSIZE)
     data = data.decode('utf-8')
     if data == '0':
      print('Success to register!')
      return True
     elif data == '1':
      print('Failed to register, account existed!')
      return False
     else:
      print('Failed for exceptions!')
      return False
    
    def login():
    #用户登录函数
     print("""
     Welcome to login in!
     """)
     accout = input('Account: ')
     if not re.findall(myre, accout):
      print('Account illegal!')
      return None
     password = input('Password: ')
     if not password:
      print('Password illegal!')
      return None
     global userAccount
     userAccount = accout
     loginInfo = [accout, password,'login']
     datastr = json.dumps(loginInfo)
     tcpCliSock.send(datastr.encode('utf-8'))
     data = tcpCliSock.recv(BUFSIZE)
     if data == '0':
      print('Success to login!')
      return True
     else:
      print('Failed to login in(user not exist or username not match the password)!')
      return False
    def addGroup():
    #群组添加
     groupname = input('Please input group name: ')
     if not re.findall(myre, groupname):
      print('group name illegal!')
      return None
     return groupname
    
    def chat(target):
    #进入聊天（群聊和点对点聊天可以选择）
     while True:
      print('{} -> {}: '.format(userAccount,target))
      msg = input()
      if len(msg) > 0 and not msg in 'qQ':
       if 'group' in target:
        optype = 'cg'
       else:
        optype = 'cp'
    
       dataObj = {'type': optype, 'to': target, 'msg': msg, 'froms': userAccount}
       datastr = json.dumps(dataObj)
       tcpCliSock.send(datastr.encode('utf-8'))
       continue
      elif msg in 'qQ':
       break
      else:
       print('Send data illegal!')
    class inputdata(threading.Thread):
    #用户输入选择然后执行不同的功能程序
     def run(self):
      menu = """
          (CP): Chat with individual
          (CG): Chat with group member
          (AG): Add a group
          (EG): Enter a group
          (H): For help menu
          (Q): Quit the system
          """
      print(menu)
      while True:
       operation = input('Please input your operation("h" for help): ')
       if operation in 'cPCPCpcp':
       #进入个人聊天
        target = input('Who would you like to chat with: ')
        chat(target)
        continue
    
       if operation in 'cgCGCgcG':
       #进入群聊
        target = input('Which group would you like to chat with: ')
        chat('group'+target)
        continue
       if operation in 'agAGAgaG':
       #添加群组
        groupName = addGroup()
        if groupName:
         dataObj = {'type': 'ag', 'groupName': groupName}
         dataObj = json.dumps(dataObj)
         tcpCliSock.send(dataObj.encode('utf-8'))
        continue
    
       if operation in 'egEGEgeG':
       #入群
        groupname = input('Please input group name fro entering: ')
        if not re.findall(myre, groupname):
         print('group name illegal!')
         return None
        dataObj = {'type': 'eg', 'groupName': 'group'+groupname}
        dataObj = json.dumps(dataObj)
        tcpCliSock.send(dataObj.encode('utf-8'))
        continue
       if operation in 'hH':
        print(menu)
        continue
    
       if operation in 'qQ':
        sys.exit(1)
       else:
        print('No such operation!')
    
    class getdata(threading.Thread):
    #接收数据线程
     def run(self):
      while True:
       data = tcpCliSock.recv(BUFSIZE).decode('utf-8')
       if data == '-1':
        print('can not connect to target!')
        continue
       if data == 'ag0':
        print('Group added!')
        continue
    
       if data == 'eg0':
        print('Entered group!')
        continue
    
       if data == 'eg1':
        print('Failed to enter group!')
        continue
    
       dataObj = json.loads(data)
       if dataObj['type'] == 'cg':
       #群组消息的格式定义
        print('{}(from {})-> : {}'.format(dataObj['froms'], dataObj['to'], dataObj['msg']))
       else:
       #个人消息的格式定义
        print('{} ->{} : {}'.format(dataObj['froms'], userAccount, dataObj['msg']))
    
    
    def main():
    
      try:
       tcpCliSock.connect(ADDR)
       print('Connected with server')
       while True:
        loginorReg = input('(l)ogin or (r)egister a new account: ')
        if loginorReg in 'lL':
         log = login()
         if log:
          break
        if loginorReg in 'rR':
         reg = register()
         if reg:
          break
    
       myinputd = inputdata()
       mygetdata = getdata()
       myinputd.start()
       mygetdata.start()
       myinputd.join()
       mygetdata.join()
    
      except Exception:
       print('error')
       tcpCliSock.close()
       sys.exit()
    
    
    if __name__ == '__main__':
     main()
    
    

服务端

​
    #coding:utf-8
    '''
    file:server.py
    date:2017/9/11 14:43
    author:lockey
    email:lockey@123.com
    platform:win7.x86_64 pycharm python3
    desc:p2p communication serverside
    '''
    import socketserver,json,time
    import subprocess
    
    connLst = []
    groupLst = []
    ## 代号 地址和端口 连接对象
    #optype = {'ag':'group adding','cp':'chat with individual','cg':'chat with group'}
    class Connector(object): ##连接对象类
     def __init__(self,account,password,addrPort,conObj):
      self.account = account
      self.password = password
      self.addrPort = addrPort
      self.conObj = conObj
    
    class Group(object):#群组类
     def __init__(self,groupname,groupOwner):
      self.groupId = 'group'+str(len(groupLst)+1)
      self.groupName = 'group'+groupname
      self.groupOwner = groupOwner
      self.createTime = time.time()
      self.members=[groupOwner]
    
    class MyServer(socketserver.BaseRequestHandler):
    
     def handle(self):
      print("got connection from",self.client_address)
      userIn = False
      global connLst
      global groupLst
      while not userIn:
       conn = self.request
       data = conn.recv(1024)
       if not data:
        continue
       dataobj = json.loads(data.decode('utf-8'))
       #如果连接客户端发送过来的信息格式是一个列表且注册标识为False时进行用户注册或者登陆
       ret = '0'
       if type(dataobj) == list and not userIn:
        account = dataobj[0]
        password = dataobj[1]
        optype = dataobj[2]
        existuser = False
        if len(connLst) > 0:
         for obj in connLst:
          if obj.account == account:
           existuser = True
           if obj.password == password:
            userIn = True
            print('{} has logged in system({})'.format(account,self.client_address))
            break
        if optype == 'login' and (not userIn or not existuser):
         ret = '1'
         print('{} failed to logged in system({})'.format(account, self.client_address))
        else:
         if existuser:
          ret = '1'
          print('{} failed to register({}),account existed!'.format(account, self.client_address))
         else:
          try:
           conObj = Connector(account,password,self.client_address,self.request)
           connLst.append(conObj)
           print('{} has registered to system({})'.format(account,self.client_address))
           userIn = True
          except:
           print('%s failed to register for exception!'%account)
           ret = '99'
       conn.sendall(ret.encode('utf-8'))
       if ret == '0':
        break
    
      while True:
      #除登陆注册之外的请求的监听
       conn = self.request
       data = conn.recv(1024)
       if not data:
        continue
       print(data)
       dataobj = data.decode('utf-8')
       dataobj = json.loads(dataobj)
       if dataobj['type'] == 'ag' and userIn:
       #如果判断用户操作请求类型为添加群组则进行以下操作
        groupName = dataobj['groupName']
        groupObj = Group(groupName,self.request)
        groupLst.append(groupObj)
        conn.sendall('ag0'.encode('utf-8'))
        print('%s added'%groupName)
        continue
    
       if dataobj['type'] == 'eg' and userIn:
       #入群操作
        groupName = dataobj['groupName']
        ret = 'eg1'
        for group in groupLst:
         if groupName == group.groupName:
          group.members.append(self.request)
          print('{} added into {}'.format(self.client_address,groupName))
          ret = 'eg0'
          break
        conn.sendall(ret.encode('utf-8'))
        continue
    
       #客户端将数据发给服务器端然后由服务器转发给目标客户端
       print('connLst',connLst)
       print('grouplst',groupLst)
       if len(connLst) > 1:
        sendok = False
        if dataobj['type'] == 'cg':
        #群内广播（除发消息的人）
         print('group',data)
         for obj in groupLst:
          if obj.groupName == dataobj['to']:
           for user in obj.members:
            if user != self.request:
             user.sendall(data)
        else:
        #个人信息发送
         for obj in connLst:
          if dataobj['to'] == obj.account:
           obj.conObj.sendall(data)
           sendok = True
         if sendok == False:
          print('no target valid!')
       else:
        conn.sendall('-1'.encode('utf-8'))
        continue
    
    if __name__ == '__main__':
     server = socketserver.ThreadingTCPServer(('192.168.1.7',8022),MyServer)
     print('waiting for connection...')
     server.serve_forever()
    
    

运行结果示例

服务端（记录着各客户端的操作）：

客户端1：

有注册、建群、群聊、点对点聊天

客户端2：

客户端3：

要拷贝代码运行的话请注意平台（win7.x86_64）和python版本号(python3.5)！！！