初學python網絡編程socket,並實現文件傳輸【含源代碼】

什麼是Socket

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

socket()函數

Python 中,我們用 socket()函數來創建套接字,語法格式如下:

socket.socket(family, type, protocol)

參數: 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() 創建一個與該套接字相關連的文件

-----**** 上述來源於網絡 ****-----------

實例:實現文件傳輸

實例演示:

服務器端打開
在這裏插入圖片描述
客戶端運行,選擇文件上傳
在這裏插入圖片描述
客戶端文件上傳完畢
在這裏插入圖片描述
服務器端收到文件
在這裏插入圖片描述
注:本次演示不是在局域網上,而是在實際上的服務器上的文件傳輸演示

源代碼:

客戶端

"""
author: 全洛

"""
import socket
import easygui as eg
import json
import os
import time

'''讀取配置文件'''
f = open('clientconf.json', 'r', encoding='utf-8')
date = json.load(f)
ip = date.get('IP')
port = date.get('port')
f.close()

'''創建socket對象'''
client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

try:
    '''連接服務器, 指定IP和端口'''
    client.connect((ip, port))

    '''接收小於 1024 字節的數據'''
    msg = client.recv(1024)

    '''輸出連接成功信息'''
    print(msg.decode('utf-8'))
except Exception as e:
    print(f'連接服務器失敗', e)
else:
    '''選擇發送的文件'''
    filepath = eg.fileopenbox(title='選擇文件')
    '''獲取文件名,文件大小'''
    filename = filepath.split("\\")[-1]
    filesize = os.path.getsize(filepath)
    # 先將文件名傳過去
    # 編碼文件名
    client.send(filename.encode())
    time.sleep(0.5)
    # 再將將文件大小傳過去
    # 編碼文件大小
    client.send(filesize.to_bytes(filesize.bit_length(), byteorder='big'))
    try:
        '''傳輸文件'''
        start_time = time.time()
        with open(filepath, 'rb') as f:
            size = 0
            while True:
                #讀取文件數據,每次1024KB
                f_data = f.read(1024)
                # 數據長度不爲零,傳輸繼續
                if f_data:
                    client.send(f_data)
                    size += len(f_data)
                    if time.time() - start_time == 0:
                        time.sleep(0.5)
                    speed = (size) / (time.time() - start_time)
                    print('\r' + '【上傳進度】:%s%.2f%%, Speed: %.2fMB/s' % ('>' * int(size * 50 / filesize), float(size / filesize * 100), float(speed / 1024 / 1024)), end=' ')
                # 數據長度爲零傳輸完成
                else:
                    print(f'{filename},{float(filesize/1024/1024):.2f}MB, 傳輸完成')
                    break
    except Exception as e:
        print(f'傳輸異常', e)
finally:
    '''關閉客戶端'''
    client.close()

客戶端配置:
{
“IP”: “182.92.112.155”,
“port”: 3636
}

服務器端

"""
author: 全洛

"""
import socket
import os
import json
import threading
import time


'''導入配置文件'''
f = open('serverconf.json', 'r', encoding='utf-8')
dirsave = json.load(f).get('dirSave')
f.close()
if not os.path.exists(dirsave):  # 看是否有該文件夾,沒有則創建文件夾
    os.mkdir(dirsave)

class server:
    def __init__(self):
        ''''''
        ''' 創建socket對象'''
        self.server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

        '''獲取本地'''
        self.host = socket.gethostname()

        '''設置端口'''
        self.port = 3636

        '''綁定地址'''
        self.server.bind((self.host, self.port))

        '''設置最大連接數, 超過後排隊'''
        self.server.listen(12)

        print('服務器已上線\n等待客戶端連接...\n')

    def server100(self):
        while True:
            '''建立客戶端連接'''
            client, addr = self.server.accept()
            print(f'客戶端: {str(addr)}, 連接上服務器')
            msg = f'已連接到服務器{self.host}!\r\n'
            client.send(msg.encode('utf-8'))
            # 有客戶端連接後,創建一個線程給客戶端
            t1 = threading.Thread(self.taskfilethread(client))
            # 設置線程守護
            t1.setDaemon(True)
            # 啓動線程
            t1.start()

    def taskfilethread(self, client):
        '''接收文件名,文件大小'''
        filename = client.recv(1024)
        # time.sleep(0.5)
        filesize = client.recv(1024)
        # 解碼文件名,文件大小
        filename = filename.decode()
        filesize = int.from_bytes(filesize, byteorder='big')
        '''接收文件'''
        try:
            f = open(dirsave+"\\"+ filename, 'wb')
            size = 0
            start_time = time.time()
            while True:
                # 接收數據
                f_data = client.recv(1024)
                # 數據長度不爲零,接收繼續
                if f_data:
                    f.write(f_data)
                    size += len(f_data)
                    if time.time() - start_time == 0:
                        time.sleep(0.5)
                    speed = (size) / (time.time() - start_time)
                    print('\r' + '【下載進度】:%s%.2f%%, Speed: %.2fMB/s' % ('>' * int(size * 50 / filesize), float(size / filesize * 100), float(speed/1024/1024)),end=' ')
                # 數據長度爲零接收完成
                else:
                    break
        except Exception as e:
            print(f'接收異常', e)
        else:
            f.flush()
            print(f'{filename},{float(filesize/1024/1024):.2f}MB, 接收完成')
            f.close()

if __name__ =='__main__':
    server = server()
    server.server100()

服務器端配置:
{
“dirSave”: “C:\Users\Administrator\Desktop\temp”
}


最後

版權聲明:本文爲博主原創文章,遵循 CC 4.0 BY-SA 版權協議,轉載請附上原文出處鏈接和本聲明。
本文鏈接:https://blog.csdn.net/weixin_43347550/article/details/106199850
————————————————

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章