使用python编写linux文件传输命令

2019独角兽企业重金招聘Python工程师标准>>>

1.这里使用到的模块有：socket , os ,sys , struct ,fire ,将其导入，如下图

import socket
import os
import sys
import struct
import fire

2.发送端

def tran(self,ip,tfname='file'):#默认文件名是file
        p=ProgressBar(total=100) #设置进度条总数为100
        tserver_adress = (ip, 1997) #用元组保存ip和端口，端口默认是1997
        max_size = 1024 #设定每一次传输1024字节
        if os.path.isfile(tfname): #判断是否是一个文件
            client = socket.socket(socket.AF_INET, socket.SOCK_STREAM) #使用tcp
            client.connect(tserver_adress) 
            print('Starting the tran at')
            asize=os.stat(tfname).st_size #获取文件大小
            fmessage=struct.pack('128sl',os.path.basename(tfname).encode('utf-8'),asize) #将文件名，文件大小转换为字符串
            client.send(fmessage) #先把文件信息发给接收端
            b=0
            w=open(tfname,'rb') #用二进制打开该文件
            while True:
                p.log('%.2lf' %(b/asize*100)) #进度条精确到小数点后俩位
                s = w.read(max_size) #读取
                if not s:
                    break
                client.send(s)
                b += 1024
            w.close()
            client.close()
        print('\nsend sucess!!\n')

3.接收端

def recv(self,ip):
        p=ProgressBar(total=100)
        max_size = 1024
        print('Starting the recv at')
        server_address = (ip, 1997)
        server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        server.bind(server_address)
        server.listen(1)
        client, addr = server.accept()
        client.settimeout(1000)
        file_size = struct.calcsize('128sl') #设定格式
        print('wait for a client to call')
        bf=client.recv(file_size) #按上面file_size格式接受文件信息
        if bf:
            fname, fsize = struct.unpack('128sl', bf) 
            newfname = fname.strip(b'\x00') #去掉文件名后面的...\x00...字符
            print('now recv a file %s, filesize is %s' %(newfname,fsize))
            k=input('please input y or n,Confirm acceptance:')
            if k == 'y':
                r = open(newfname, 'wb')
                print('start receiving....')
                alldata=0
                while not alldata == fsize:
                    p.log('%.2lf' %(alldata / fsize * 100))
                    if fsize-alldata>1024:
                        data= client.recv(max_size)
                        alldata+=len(data)
                    else:
                        data=client.recv(fsize-alldata)
                        alldata=fsize
                    r.write(data)
                r.close()
            elif k=='n':
                print('over!')
                os._exit(0)
            server.close()
            print('\nrecv success!!!\n')

接收端代码和发送端没多大区别。。。

4.实现命令行进度条

class ProgressBar:
    def __init__(self, count = 0, total = 0, width = 50):
        self.count = count
        self.total = total
        self.width = width
        self.pp=0
    def move(self,d):
        self.count = d
    def log(self,d):
        sys.stdout.flush()
        p=int(float(d)) #将base 10字符串装换为int型
        if p-self.pp == 1:
            sys.stdout.write('{0:3}/{1:3}: '.format(d, self.total))
            self.pp=p
            sys.stdout.write('#' * int(p/2) + '-' * (self.width - int(p/2)) + '\r')
            if p == self.width:
                sys.stdout.write('\n')
            sys.stdout.flush()

5.运行效果

确认是否接收

传输过程

传输完成

6.使用方法

Usage:        1.recv:ftf recv [ip]
              2.tran:ftf tran [ip] [filename]

When using, open the receiver first(recv), development of sending again(tran)!（有道翻译的，英文太差！！）

 

7.完整项目代码

https://git.oschina.net/nanxun/file_to_file

转载于:https://my.oschina.net/u/3264690/blog/863439