第七周作业——简单FTP

开发简单的FTP：

1. 用户登陆

2. 上传/下载文件

3. 不同用户家目录不同

4. 查看当前目录下文件

5. 充分使用面向对象知识

1、目录结构
zuoye-ftp
├── chenliang #用户chenliang的FTP根目录
├── download #所有用户下载的时候默认文件路径
├── lw #用户lw的FTP根目录
├── lzd #用户lzd的FTP根目录
├── wangjun #用户wangjun的FTP根目录
├── wdl #用户wdl的FTP根目录
├── xueli #用户xueli的FTP根目录
├── create-users.py #创建用户专用py文件
├── ros.zip #模拟上下载的测试文件ros.zip
├── socket_client.py #客户端文件
├── socket_server.py #服务端文件
├── user_data.bak #shelve文件
├── user_data.dat #shelve文件
└── user_data.dir #shelve文件

2、创建用户运行create-users.py，默认帐号密码是一致的。已有用户chenliang,lw,lzd,wangjun,wdl,xueli
3、创建的用户，在执行服务端的时候，自动会生成一个自己用户的FTP根目录，同时便于测试会复制ROS.zip到自己根目录
4、先运行socket_server.py，默认socket的服务端端口号为6969
5、再运行socket_client.py

create-users.py

#!/usr/bin/env python
#-*-coding:utf-8-*-
#Author:lzd
#创建用户名和密码，默认用户名和密码一致
import shelve
userlist=[]
user_data=shelve.open("user_data")
# for i in user_data.items():
#     print(i)
print("当前已有的所有用户的用户名如下：")
for i in user_data.keys():
    userlist.append(user_data[i])
print(userlist)
while True:
    username=input("请输入用户名\n")
    if username in userlist:
        print("用户名重复啦，请重新输入用户名!")
        continue
    else:
        password=input("请输入用户密码\n")
    break
user_data[username]=password
user_data.close()
print("用户创建成功~！")

socket_server.py

#!/usr/bin/env python
#-*-coding:utf-8-*-
#Author:lzd
#服务器端
import os
gongzuopath=os.getcwd()
# print(gongzuopath)
import time
#print(time.strftime("%Y%m%d%H%M%S",time.localtime()))
import socket
server = socket.socket()
server.bind(('localhost',6969)) #绑定要监听端口
server.listen() #监听
import shutil
import shelve
userlist=[]
user_data=shelve.open("user_data")
for i in user_data.keys():
    userlist.append(user_data[i])
    if not os.path.exists(gongzuopath+r"/"+i):
        os.mkdir(gongzuopath+r"/"+i)
    if not os.path.exists(gongzuopath+r"/"+i+"/ros.zip"):
        shutil.copyfile(gongzuopath+r"/ros.zip",gongzuopath+r"/"+i+"/ros.zip")
print("我要开始等电话了")
while True:
    Flag = True
    conn, addr = server.accept()  # 等电话打进来
    # conn就是客户端连过来而在服务器端为其生成的一个连接实例
    print(conn, addr)
    print("电话来了")
    while Flag:
        data = conn.recv(1024)
        print("recv:",data)
        if not data:
            print("client has lost...")
            break
        clientuser=data.decode("utf-8")
        if clientuser in userlist:
            conn.send("请输入用户密码".encode("utf-8"))
            data = conn.recv(1024)
            print("recv:",data)
            if not data:
                print("client has lost...")
                break
            if data.decode("utf-8")==user_data[clientuser]:
                conn.send("用户名密码验证成功".encode("utf-8"))
                while Flag:
                    data = conn.recv(1024)
                    print("recv:", data)
                    if data=="choose3".encode("utf-8"):
                        if ",".join(os.listdir(gongzuopath+r"/"+clientuser)).encode("utf-8"):
                            conn.send(",".join(os.listdir(gongzuopath+r"/"+clientuser)).encode("utf-8"))
                        else:
                            conn.send("当前目录下为空".encode("utf-8"))
                    elif data=="choose1".encode("utf-8"):
                        conn.send("服务端准备接收文件".encode())
                        data = conn.recv(1024).decode("utf-8") #收到文件大小，单位byte
                        filesize=data
                        print("recv:文件大小为", data)
                        tmpdata="服务端准备接收大小为"+data+"的文件"
                        conn.send(tmpdata.encode())
                        serverfile=gongzuopath+r"/"+clientuser+"/"+time.strftime("%Y%m%d%H%M%S",time.localtime())
                        f = open(serverfile, "wb")
                        while str(os.stat(serverfile).st_size) != filesize:
                            data=conn.recv(10240000)
                            f.write(data)
                            f.flush()
                        else:
                            conn.send("文件发送完毕".encode("utf-8"))
                    elif data=="choose2".encode("utf-8"):
                        if ",".join(os.listdir(gongzuopath + r"/" + clientuser)).encode("utf-8"):
                            conn.send(",".join(os.listdir(gongzuopath + r"/" + clientuser)).encode("utf-8"))
                        else:
                            conn.send("当前目录下为空".encode("utf-8"))
                            continue
                        data = conn.recv(1024).decode("utf-8")
                        print(data)
                        if not data:
                            continue
                        print("recv:客户端要下载的文件名", data)
                        fpath=gongzuopath+r"/"+clientuser+r"/"+data
                        fpath_size = str(os.stat(fpath).st_size)
                        f = open(fpath, "rb")
                        filedata = f.read()
                        conn.send(fpath_size.encode("utf-8"))
                        conn.sendall(filedata)
                        data = conn.recv(1024)
                        print(data.decode("utf-8"))  # 文件接收完毕
                    elif data=="choose4".encode("utf-8"):
                        Flag=False
            else:
                conn.send("用户名密码验证失败".encode("utf-8"))
        else:
            conn.send("输入的用户名不存在".encode())

