TCP简单实现文件上传与下载

服务器端：

import socket
from ms.login import read
from ms.login import down
from ms.login import up

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')))

    choice=conn.recv(4).decode('utf-8')   #客户端选择上传还是下载

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

    elif choice=='D':   # 文件下载
        down(conn)  #调下载方法
    else:
        print('error，错误的操作指令！')
else:   #登录失败 关闭连接和对象
    conn.send(bytes('用户登录失败，断开连接...'.encode('utf-8')))
    conn.close()
    sk.close()

 

 

调用函数的模块：

import os
import struct
import operator
import time
import json


#登录验证
def read(user,psd):
    with open('info','r',encoding='utf-8')as f:
        for i in f:     #读取所有记录
            i=i.strip()     #去空格
            if user in i:   #如果用户名相匹配
                n_user,n_psd=i.split('|')     #分割字符串 得到用户名密码
                if n_psd==psd:
                    return True     #密码正确则为TRUE
                else:
                    return False    #错误则为FALSE
#上传
def up(conn,buffer):
    long=conn.recv(4)   #4字节的数据长度包
    pack_long=struct.unpack('i',long)[0]  #报头的长度
    filejson=conn.recv(pack_long).decode('utf-8')
    fileinfo=json.loads(filejson) #报头内容

    filename = fileinfo['filename']
    filesize = fileinfo['filesize']

    with open(filename, 'wb')as f:  #将接收到的数据写入文件
        while filesize:
            if filesize >= buffer:   #如果文件大小 大于读取数据量
                filedata = conn.recv(buffer)
                f.write(filedata)     #就将读取的数据量写入
                filesize -= buffer
            else:
                filedata = conn.recv(filesize)
                f.write(filedata)    #直接把文件写入
                break
        print('文件%s上传成功' % filename)


#下载
def down(conn):
    files = os.listdir()
    lis_file = ''
    for i in files:
        lis_file += i + '\n'
    if operator.eq(lis_file, ''):  # 如果文件列表为空
        conn.send(''.encode('utf-8'))
    else:
        conn.send(lis_file.encode('utf-8'))  #把文件列表发给客户端
        while True:
            filename=conn.recv(1024).decode('utf-8')  #接收要下载的文件名
            if filename not in files:
                conn.send(bytes('文件名不存在!'.encode('utf-8')))
            else:
                conn.send(bytes('开始文件传输!'.encode('utf-8')))
                break
        size=os.stat(filename).st_size
        info=struct.pack('i',size)
        conn.send(info)    #文件信息大小打包发送给客户端

        with open(filename,'rb')as f:
            while True:
                file_data=f.read(4096)
                if not file_data:
                    break
                conn.send(file_data)

        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)
        print(show)
        path=os.path.dirname(os.path.abspath(__file__))
        path=os.path.join(path,'log')

        os.chdir(path)    #临时把当前目录变为 E:\program\ms\log
        with open('data_log','a',encoding='utf-8')as f:
            f.write(show)              #把输出信息记录入日志

 

客户端：

import socket
import struct
import os,sys
import operator
import time
import json

buffer=1024

sk=socket.socket()
sk.connect(('127.0.0.1',8080))
user=input('请输入用户名：')
psd=input('请输入密码：')
login=str(user)+'|'+str(psd)
sk.send(bytes(login.encode('utf-8')))  #传用户名密码给客户端
ret=sk.recv(1024).decode('utf-8')   #接收登录状态  ok

if ret=='ok':
    menu=sk.recv(1024).decode('utf-8')
    print(menu)              #显示操作菜单
    do=input('请输入要执行的操作...').upper()
    sk.send(bytes(do.encode('utf-8')))    #输入上传还是下载

    #选择上传
    if do=='U':
        filepath = input('请输入要上传的文件路径\n')
        filename = input('请输入文件名\n')

        path = os.path.join(filepath, filename)
        filesize = os.stat(path).st_size  # 文件大小
        fileinfo = {'filename': filename, 'filepath': filepath,'filesize':filesize}
        filejson=json.dumps(fileinfo).encode('utf-8')
        long=len(filejson) #报头的长度
        pack_long=struct.pack('i',long)  #报头长度变为4字节
        sk.send(pack_long)
        sk.send(filejson)

        with open(path, 'rb')as f:
            while filesize:
                if filesize >= 1024:
                    filedata = f.read(buffer)
                    sk.send(filedata)
                    filesize -= buffer
                else:
                    filedata = f.read(filesize)
                    sk.send(filedata)
                    break

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


    # 选择下载
    elif do=='D':
        r=sk.recv(1024).decode('utf-8')   #收到文件列表
        if operator.eq(r,''):
            print('没有可以下载的文件')
        print(r)
        filename= input('请输入要下载的文件名>>>')
        sk.send(bytes(filename.encode('utf-8')))   #发送要下载的文件名
        state=sk.recv(1024).decode('utf-8')  #文件下载状态
        print(state)

        info=sk.recv(4)   #4字节的包
        size=struct.unpack('i',info)[0]  #文件的字节大小
        print(filename,size)

        data=sk.recv(size).decode('utf-8')

        path = os.path.dirname(os.path.abspath(__file__))
        path = os.path.join(path, 'down')
        os.chdir(path)
        with open(filename,'w+',encoding='utf-8')as f:
            f.write(data)
        print(data)

    else:
        print('error，错误的操作指令！')


sk.close()