Python编程笔记 - 基于udp协议的聊天小程序

一、需求说明

基于udp协议的聊天小程序：

1. 输入发送方的ip地址
2. 输入发送方的端口号
3. 输入本机接受信息时要绑定的端口号
4. 输入要发送的消息

可给别人发送消息，同时也可以接受别人发来的消息。

二、实现源码

# -*- coding: utf-8 -*-
# __author : Demon
# date : 1/19/18 4:34 PM

# 能收也能发
from threading import *
from socket import *
import socket

def recvData():
    udpSocket.bind(("", recvPort))
    while True:
        recvInfo = udpSocket.recvfrom(1024)
        print("\r>>>%s\n<<<" % recvInfo[0].decode("utf-8"), end="")

def sendData():
    while True:
        msg = input("<<<")
        udpSocket.sendto(msg.encode("utf-8"), (ipAddr, sendPort))

udpSocket = socket.socket(AF_INET, SOCK_DGRAM)
ipAddr = None
sendPort = None
recvPort = None

def main():
    global sendPort
    global recvPort
    global ipAddr
    ipAddr = input("请输入要发送的ip地址: ")
    sendPort = int(input("请输入要发送的端口号: "))
    recvPort = int(input("请输入本机要绑定的端口号: "))
    tr = Thread(target = recvData)
    tr.start()

    ts = Thread(target = sendData)
    ts.start()

    tr.join()
    ts.join()
    udpSocket.close()

if __name__ == '__main__':
    main()

三、代码说明和演示

1. 要发送的ip地址，如果不输入，即代表你输入的是""，则表示是当前机器的ip地址
2. \r 表示打印时，先清除原有的输入
3. 代码采用utf-8编码，要注意发送与接收时要采用相同的编码与解码方式