效果圖:
1.gif
使用Python socket編程實現(xiàn)簡單的聊天室功能缴饭。
服務(wù)器和客戶端使用UDP編程,客戶端兩個線程一個負責接收,一個負責發(fā)送竭贩。
服務(wù)器:接收消息并保存地址,如果觸發(fā)‘EXIT’關(guān)鍵字則從地址表中移除該地址
def main():
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
addr = ('127.0.0.1', 9999)
s.bind(addr)
logging.info('UDP Server on %s:%s...', addr[0], addr[1])
user = {} # {addr:name}
while True:
try:
data, addr = s.recvfrom(1024)
if not addr in user:
for address in user:
s.sendto(data + ' 進入聊天室...'.encode(), address)
user[addr] = data.decode('utf-8')
continue
if 'EXIT' in data.decode('utf-8'):
name = user[addr]
user.pop(addr)
for address in user:
s.sendto((name + ' 離開了聊天室...').encode(), address)
else:
print('"%s" from %s:%s' %
(data.decode('utf-8'), addr[0], addr[1]))
for address in user:
if address != addr:
s.sendto(data, address)
except ConnectionResetError:
logging.warning('Someone left unexcept.')
客戶端:兩個線程莺禁,并設(shè)置接收線程為守護線程
def recv(sock, addr):
'''
一個UDP連接在接收消息前必須要讓系統(tǒng)知道所占端口
也就是需要send一次留量,否則win下會報錯
“ data=sock.recv(1024)
OSError: [WinError 10022] 提供了一個無效的參數(shù)。 ”
'''
sock.sendto(name.encode('utf-8'), addr)
while True:
data = sock.recv(1024)
print(data.decode('utf-8'))
def send(sock, addr):
while True:
string = input()
message = name + ' : ' + string
data = message.encode('utf-8')
sock.sendto(data, addr)
if string == 'EXIT':
break
def main():
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
server = ('127.0.0.1', 9999)
tr = threading.Thread(target=recv, args=(s, server), daemon=True)
ts = threading.Thread(target=send, args=(s, server))
tr.start()
ts.start()
ts.join()
s.close()
if __name__ == '__main__':
print("-----歡迎來到聊天室,退出聊天室請輸入'EXIT'-----")
name = input('請輸入你的名稱:')
print('-----------------%s------------------' % name)
main()
可以通過繼承socket.socket類來重構(gòu)哟冬,那樣子顯得跟規(guī)范一些