基于python聊天脚本

展开查看代码

import threading
import socket
 
#接受信息函数
def rec(upd):
    while True:
        #接收消息，最多为1024字节
        data = upd.recvfrom(1024)
        #data为一个元组，info为信息内容，frm为发送者ip和端口号
        info, frm = data
        #设置编码格式
        info = info.decode("utf-8")
        print("收到来自"+str(frm)+'的消息：'+info)
 
#发送信息函数
def send(udp,dest_ip,dest_port):
    while True:
        data = input("请输入要发送的数据:")
        #发送消息
        udp.sendto(data.encode("utf-8"), (dest_ip, dest_port))
 
 
def main():
    #创建套接字
    udp = socket.socket(socket.AF_INET,socket.SOCK_DGRAM)
    #设置固定端口
    port_self = int(input("请输入本地端口号"))
    udp.bind(("",port_self))
    dest_ip = input("请输入对方ip：")
    dest_port = int(input("请输入端口号"))
    #创建接收进程
    t1 = threading.Thread(target=rec, args=(udp,))
    #创建发送进程
    t2 = threading.Thread(target=send, args=(udp, dest_ip, dest_port))
    #开始接收进程
    t1.start()
    #开始发送进程
    t2.start()
 
 
if __name__ == '__main__':
    main()