TCP簡單實現文件上傳與下載

服務器端：

import socket  2 from ms.login import read  3 from ms.login import down  4 from ms.login import up  5 
buffer=1024  #一次讀取內容長度字節
sk=socket.socket()   #創建socket對象
sk.bind(('127.0.0.1',8080))  #綁定地址
sk.listen()  #監聽

conn,addr=sk.accept() #建立一個連接

info=conn.recv(50).decode('utf-8')   #接收客戶端的用戶名密碼
user,psd=info.split('|')   #分割獲取用戶名 和 密碼

ret=read(user,psd)  #調read讀取用戶名信息 驗證用戶名密碼是否正確
if ret:  #登錄成功
    conn.send(bytes('ok'.encode('utf-8')))   #返回一個OK
    conn.send(bytes('用戶登錄成功,請選擇要執行的操作【U：上傳文件】【D：下載文件】'.encode('utf-8'))) 20 
    choice=conn.recv(4).decode('utf-8')   #客戶端選擇上傳還是下載

    if choice=='U':  #文件上傳
        up(conn,buffer)    #調上傳方法

    elif choice=='D':   # 文件下載
        down(conn)  #調下載方法
    else: 29         print('error，錯誤的操作指令！') 30 else:   #登錄失敗 關閉連接和對象
    conn.send(bytes('用戶登錄失敗，斷開連接...'.encode('utf-8'))) 32  conn.close() 33     sk.close()

 

 

調用函數的模塊：

import os  2 import struct  3 import operator  4 import time  5 import json  6 

#登錄驗證
def read(user,psd): 10     with open('info','r',encoding='utf-8')as f: 11         for i in f:     #讀取所有記錄
            i=i.strip()     #去空格
            if user in i:   #如果用戶名相匹配
                n_user,n_psd=i.split('|')     #分割字符串 得到用戶名密碼
                if n_psd==psd: 16                     return True     #密碼正確則為TRUE
                else: 18                     return False    #錯誤則為FALSE
#上傳
def up(conn,buffer): 21     long=conn.recv(4)   #4字節的數據長度包
    pack_long=struct.unpack('i',long)[0]  #報頭的長度
    filejson=conn.recv(pack_long).decode('utf-8') 24     fileinfo=json.loads(filejson) #報頭內容

    filename = fileinfo['filename'] 27     filesize = fileinfo['filesize'] 28 
    with open(filename, 'wb')as f:  #將接收到的數據寫入文件
        while filesize: 31             if filesize >= buffer:   #如果文件大小 大於讀取數據量
                filedata = conn.recv(buffer) 33                 f.write(filedata)     #就將讀取的數據量寫入
                filesize -= buffer 35             else: 36                 filedata = conn.recv(filesize) 37                 f.write(filedata)    #直接把文件寫入
                break
        print('文件%s上傳成功' % filename) 40 

#下載
def down(conn): 44     files = os.listdir() 45     lis_file = ''
    for i in files: 47         lis_file += i + '\n'
    if operator.eq(lis_file, ''):  # 如果文件列表為空
        conn.send(''.encode('utf-8')) 50     else: 51         conn.send(lis_file.encode('utf-8'))  #把文件列表發給客戶端
        while True: 53             filename=conn.recv(1024).decode('utf-8')  #接收要下載的文件名
            if filename not in files: 55                 conn.send(bytes('文件名不存在!'.encode('utf-8'))) 56             else: 57                 conn.send(bytes('開始文件傳輸!'.encode('utf-8'))) 58                 break
        size=os.stat(filename).st_size 60         info=struct.pack('i',size) 61         conn.send(info)    #文件信息大小打包發送給客戶端

        with open(filename,'rb')as f: 64             while True: 65                 file_data=f.read(4096) 66                 if not file_data: 67                     break
 conn.send(file_data) 69 
        show='{0}\n下載文件：{1}:\n{2}byte\nSuccessful...\n\n'.format(time.strftime('%Y-%m-%d:%H:%M:%S',time.localtime(time.time())),filename,size) 71         print(show) 72         path=os.path.dirname(os.path.abspath(__file__)) 73         path=os.path.join(path,'log') 74 
        os.chdir(path)    #臨時把當前目錄變為 E:\program\ms\log
        with open('data_log','a',encoding='utf-8')as f: 77             f.write(show)              #把輸出信息記錄入日志

 

客戶端：

import socket  2 import struct  3 import os,sys  4 import operator  5 import time  6 import json  7 
buffer=1024

sk=socket.socket() 11 sk.connect(('127.0.0.1',8080)) 12 user=input('請輸入用戶名：') 13 psd=input('請輸入密碼：') 14 login=str(user)+'|'+str(psd) 15 sk.send(bytes(login.encode('utf-8')))  #傳用戶名密碼給客戶端
ret=sk.recv(1024).decode('utf-8')   #接收登錄狀態 ok

if ret=='ok': 19     menu=sk.recv(1024).decode('utf-8') 20     print(menu)              #顯示操作菜單
    do=input('請輸入要執行的操作...').upper() 22     sk.send(bytes(do.encode('utf-8')))    #輸入上傳還是下載

    #選擇上傳
    if do=='U': 26         filepath = input('請輸入要上傳的文件路徑\n') 27         filename = input('請輸入文件名\n') 28 
        path = os.path.join(filepath, filename) 30         filesize = os.stat(path).st_size  # 文件大小
        fileinfo = {'filename': filename, 'filepath': filepath,'filesize':filesize} 32         filejson=json.dumps(fileinfo).encode('utf-8') 33         long=len(filejson) #報頭的長度
        pack_long=struct.pack('i',long)  #報頭長度變為4字節
 sk.send(pack_long) 36  sk.send(filejson) 37 
        with open(path, 'rb')as f: 39             while filesize: 40                 if filesize >= 1024: 41                     filedata = f.read(buffer) 42  sk.send(filedata) 43                     filesize -= buffer 44                 else: 45                     filedata = f.read(filesize) 46  sk.send(filedata) 47                     break

        show = '{0}\n上傳文件：{1}:\n{2}byte\nSuccessful...\n\n'.format( 50             time.strftime('%Y-%m-%d:%H:%M:%S', time.localtime(time.time())), filename,filesize) 51         print(show) 52 

    # 選擇下載
    elif do=='D': 56         r=sk.recv(1024).decode('utf-8')   #收到文件列表
        if operator.eq(r,''): 58             print('沒有可以下載的文件') 59         print(r) 60         filename= input('請輸入要下載的文件名>>>') 61         sk.send(bytes(filename.encode('utf-8')))   #發送要下載的文件名
        state=sk.recv(1024).decode('utf-8')  #文件下載狀態
        print(state) 64 
        info=sk.recv(4)   #4字節的包
        size=struct.unpack('i',info)[0]  #文件的字節大小
        print(filename,size) 68 
        data=sk.recv(size).decode('utf-8') 70 
        path = os.path.dirname(os.path.abspath(__file__)) 72         path = os.path.join(path, 'down') 73  os.chdir(path) 74         with open(filename,'w+',encoding='utf-8')as f: 75  f.write(data) 76         print(data) 77 
    else: 79         print('error，錯誤的操作指令！') 80 

sk.close()