python核心---socket编程实现聊天室

1.udp方式

from threading import Thread
from socket import *

def recvData():
    while True:
        recvInfo = udpSocket.recvfrom(1024)
        print('\r>>%s:%s' %(recvInfo[1], recvInfo[0]))
        print11('<<')

def sendData():
    while True:
        sendInfo = input('<<')
        udpSocket.sendto(sendInfo.encode('utf-8'), (destIp, destPort))

udpSocket = None
destIp = ''
destPort = 0

def main():
    udpSocket = socket(AF_INET, SOCK_DGRAM)
    udpSocket.bind(('', 4567))

    destIp = input('请输入对方的ip:')
    destPost = int(input('请输入对方的端口号:'))

    tr = Thread(target=recvData)
    ts = Thread(target=sendData)

    tr.start()
    ts.start()

    tr.join()
    ts.join()

if __name__=='__main__':
    main()

2.TCP方式
(1)客户端

from socket import *

# 创建套接字
severSocket = socket(AF_INET, SOCK_STREAM)
# 绑定端口
severSocket.bind(("", 8899))
# 监听事件
serverSocket.listen(5)
while True:
    # 接收请求
    clientSocket, clientInfo = serverSocket.accept()
    while True:
        recvData = clientSocket.recv(1024)
        if len(recvData)>0:
            print("%s:%s" %(str(clientInfo), recvData))
        else:
            break

        sendData = raw_input('send:')
        clientSocket.send(sendData)
    # 关闭套接字
    clientSocket.close()

(2)客户端

from socket import *

# 创建套接字
clientSocket = socket(AF_INET, SOCK_STREAM)
# 连接
clientScoket.connect(('127.0.0.1', 8989))

clientSocket.send('haha'.encode('utf-8'))

recvData = clientSocket.recv(1024)
print('recvData:%s' %recvData)

# 关闭套接字
clientSocket.close()