python模拟简单ftp

需求：开发简单的FTP

用户登陆

上传/下载文件

不同用户家目录不同

查看当前目录下文件

充分使用面向对象知识
代码目录结构：



流程图：



客户端代码：

import os,sys
import getpass
import platform
if platform.system() == "Windows":
BASE_DIR = "\\".join(os.path.abspath(os.path.dirname(__file__)).split("\\")[:-1])
else:
BASE_DIR = "/".join(os.path.abspath(os.path.dirname(__file__)).split("/")[:-1])
sys.path.insert(0,BASE_DIR)
from modules import ssh

#print(sys.path)

if name == "main":
#host = input("请输入服务器端IP：").strip()
#port = input("请输入端口：").strip()
username = input("请输入用户名：").strip()
password = input("请输入密码：").strip()
#password = getpass.getpass("请输入密码：").strip()
ssh.SSH("host","port",username,password)

服务端代码：

import os,sys
import socket
import platform

if platform.system() == "Windows":
BASE_DIR = "\".join(os.path.abspath(os.path.dirname(file)).split("\")[:-1])
else:
BASE_DIR = "/".join(os.path.abspath(os.path.dirname(file)).split("/")[:-1])
sys.path.insert(0,BASE_DIR)
from modules import ftp

#从文件获取用户账号信息
def get_acount():
acount_dict = {}
f = open(BASE_DIR+"\conf\acount.conf")
for account in f:
acount_dict[account.split()[0]] = account.split()[1]
return acount_dict

if name == "main":
server = socket.socket()
server.bind(("localhost",10620))
server.listen(5)
print("准备接受连接.....")
while True:
conn, addr = server.accept()
auth = conn.recv(1024)
username = auth.decode("utf-8").split()[0]
password = auth.decode("utf-8").split()[1]
acount = get_acount()
if username in acount and password == acount[username]:
conn.send(b"success")
if os.path.exists(("C:\Users\%s" %username)):
home_dir = "C:\Users\%s" %username
conn.send(home_dir.encode("utf-8"))
else:
os.mkdir("C:\Users\%s" %username)
conn.send(home_dir)
os.chdir(home_dir)
conn.send("登录成功！命令帮助请输入h或？".encode("utf-8"))
help = '''
put [file_name] 上传本地文件到ftp服务器。例如：put file.txt
get [file_name] 下载ftp服务器文件到本地。例如：get file.txt
command 执行操作系统命令。例如:dir ipconfig
quit|exit|q 退出登录。例如：q
'''
while True:
command = conn.recv(1024).decode("utf-8")
if command == "?" or command == "h":
conn.send(help.encode("utf-8"))
elif command.split()[0] == "get":# or command.split()[0] == "put": #判断用户是否执行ftp命令
if len(command.split()) != 2:
conn.send(("%s命令格式为：%s FileName" %(command.split()[0],command.split()[0])).encode("utf-8"))
continue
else:
Ftp = ftp.FTP(conn,command.split()[0],command.split()[1]) #调用FTP模块传输文件
Ftp.download()
elif command.split()[0] == "put": # 判断用户是否执行ftp命令
f = open(command.split()[1], "wb")
fsize = conn.recv(1024).decode("utf-8")
conn.send(b"ok")
while True:
data = conn.recv(102400)
f.write(data)
f.flush()
if os.path.getsize(command.split()[1]) == int(fsize):
break
f.close()
elif command == "q" or command == "quit" or command == "exit":
conn.send("q".encode("utf-8"))
conn.close()
break
else:
res = os.popen(command).read()
if len(res) == 0: #如果是不存在的系统命令，则提醒用户输入错误
conn.send(("%s：command not found" % command).encode("utf-8"))
else: #以上条件都不符合后执行此步骤，此块内容为执行系统命令
conn.sendall(res.encode("utf-8"))
continue
else:
print(2222)
conn.close()
continue
server.close()

ssh模块：

import os,sys
import socket
import platform
if platform.system() == "Windows":
BASE_DIR = "\".join(os.path.abspath(os.path.dirname(file)).split("\")[:-1])
else:
BASE_DIR = "/".join(os.path.abspath(os.path.dirname(file)).split("/")[:-1])
sys.path.insert(0,BASE_DIR)
from modules import ftp

class SSH(object):
def init(self,host,port,username,password):
client = socket.socket()
self.host = host
self.port = port
self.username = username
self.password = password
#client.connect((self.host,int(self.port)))
client.connect(("localhost",10620))
#登录时将账号发送到服务端进行验证，验证通过后进入循环命令输入
#client.send((self.username+" "+self.password).encode("utf-8"))
client.send(("zhaohh" + " " + "123").encode("utf-8"))
auth_res = client.recv(1024)
if auth_res.decode("utf-8") == "success":
home_dir = client.recv(1024) #获取用户登录成功后的家目录
welcom = client.recv(1024)
print(welcom.decode("utf-8"))
while True:
command = input("[%s]$ "%home_dir.decode("utf-8")).strip()
if len(command) == 0:continue
client.send(command.encode("utf-8"))
if command.split()[0] == "get":
f = open(command.split()[1],"wb")
fsize = client.recv(1024).decode("utf-8")
client.send(b"ok")
while True:
data = client.recv(102400)
f.write(data)
f.flush()
if os.path.getsize(command.split()[1]) == int(fsize):
break
f.close()
elif command.split()[0] == "put":
if len(command.split()) != 2:
print("%s命令格式为：%s FileName" % (command.split()[0], command.split()[0]))
continue
else:
Ftp = ftp.FTP(client, command.split()[0], command.split()[1]) # 调用FTP模块传输文件
Ftp.upload()
else:
res = client.recv(102400)
if res.decode("utf-8") == "q":
exit("已退出登录！")
else:
print(res.decode())
else:
input("账号错误！")

ftp模块：

import os
class FTP(object):
def init(self,conn,command,filename):
self.command = command
self.filename = filename
self.conn = conn

#上传文件
def upload(self):
print("上传文件:%s" %self.filename)
f = open(self.filename,"rb")
data = f.read()
fsize = os.path.getsize(self.filename)
self.conn.send(str(fsize).encode("utf-8"))
self.conn.recv(1024)
self.conn.sendall(data)

#下载文件
def download(self):
f = open(self.filename,"rb")
data = f.read()
fsize = os.path.getsize(self.filename)
self.conn.send(str(fsize).encode("utf-8"))
self.conn.recv(1024)
self.conn.sendall(data)

用户账号文件内容格式：
zhaohh 123
alex 123