用python打造一個(gè)基于socket的文件(夾)傳輸系統(tǒng)

前言

這段時(shí)間在學(xué)習(xí)python,接觸到了網(wǎng)絡(luò)編程中的socket這塊,加上自己在用的Linux服務(wù)器都是原生支持python的,于是乎有了個(gè)做文件傳輸功能程序的想法。畢竟python語言中,有下載功能的框架一抓一大把,但是主機(jī)與主機(jī)間快速搭建文件(夾)傳輸通道的程序似乎不常見,因?yàn)槲覄偨佑|python不久,有但是我不知道也不奇怪,總得來說,自己做一個(gè)練手,成就感滿滿。

項(xiàng)目地址

https://github.com/Ccapton/python-stuff/tree/master/filetransporter

實(shí)操預(yù)覽

本地主機(jī)發(fā)送文件到遠(yuǎn)端服務(wù)器主機(jī)

思維導(dǎo)圖

文件傳輸系統(tǒng)

以上思維導(dǎo)圖僅供參考,畢竟表達(dá)能力有限,具體功能要追究到代碼處才能分析其原理。

原理

基于socket的通信,相信會編程的朋友都不陌生,而通過socket來傳輸文件也是很常見的,但是這僅僅是對于單個(gè)文件來說很容易實(shí)現(xiàn)。如果是多文件呢?我在實(shí)現(xiàn)本系統(tǒng)之前嘗試了幾次,用單一socket通道來傳輸多文件不切實(shí)際,因?yàn)檎{(diào)用socket.recv()方法的時(shí)候,返回的數(shù)據(jù)格式是原始數(shù)據(jù)str類型,要分割不同文件的數(shù)據(jù)有很大難度。

因?yàn)樯婕暗浇邮铡l(fā)送方兩端文件數(shù)據(jù)接收和發(fā)送進(jìn)度的統(tǒng)一性,就要用另外的指令來控制傳輸工作不亂套,于是我想到多開一個(gè)socket作為傳輸指令的通道,這樣指令和數(shù)據(jù)就分離了,也就容易控制傳輸工作了!所以有了指令線程數(shù)據(jù)線程之分。

要實(shí)現(xiàn)傳輸整個(gè)文件夾,首先要遍歷這個(gè)文件夾,把在其內(nèi)的所有文件結(jié)構(gòu)準(zhǔn)確無誤的還原出來;因?yàn)槭峭ㄟ^一個(gè)socket通過傳輸數(shù)據(jù),所以傳輸文件只能一個(gè)接一個(gè)來,這樣,文件的遍歷工作只能等前一個(gè)文件傳輸完畢后才能繼續(xù)進(jìn)行,于是又要對遍歷工作設(shè)計(jì)一番。經(jīng)過改造,我在文件查找(遍歷)器內(nèi)加入了while循環(huán)體和供外界繼承的回調(diào)類,這樣就能達(dá)到我想要的文件通過socket按順序傳輸?shù)男Ч恕?br> 文件查找器FileFinder(阻塞型)源碼:

import os,time
# 文件、文件夾尋找類 (阻塞型)
# 阻塞的設(shè)計(jì): 為了等待調(diào)用者的耗時(shí)操作【否則很快就完成了文件的遍歷任務(wù),調(diào)用者達(dá)不到順序操作文件(夾)的意圖】
class FileFinder:
   def __init__(self,finderCallback):
       self.finderCallback = finderCallback
       # 文件(夾)路徑下所有文件的總大小
       self.sum_size = 0
       # 調(diào)用者控制的參數(shù),若為False,則遍歷工作繼續(xù)進(jìn)行,若為True,則阻塞任務(wù),等待調(diào)用者完成它的其他耗時(shí)操作后在考慮是否改變此值
       self.recycle = True
       # 調(diào)用者控制的參數(shù),若為False,則正常工作,若為True,則當(dāng)recycle為False時(shí)遍歷工作不阻塞快速完成,recycle為True時(shí)遍歷工作阻塞
       self.off = False

    # 文件(夾)找到時(shí)的回調(diào)類
   class FinderCallback:
       # 找到文件夾
       def onFindDir(self,dir_path):
           pass
       # 找到文件
       def onFindFile(self,file_path,size):
           pass
       # 預(yù)留的刷新函數(shù)
       def onRefresh(self):
           pass

   # 查找文件(夾)方法
   def list_flie(self,root_dir):
       if  os.path.isfile(root_dir):
           while self.recycle:
               time.sleep(0.05)
           if self.finderCallback:
               self.finderCallback.onFindFile(root_dir,os.path.getsize(root_dir))
               self.finderCallback.onRefresh()
               if not self.off:
                  self.recycle = True
       else:
           dirlist = os.listdir(root_dir)  # 列出文件夾下所有的目錄與文件
           for dir in dirlist:
               path = os.path.join(root_dir, dir)
               if os.path.isfile(path):
                   while self.recycle:
                       time.sleep(0.05)
                   if self.finderCallback:
                       self.finderCallback.onFindFile(path,os.path.getsize(path))
                       self.finderCallback.onRefresh()
                       if not self.off:
                          self.recycle = True
               else:
                   while self.recycle:
                       time.sleep(0.05)
                   if self.finderCallback:
                       self.finderCallback.onFindDir(path)
                       self.finderCallback.onRefresh()
                       if not self.off:
                          self.recycle = True
                   # 遞歸調(diào)用(當(dāng)遍歷到文件夾時(shí),繼續(xù)遍歷,直到當(dāng)前文件夾下沒有文件夾為止)
                   self.list_flie(path)

