python socket 实现模拟ftp 上传、下载文件

ftp 实现功能如下：

1、实现多用户登陆，需使用用户名密码校验登陆过程

2、每个用户有自己的加目录

3、可使用ls、cd指令切换查看目录

4、可是实现上传下载文件的功能

5、可以断点续传

 

 

目前只是简单实现，还存在较多bug

客户端代码

#!/usr/bin/python
# -*- coding: utf-8 -*-
#Author:xianyanghua
# @Time    :2019/7/9 10:47
# @File    : socket_client.py
# @Software: PyCharm
import struct
from socket import *
import os
import sys
class mysocketClient(object):
    def __init__(self,HOST,PORT,dire=None):
        self.sock= socket(AF_INET,SOCK_STREAM)
        self.sock.connect((HOST,PORT))
        self.dire=dire
    def start(self):
        self.auth()
    def auth(self):
        while True:
            msg=raw_input('请输入用户名密码 username:password >>>')
            if msg.count(':')==1:
                self.sock.send(msg.encode('utf-8'))
                data=self.sock.recv(1024)
                if data.startswith('你好'):
                    username,password=msg.split(':')
                    self.dire=username
                    self.interactive()
                else:
                    continue
            else:
                print('输入格式有误')
                continue
    def interactive(self):
        while True:
            print('instruction  fuction [filename]')
            try:
                if not self.dire:
                    msg = raw_input('>>>>%s '%self.dire)
                else:
                    msg= raw_input('>>>[%s]'%self.dire )
                if msg.count(' ')==1:
                     fuc,filename=msg.split(' ')
                     print('.........1.........')
                     print(fuc,filename)
                elif not msg:
                    print('不能为空')
                    continue
                else:
                    fuc=msg
                    filename=None
                print('2......................')
                print(fuc,filename)
                if hasattr(self,fuc):
                    print('...............')
                    func=getattr(self,fuc)
                    print(func)
                    func(filename)
            except Exception as e:
                print(e)
                continue
    def push(self,filename):
        print(filename)
        msg='push '+filename
        self.sock.send(msg.encode('utf-8'))
        data = self.sock.recv(1024)
        print(data)
        if data.decode('utf-8') == 'ready':
            file_size = os.path.getsize(filename)
            print(file_size)
            length = struct.pack('i', file_size)
            self.sock.send(length)
            f = file(filename, 'rb')
            send_size = 0
            while not file_size == send_size:
                if file_size - send_size > 1024:
                    msg = f.read(1024)
                    send_size += len(msg)
                else:  # left data less than 1024
                    msg = f.read(file_size - send_size)
                    send_size += (file_size - send_size)
                self.sock.send(msg)
                per = int((float(send_size) / float(file_size)) * 100)
                self.processBar(per)
            f.close()
            data = self.sock.recv(1024)
            if data.decode('utf-8') == 'success':
                print('')
                self.sock.send('success'.encode('utf-8'))
                print('传输完成')

    def get(self,filename):
        pass

    def ls(self,filename):
        if not filename:
            self.sock.send('ls'.encode('utf-8'))
            data=self.sock.recv(1024)
            print('[[[[]]]')
            print(data)
    def cd(self,filename):
        if not filename:
            print('filename 不能为空')
        self.sock.send('cd '+filename)
        data=self.sock.recv(1024)
        print(data)
        if data=='cdSuccess':
          self.dire=filename
          print(data)
        else:
            print(data)
    def processBar(self,percent):
        bar_length = 100
        hashes = '#' * int(percent / 100.0 * bar_length)
        spaces = ' ' * (bar_length - len(hashes))
        sys.stdout.write("\rPercent: [%s] %d%%" % (hashes + spaces, percent))
        sys.stdout.flush()
if __name__=='__main__':
    HOST='127.0.0.1'
    PORT=8003
    s1=mysocketClient(HOST,PORT)
    s1.start()

服务端代码

#!/usr/bin/python
# -*- coding: utf-8 -*-
#Author:xianyanghua
# @Time    :2019/7/16 12:52
# @File    : ftpserver.py
# @Software: PyCharm
import socketserver
import os
import sys
import struct
class MyTCPHandler(socketserver.BaseRequestHandler):
    def handle(self):
        try:
            BASEDIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
            os.chdir(BASEDIR)
            os.chdir('ftpserver')
            userconf={'hua': '123', 'liu': '321'}
            data=self.request.recv(1024)
            user,password=data.split(':')
            if userconf[user]==password:
                self.request.send('你好'+user+'欢迎登录ftp服务器')
                platform=sys.platform
                if platform=='win32':
                    pwd=os.getcwd()
                    os.chdir('hua')
                    pwd = os.getcwd()
                print(platform,pwd)
            else:
                self.request.send('用户名密码错误')
                self.request.close()
        except Exception as e:
            print(e)
            self.request.close()
        try:
            while True:
                data = self.request.recv(1024)
                if not data:
                    continue
                if data=='ls':
                    data=os.listdir('.')
                    print( data)
                    data=str(data)
                try:
                    if data.startswith('cd'):
                        fuc,filename=data.split(' ')
                        print(data)
                        print (fuc,filename)
                        os.chdir(filename)
                        data='cdSuccess'
                except Exception as e:
                    print(e)
                    data=e

                if data.startswith('push'):
                    filename=data.split(' ')[1]
                    self.request.send('ready')
                    print('......')
                    file_size = self.request.recv(4)
                    file_size=struct.unpack('i',file_size)[0]
                    print(file_size,',,,,,,,,,,,')
                    recv_size=0
                    recv_msg=''
                    f = file(filename, 'wb')
                    while not file_size == recv_size:  # 只要文件总大小和已收到的大小不想等，就代表还没收完
                        if file_size - recv_size > 1024:  # 文件总大下减已收到的大小等于还剩下没收到的大小，如果这个数大于1024，代表一次肯定收不完，那就还得多循环几次
                            data = self.request.recv(1024)  # 这次收1024字节，但实际上收到的可能比1024小，所以需要以实际收到的数为准
                            recv_size += len(data)  # 已收到的大小加上这一次循环收到的实际大小
                        else:  # 如果最后剩下的少于1024，那就一次性把剩下的都收过来
                            data = self.request.recv(file_size - recv_size)
                            # recv_size = file_size #不能这么写，因为这一次依然不一定能一次性收完，因为实际收到的数据可能比你规定的数据要少， 所以需要按下面这行的方式写
                            recv_size += (file_size - recv_size)
                        f.write(data)
                    print(recv_msg)
                    f.close()
                    self.request.send('success')
                    data=self.request.recv(1024)
                    print(data)
                    continue
                self.request.send(data)
        except Exception as e:
            print(e)
            self.request.close()
            print(data)
if __name__=='__main__':
    HOST='127.0.0.1'
    PORT=8003
    server=socketserver.ThreadingTCPServer((HOST,PORT),MyTCPHandler)
    server.serve_forever()