python： 基于select的sokct ftp 客户端

#7512
import socket
import hashlib,time
import json
import sys,os
import random


BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.append(BASE_DIR)
from conf import setting
import signal


  # 切换工作目录
# def exit_handle(signum, frame):
#
#     savedict,addr = client.get_save()
#     print("dddd",savedict,addr)
#     if addr:
#         with open(addr + \'\\%s\'%savedict["user"], \'wb\') as f:
#             f.write(json.dumps(savedict).encode())
#     print(\'\'\'
#         *******************************
#                     暂   停
#         *******************************
#         \'\'\')
#     exit()
#
# signal.signal(signal.SIGINT,exit_handle)
# signal.signal(signal.SIGTERM,exit_handle)




class selectFTPClient(object):
    \'\'\'
        FTP客户端
        \'\'\'
    SOCK_FAMALY = socket.AF_INET  # IP v4
    SOCK_TYPE = socket.SOCK_STREAM
    SOCK_ALLOW_CONN = 1024

    def __init__(self,ip = \'localhost\' ,port = 9999):
        self.client = socket.socket(self.SOCK_FAMALY,self.SOCK_TYPE)
        try:
            self.client.connect((ip,port))
            self.msg_dict = {}
            print("登陆时间:[%s],主机地址:[%s]"%(time.strftime(\'%Y-%m-%d %H:%M:%S\',time.localtime()),self.client.getpeername()))
            self.interrupt = False
            self.interaction()
        except ConnectionRefusedError as e:
            print(\'\033[31;1m 连接中断,中断原因 [%s] ,中断时间:[%s] \033[0m\'
                  % (e, time.strftime(\'[%Y-%m-%d %H:%M:%S]\', time.localtime())))
            self.client.close()

    def com_get(self,com):
        os.chdir(setting.CLIENT_HOME)
        self.client.send(self.datahead(com, \'send\'))
        recv_data = self.client.recv(1024)
        recv_data = self.datahead(recv_data,\'recv\')
        if recv_data[\'error\'] == 777:
            filesize = recv_data[\'size\']
            file_recv = 0
            _md5 = hashlib.md5()
            file_obj = open(\'copy%d\'%random.randint(1,10000),\'wb\')
            respond = self.datahead(self.com_handle(\'rspond_ok\'),\'send\')
            self.client.send(respond)
            print(recv_data)
            while file_recv < filesize:
                if filesize - file_recv > 1024 :
                    size = 1024
                else:
                    size = filesize - file_recv
                data = self.client.recv(size)
                file_recv +=len(data)
                file_obj.write(data)
                file_obj.flush()
                _md5.update(data)
                percentage = float(file_recv / filesize) * 100
                print("%.2f"%percentage)
            else:
                file_obj.close()
                md5 = self.client.recv(size).decode()
                if md5 ==  _md5.hexdigest():
                    print(\'\033[32;1m 下载成功 ,时间:[%s] \033[0m\'
                          % (time.strftime(\'[%Y-%m-%d %H:%M:%S]\', time.localtime())))
                else:
                    print(\'\033[31;1m 文件传输过程中丢失数据！ ,时间:[%s] \033[0m\'
                          % (time.strftime(\'[%Y-%m-%d %H:%M:%S]\', time.localtime())))
        else:
            print(\'\033[31;1m 文件不存在，请重新输入! \033[0m\')
            self.interaction()



    def com_put(self,com):
        os.chdir(setting.CLIENT_HOME)
        if os.path.exists(com[\'filename\']):
            if os.path.isfile(com[\'filename\']):
                com[\'size\'] = os.stat(com[\'filename\']).st_size
                com[\'error\'] = 777
        else:
            print(\'\033[31;1m 文件不存在，请重新输入! \033[0m\')
            self.interaction()
        self.client.send(self.datahead(com,\'send\'))
        recv_data = self.client.recv(1024)
        print(recv_data)
        if recv_data == b\'rspond_ok\':
            file_obj=open(com[\'filename\'],\'rb\')
            for line in file_obj:
                self.client.send(line)
                time.sleep(1)
                print(1)
            print("上传完毕")



        #
        # if recv_data == b\'rspond_ok\':
        #     # recv_data = self.datahead(recv_data, \'recv\')
        #     send_data = {}
        #     com = com.strip().split()
        #     send_data[\'error\'] = 222;
        #     send_data[\'action\'] = com[0]
        #     if os.path.exists(com[1]):
        #         if os.path.isfile(com[1]):
        #             send_data[\'size\'] = os.stat(com[1]).st_size
        #             send_data[\'error\'] = 777;
        #             send_data[\'filename\'] = com[1];
        #     print(send_data)
        #     _send =\'upload\'+\' \'+ json.dumps(send_data)
        #     self.client.send(_send.encode())


        # recv_data = self.client.recv(1024)
        # recv_data = self.datahead(recv_data, \'recv\')

    def com_format(self,com_data):
        pass

    def interaction(self):
        while True:
                try:
                    com = input("%s>>:"%setting.CLIENT_HOME)
                    if len(com) == 0:
                        continue
                    com = self.com_handle(com)
                    antion = com[\'action\']
                    print(com)
                    if antion is not None and hasattr(self, \'com_%s\' % antion ):
                        func = getattr(self, \'com_%s\' % antion)
                        func(com)
                    else:
                        continue
                except (ConnectionRefusedError,ConnectionResetError )as e:
                    print(\'\033[31;1m 连接中断,中断原因 [%s] ,中断时间:[%s] \033[0m\'
                          % (e, time.strftime(\'[%Y-%m-%d %H:%M:%S]\', time.localtime())))
                    self.client.close()


    def com_handle(self,com):
        com = com.strip().split()
        com_dict ={}
        if len(com) > 1 :
            com_dict[\'action\'] =  com[0]
            com_dict[\'filename\'] =com[1]
        else:
            com_dict[\'action\'] = com[0]
            com_dict[\'filename\'] = \'no\'
        com_dict[\'size\'] = 0
        com_dict[\'error\'] = 222
        return com_dict


    def datahead(self,data,flag = \'else\'):
        if flag == \'send\':
            return json.dumps(data).encode()
        elif flag == \'recv\':
            return json.loads(data.decode())
        else:
            return data.encode()


if __name__ == \'__main__\':
    client = selectFTPClient()