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

#coding:utf-8'''file:client.py.pydate:2017/9/11 11:01author:lockeyemail:lockey@123.complatform:win7.x86_64 pycharm python3desc:p2p communication clientside'''from socket import *import threading,sys,json,re#引入json模块主要是为了数据的封装传输，re的话是做一些合法性的验证HOST = '192.168.1.7'PORT=8022BUFSIZE = 1024 ##缓冲区大小 1KADDR = (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 Falsedef 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 Falsedef addGroup():#群组添加 groupname = input('Please input group name: ') if not re.findall(myre, groupname):  print('group name illegal!')  return None return groupnamedef 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.pydate:2017/9/11 14:43author:lockeyemail:lockey@123.complatform:win7.x86_64 pycharm python3desc:p2p communication serverside'''import socketserver,json,timeimport subprocessconnLst = []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 = conObjclass 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'))    continueif __name__ == '__main__': server = socketserver.ThreadingTCPServer(('192.168.1.7',8022),MyServer) print('waiting for connection...') server.serve_forever()