python 点对点聊天_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)！！！