python 入門 socket 實現udp聊天室

 

什么是 Socket?

Socket又稱"套接字"，應用程序通常通過"套接字"向網絡發出請求或者應答網絡請求，使主機間或者一台計算機上的進程間可以通訊。

---

socket()函數

Python 中，我們用 socket（）函數來創建套接字，語法格式如下：

socket.socket([family[, type[, proto]]])

參數

• family: 套接字家族可以使AF_UNIX或者AF_INET
• type: 套接字類型可以根據是面向連接的還是非連接分為SOCK_STREAM或SOCK_DGRAM
• protocol: 一般不填默認為0.

Socket 對象(內建)方法

函數	描述
服務器端套接字
s.bind()	綁定地址（host,port）到套接字， 在AF_INET下,以元組（host,port）的形式表示地址。
s.listen()	開始TCP監聽。backlog指定在拒絕連接之前，操作系統可以掛起的最大連接數量。該值至少為1，大部分應用程序設為5就可以了。
s.accept()	被動接受TCP客戶端連接,(阻塞式)等待連接的到來
客戶端套接字
s.connect()	主動初始化TCP服務器連接，。一般address的格式為元組（hostname,port），如果連接出錯，返回socket.error錯誤。
s.connect_ex()	connect()函數的擴展版本,出錯時返回出錯碼,而不是拋出異常
公共用途的套接字函數
s.recv()	接收TCP數據，數據以字符串形式返回，bufsize指定要接收的最大數據量。flag提供有關消息的其他信息，通常可以忽略。
s.send()	發送TCP數據，將string中的數據發送到連接的套接字。返回值是要發送的字節數量，該數量可能小於string的字節大小。
s.sendall()	完整發送TCP數據，完整發送TCP數據。將string中的數據發送到連接的套接字，但在返回之前會嘗試發送所有數據。成功返回None，失敗則拋出異常。
s.recvfrom()	接收UDP數據，與recv()類似，但返回值是（data,address）。其中data是包含接收數據的字符串，address是發送數據的套接字地址。
s.sendto()	發送UDP數據，將數據發送到套接字，address是形式為（ipaddr，port）的元組，指定遠程地址。返回值是發送的字節數。
s.close()	關閉套接字
s.getpeername()	返回連接套接字的遠程地址。返回值通常是元組（ipaddr,port）。
s.getsockname()	返回套接字自己的地址。通常是一個元組(ipaddr,port)
s.setsockopt(level,optname,value)	設置給定套接字選項的值。
s.getsockopt(level,optname[.buflen])	返回套接字選項的值。
s.settimeout(timeout)	設置套接字操作的超時期，timeout是一個浮點數，單位是秒。值為None表示沒有超時期。一般，超時期應該在剛創建套接字時設置，因為它們可能用於連接的操作（如connect()）
s.gettimeout()	返回當前超時期的值，單位是秒，如果沒有設置超時期，則返回None。
s.fileno()	返回套接字的文件描述符。
s.setblocking(flag)	如果flag為0，則將套接字設為非阻塞模式，否則將套接字設為阻塞模式（默認值）。非阻塞模式下，如果調用recv()沒有發現任何數據，或send()調用無法立即發送數據，那么將引起socket.error異常。
s.makefile()	創建一個與該套接字相關連的文件

 

 

實現udp聊天室:

   所需模塊  socket  threading

 思路：我們需要用socket模塊實現接收數據與發送數據，由於需要我們接收與發送功能在同時進行，因此我們需要用threading 創建多線程來實現。

 

 發送：使用while True 循環  使s.sendto() 可以不斷的向遠程端發送數據

 接收：通過循環s.recvfrom()得到的udp數據打印到控制台上實現數據的實時接收

 使用threading分別為創建接收與發送的線程。

            

#注 socket只能發送字節類型的數據 所以在發送時要將發送的數據通過str.encode() 轉為字節類型，接收時在通過str.decode()轉為字符串。

 

Demo： 

import socket
import threading

#輸入
host=socket.gethostname()
port=int(input('本地端口:'))
cliip=input('目標地址:')
cliport=int(input('目標端口:'))
cliaddr=(cliip,cliport)
print('嘗試與目標地址 '+str(cliaddr)+'連接\n')



#創建socket
s=socket.socket(socket.AF_INET,socket.SOCK_DGRAM)
s.bind((host,port))


#接收
class server(threading.Thread):
  def __init__(self):
    threading.Thread.__init__(self)
  def run(self):
    print('開啟服務端\r\n')
    while True:
        
        msg,addr=s.recvfrom(2048)#接收
        print('\n接收來自'+str(addr)+'的信息:\n\t'+msg.decode()+'\n：')
          
          

#發送      
class client(threading.Thread):
  def __init__(self,cliip,cliport):
    threading.Thread.__init__(self)
    self.cliaddr=cliaddr
  def run(self):
    while True:
      date=input('：')
      s.sendto(date.encode(),cliaddr) #發送
        
        

 #啟動線程       
thread1=server()
thread2=client(cliaddr)
thread1.start()
thread2.start()