python實現的簡單點對點（p2p）聊天

大囚長發表於2018-06-26

點對點聊天首先是基於多執行緒的網路程式設計，其次就是將每一個連線都儲存為一個具有獨一屬性的物件並新增到連線列表中，對於每一個連線物件傳送過來的資訊必須要包含主要的三項內容（from,to,messages），這樣當資訊傳送到伺服器之後伺服器根據to的連線物件遍歷連線列表找到目標物件將資訊傳送給目標，目標拿到資訊後就知道是誰發過來的，然後根據id號碼進行回覆。。此實現將會繼續完善，後續新加功能將會在我個人github主頁展現

伺服器端實現：

#coding:utf-8
'''
file:server.py
date:2017/9/10 12:43
author:lockey
email:lockey@123.com
platform:win7.x86_64 pycharm python3
desc:p2p communication serverside
'''
import socketserver,json
import subprocess

connLst = []
##  連線列表，用來儲存一個連線的資訊（代號 地址和埠 連線物件）
class Connector(object):#連線物件類
    def __init__(self,account,password,addrPort,conObj):
        self.account = account
        self.password = password
        self.addrPort = addrPort
        self.conObj = conObj



class MyServer(socketserver.BaseRequestHandler):

    def handle(self):
        print("got connection from",self.client_address)
        register = False
        while True:
            conn = self.request
            data = conn.recv(1024)
            if not data:
                continue
            dataobj = json.loads(data.decode('utf-8'))
            #如果連線客戶端傳送過來的資訊格式是一個列表且註冊標識為False時進行使用者註冊
            if type(dataobj) == list and not register:
                account = dataobj[0]
                password = dataobj[1]
                conObj = Connector(account,password,self.client_address,self.request)
                connLst.append(conObj)
                register = True
                continue
            print(connLst)
            #如果目標客戶端在傳送資料給目標客服端
            if len(connLst) > 1 and type(dataobj) == dict:
                sendok = False
                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('nobody recevied!'.encode('utf-8'))
                continue

if __name__ == '__main__':
    server = socketserver.ThreadingTCPServer(('192.168.1.4',8022),MyServer)
    print('waiting for connection...')
    server.serve_forever()

客戶端實現：

#coding:utf-8
'''
file:client.py.py
date:2017/9/10 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

HOST = '192.168.1.4'  ##
PORT=8022
BUFSIZ = 1024  ##緩衝區大小  1K
ADDR = (HOST,PORT)

tcpCliSock = socket(AF_INET,SOCK_STREAM)
tcpCliSock.connect(ADDR)
userAccount = None
def register():
    myre = r"^[_a-zA-Z]\w{0,}"
    #正則驗證使用者名稱是否合乎規範
    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
    return (accout,password1)

class inputdata(threading.Thread):
    def run(self):
        while True:
            sendto = input('to>>:')
            msg = input('msg>>:')
            dataObj = {'to':sendto,'msg':msg,'froms':userAccount}
            datastr = json.dumps(dataObj)
            tcpCliSock.send(datastr.encode('utf-8'))


class getdata(threading.Thread):
    def run(self):
        while True:
            data = tcpCliSock.recv(BUFSIZ)
            dataObj = json.loads(data.decode('utf-8'))
            print('{} -> {}'.format(dataObj['froms'],dataObj['msg']))


def main():
    while True:
        regInfo = register()
        if  regInfo:
            datastr = json.dumps(regInfo)
            tcpCliSock.send(datastr.encode('utf-8'))
            break
    myinputd = inputdata()
    mygetdata = getdata()
    myinputd.start()
    mygetdata.start()
    myinputd.join()
    mygetdata.join()


if __name__ == '__main__':
    main()