1. 程式人生 > >基於UDP的聊天系統的實現

基於UDP的聊天系統的實現

基於UDP使用Python實現了一個聊天系統。

客戶端的程式碼實現(使用鍵盤對於資料的讀入):

#!/usr/bin/python
# -*- coding: UTF-8 -*-
import socket
server_addr=('127.0.0.1',8080)
s=socket.socket(socket.AF_INET,socket.SOCK_DGRAM)
while True:
	data = input('Please enter the message:')
	s.sendto(data,server_addr)
	print(s.recv(1024))
s.close()

伺服器端的程式碼實現(對於client端的請求進行響應並且將資料儲存到資料庫中):

#!/usr/bin/python
# -*- coding: UTF-8 -*-
import socket
import sqlite3
server_addr=('127.0.0.1',8080)
s=socket.socket(socket.AF_INET,socket.SOCK_DGRAM)
#connect the database and if not have the 'chat_data.db',while to create database file 
conn_db=sqlite3.connect('chat_data.db')
#create a cursor
cursor=conn_db.cursor()
#create table
cursor.execute('create table user1 (ip varchar(100),port int(7),data varchar(255))')
s.bind(server_addr)
print('Bind UDP on 9999')
while True:
	data,addr=s.recvfrom(1024)
	ip=addr[0]
	port=addr[1]
	message=data
	print('Received from :',addr,data)
	s.sendto(b'Hello:'+data,addr)
	#insert message to database
	cursor.execute("insert into user1(ip,port,data)values('%s',%d,'%s')" % (ip,port,message))
	cursor.execute("select * from user1")
	values=cursor.fetchall()
	print(values)
cursor.close()
conn_db.commit()
conn_db.close()

注:由於作者水平有限,若有問題,請留言,謝謝!