server.close()

socket_client.py

#!/usr/bin/env python
#-*-coding:utf-8-*-
#Author:lzd
#客户端
import socket
import os
gongzuopath=os.getcwd()
# print(os.stat(gongzuopath+"/lzd/"+"ros.zip").st_size)
client = socket.socket() #声明socket类型，同时生成socket连接对象
client.connect(('localhost',6969))
while True:
    username=input("请输入用户名\n")
    if len(username) == 0: continue
    client.send(username.encode("utf-8"))
    data=client.recv(1024)
    print(data.decode())
    if data.decode()=="请输入用户密码":
        password=input("")
        client.send(password.encode("utf-8"))
        data=client.recv(1024)
        print(data.decode())
        if data.decode()=="用户名密码验证失败":
            continue
        elif data.decode()=="用户名密码验证成功":
            while True:
                print("%s当前的工作目录为/%s"%(username,username))
                print("""
                1、上传文件；
                2、下载文件；
                3、查看当前目录下的所有文件；
                4、退出程序
                请输入你想操作的数字？
                """)
                choose=input("")
                if choose=="":
                    print("当前用户的工作目录下所有文件如下：")
                    print(",".join(os.listdir(gongzuopath + r"/" + username)))
                    fpath2=input("请输入你想上传到服务器的文件名，为便于测试，当前目录下有个1M的文件ros.zip\n")
                    print("默认会在当前用户的目录下生成一个当前时间的文件。")
                    if not fpath2:
                        continue
                    fpath=gongzuopath+r"/"+username+r"/"+fpath2
                    fpath_size=str(os.stat(fpath).st_size)
                    f=open(fpath,"rb")
                    filedata=f.read()
                    client.send("choose1".encode("utf-8"))
                    data = client.recv(1024)
                    print(data.decode("utf-8")) #服务器准备接收文件
                    client.send(fpath_size.encode("utf-8"))
                    data = client.recv(1024)
                    print(data.decode("utf-8")) #服务端准备接收大小为%s的文件
                    client.sendall(filedata)
                    data = client.recv(1024)
                    print(data.decode("utf-8"))  # 文件发送完毕
                if choose=="":
                    client.send("choose2".encode("utf-8"))
                    data = client.recv(1024)
                    print("您当前的工作目录下，所有的文件如下：")
                    print(data.decode("utf-8"))  # 显示当前工作目录下所有文件
                    if data.decode("utf-8")=="当前目录下为空":
                        continue
                    print("您想下载哪个文件到本地？默认下载到当前脚本的工作目录的download目录")
                    fpath2 = input("请输入你想下载的文件名\n")
                    client.send(fpath2.encode("utf-8"))
                    fpath = gongzuopath + r"/download/" + fpath2
                    data = client.recv(1024).decode("utf-8")
                    file_size=data
                    print("您想下载的文件大小为：",data)
                    f = open(fpath, "wb")
                    while str(os.stat(fpath).st_size) != file_size:
                        data = client.recv(10240000)
                        f.write(data)
                        f.flush()
                    else:
                        client.send("文件接收完毕".encode("utf-8"))
                        print("文件接收完毕")
                if choose=="":
                    client.send("choose3".encode("utf-8"))
                    data = client.recv(1024)
                    print("您当前的工作目录下，所有的文件如下：")
                    print(data.decode("utf-8"))
                if choose=="":
                    client.send("choose4".encode("utf-8"))
                    client.close()
                    exit()