python 学习分享-实战篇简单的ftp

import socket
import os
import time
import pickle
Basedb = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
class client_ftp(object):
    #客户端类
    def __init__(self,add,port):
        #客户端属性，地址和端口
        self.add = add
        self.port = port
        #注册用户
    def register(self):
        username = input('请输入用户名：').strip()
        password = input('请输入密码：').strip()
        if not username or  not password:
            print('输入为空！')
        elif username  in os.listdir('%s/db'%Basedb):
            print('用户已存在！')
        else:
            save = {'username':username,'password':password}
            db_path = Basedb + '/db/' + username
            #存入用户信息
            pickle.dump(save,open(db_path,'wb'))
            #生成用户文件夹
            os.mkdir(Basedb + '/client_file/' + username)
            print('注册成功！')

    def login_in(self):
        #登陆验证方法
        username = input('请输入用户名：').strip()
        password = input('请输入密码：').strip()
        if username in os.listdir('%s/db'%Basedb):
            db_path = Basedb + '/db/' + username
            db_dict = pickle.load(open(db_path,'rb'))
            if password == db_dict['password']:
                print('登陆成功!')
                flag = True
            else:
                print('密码错误！')
                flag = False
        else:
            print('用户名错误！')
            flag = False
        return flag
    def conn_server(self):
        #连接方法
        client = socket.socket()
        client.connect((self.add,self.port))
        code_in = input('请输入用户名：').strip()
        client.send(code_in.encode('UTF-8'))
        data = client.recv(1024000).decode()
        print(data)
        #如果返回成功
        if data == 'success':
            server_file = os.listdir(Basedb + '/client_file/' + code_in)
            send_file = os.listdir(Basedb + '/send')
            while 1:
                option = input('''
                -----option-----
                1.输入1获取文件目录；
                2.直接输入文件名称下载文件
                3.将文件拖至send文件夹中，并输入文件名，上传到服务器
                4.退出
                ''').strip()
                if option == '':
                    client.send('check'.encode('utf-8'))
                    print(client.recv(1024000).decode())
                elif option in server_file:
                    client.send(option.encode('utf-8'))
                    data1 = client.recv(1024000).decode()
                    file_name = Basedb + '/send'
                    time_now =time.strftime("%Y-%m-%d-%H-%M-%S",time.localtime(time.time()))
                    with open('%s/%s'%(file_name,time_now),'w') as f:
                        f.write(data1)
                    print('接收成功！')
                elif option in send_file:
                    client.send('load'.encode('utf-8'))
                    print(client.recv(1024000).decode())
                    f_open = open('%s/send/%s'%(Basedb,option),'r')
                    back_data = f_open.read()
                    client.send(str(back_data).encode('utf-8'))
                    f_open.close()
                    print('传输成功!')
                else:
                    exit()

if __name__ == '__main__':
    client_start = client_ftp('localhost',6969)
    while 1:
        option = input('''
        --------option--------
        1.注册
        2.登陆
        3.退出
        ''').strip()
        if option == '':
            client_start.register()
        elif option =='':
            flag = client_start.login_in()
            if flag== True :
                client_start.conn_server()
            else:
                print('登陆失败了哥们')
                break
        else:
            exit()

import socket
import os
import time
Base_path = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
class Server_ftp(object):
    #此类为ftp服务端
    def __init__(self,add,port):
        self.add = add
        self.port = port
    def conn_listen(self):
        server = socket.socket()
        server.bind((self.add,self.port))
        server.listen(5)
        print('监听正在进行！监听地址为：',self.add,'监听端口为：',self.port)
        while 1:
            conn,addr = server.accept()
            print('已建立连接!')
            data = conn.recv(1024000).decode()
            print("recv:", data)
            if not data:
                print("client has lost...")
                break
            elif os.path.exists(Base_path + '/client_file/' + data):
                allfile = os.listdir(Base_path + '/client_file/' + data)
                conn.send('success'.encode('utf-8'))
                while 1:
                    data1 = conn.recv(1024000).decode()
                    if not data1:
                        print("client has lost...")
                        break
                    elif data1 == 'check':
                        conn.send(str(allfile).encode('utf-8'))
                        break
                    elif data1 in allfile:
                        file_name = Base_path + '/client_file/' + data
                        f_open = open('%s/%s'%(file_name,data1),'rb')
                        back_data = f_open.read()
                        conn.send(str(back_data).encode('utf-8'))
                        f_open.close()
                        break
                    elif data1 == 'load':
                        conn.send('reading'.encode('utf-8'))
                        data2 = conn.recv(1024000).decode()
                        if not data2:
                            print("client has lost...")
                            break
                        else:
                            file_name = Base_path + '/client_file/' + data
                            time_now =time.strftime("%Y-%m-%d-%H-%M-%S",time.localtime(time.time()))
                            with open('%s/%s'%(file_name,time_now),'w') as f:
                                f.write(data2)
                            print('接收成功！')
                            conn.send('接收成功！'.encode('utf-8'))
                            break
                    else:
                        print('data1输入错误!')
                        conn.send('data1输入错误！'.encode('utf-8'))
            else:
                print('接收错误！')
                conn.send('错误输入！'.encode('utf-8'))
                break

if __name__ == '__main__':
    sever_start = Server_ftp('localhost',6969)
    sever_start.conn_listen()