通過繼承FileFinder的回調(diào)類FinderCallback,重寫其 onFindDir(self,dir_path) 和onFindFile(self,file_path,size)方法,在這兩個(gè)方法里面發(fā)送相關(guān)指令到接收端。

但最難的就是兩端的指令線程內(nèi)的socket指令信息的處理,這需要深入代碼去具體研究,本人能力有限,也就不細(xì)說了。需要研究一番的朋友,請移步該項(xiàng)目地址 https://github.com/Ccapton/python-stuff/tree/master/filetransporter

注意

運(yùn)行接收端程序,需要一個(gè)能訪問的地址,也即是說最好是局域網(wǎng)內(nèi)進(jìn)行文件傳輸工作,因?yàn)榫钟蚓W(wǎng)本地ip都是可以直接訪問的,若是在公共網(wǎng)絡(luò)傳輸文件,必須知道接收方主機(jī)的公網(wǎng)ip和內(nèi)網(wǎng)ip。
例如我現(xiàn)在用到接收方主機(jī)是騰訊云的主機(jī),內(nèi)網(wǎng)ip是10.135.xxx.xxx,公網(wǎng)ip是111.120.xxx.xxx。該主機(jī)內(nèi),運(yùn)行接收(服務(wù))端程序

python3 ftserver.py -i 10.135.xxx.xxx -d /home/ubuntu/downloads

而在你的主機(jī)運(yùn)行發(fā)送(客戶)端程序發(fā)送文件夾bilibili

python3 ftclient.py -i 111.120.xxx.xxx -f /Users/capton/desktop/bilibili

因?yàn)轵v訊云、阿里云等國內(nèi)虛擬主機(jī)供應(yīng)商是采用NAT地址轉(zhuǎn)換對云主機(jī)進(jìn)行地址分配的,所以按照我上面的步驟來運(yùn)行兩端程序才能連通。

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

  • 說明 本文 翻譯自 realpython 網(wǎng)站上的文章教程 Socket Programming in Pytho...
    keelii閱讀 2,432評論 0 16
  • 網(wǎng)絡(luò)編程 一.楔子 你現(xiàn)在已經(jīng)學(xué)會了寫python代碼,假如你寫了兩個(gè)python文件a.py和b.py,分別去運(yùn)...
    go以恒閱讀 2,252評論 0 6
  • 最近在學(xué)習(xí)Python看了一篇文章寫得不錯(cuò),是在腳本之家里的,原文如下,很有幫助: 一、網(wǎng)絡(luò)知識的一些介紹 soc...
    qtruip閱讀 2,846評論 0 6
  • Socket創(chuàng)建函數(shù) socket.socket(socket_family, socket_type, prot...
    JianMing閱讀 6,100評論 1 13
  • 個(gè)人認(rèn)為,Goodboy1881先生的TCP /IP 協(xié)議詳解學(xué)習(xí)博客系列博客是一部非常精彩的學(xué)習(xí)筆記,這雖然只是...
    貳零壹柒_fc10閱讀 5,215評論 0 8

友情鏈接更多精彩內(nèi)容