![从零开始学Python08作业源码：开发简单的FTP（仅供参考）]( "从零开始学Python08作业源码：开发简单的FTP（仅供参考）")
#!usr/bin/env python
# -*- coding:utf-8 -*-
# auther:Mr.chen
# 描述：
import socket
import os
import threading
import time
import json
from user_users import PersonInfo
DIR = os.path.dirname(os.path.abspath(__file__))
DIR = DIR+'/Folder/'
TAG = True
def tcplink(conn,addr):
"""
tcp请求分析函数
:param conn: tcp连接对象
:param addr: 连接地址
:return:
"""
print ("收到来自{0}的连接请求".format(addr))
conn.send('与主机通信中...')
while TAG:
try:
data = conn.recv(4096)
time.sleep(1)
if not data:
break
else:
print (data)
if data == 'ls':
P.view_file(conn)
continue
action,filename = data.strip().split()
action = action.lower()
if action == 'put':
re = P.Recvfile(conn,filename)
if re == True:
print ("文件接收成功！")
else:
print ("文件接收失败！")
elif action == 'get':
P.Sendfile(conn,filename)
elif action == 'login':
name, password = filename.split(',')
P = PersonInfo(name, password)
re = P.login()
if re == True:
conn.send('Ready!')
else:
conn.send('False!')
elif action == 'register':
name,password = filename.split(',')
P = PersonInfo(name, password)
re = P.register()
if re == True:
conn.send('Ready!')
else:
conn.send('False!')
else:
print ("请求方的输入有错！")
continue
except Exception,e:
print "tcplink处理出现问题",e
break
if __name__ == '__main__':
host = 'localhost'
port = 8888
s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
s.bind((host,port))
s.listen(5)
print ("服务运行中：正在监听{0}地址的{1}端口：".format(host,port))
while TAG:
# 接受一个新连接
conn,addr = s.accept()
# 创建一个新线程处理TCP连接
t = threading.Thread(target=tcplink,args=(conn,addr))
t.start()
#!usr/bin/env python
# -*- coding:utf-8 -*-
# auther:Mr.chen
# 描述：
import time
import os
import pickle
class PersonInfo:
"""
用户模型类
"""
DIR = os.path.dirname(os.path.abspath(__file__))
DIR = DIR + '/Folder/'
ConfigDir = DIR.replace('Folder','db')
def __init__(self,name,password):
self.Name = name #用户名
self.Password = password #密码
self.DIR = PersonInfo.DIR + self.Name +'/' #用户家目录
def login(self):
"""
用户登陆
:return:
"""
dict = PersonInfo.config_read()
if dict == None:
return False
if self.Name in dict:
if dict[self.Name] == self.Password:
return True
return False
def register(self):
"""
用户注册
:return:
"""
if os.path.exists(self.DIR) != True:
os.system('mkdir'+' '+ self.DIR)
try:
dict = PersonInfo.config_read()
if dict == None:
dict = {}
if self.Name not in dict:
dict[self.Name] = self.Password
else:
print ("姓名重复")
return False
re = PersonInfo.config_write(dict)
if re == True:
return True
except Exception,e:
print "注册出现异常！",e
return False
def view_file(self,conn):
"""
查看用户家目录
:param conn:
:return:
"""
data = os.popen('ls'+' '+ self.DIR).read()
conn.sendall(data)
def Recvfile(self,conn,filename):
"""
接收文件方法
:param conn:tcp连接对象
:param filename:目标文件名
:return:
"""
print ("开始接收文件...")
conn.send('Ready!')
buffer = []
while True:
d = conn.recv(4096)
if d == 'exit':
break
else:
buffer.append(d)
data = ''.join(buffer)
if data == '':
return False
print (data)
print (filename)
print (self.DIR)
with open(self.DIR + filename, 'w') as f:
f.write(data)
return True
def Sendfile(self,conn,filename):
"""
放送文件方法
:param conn: tcp连接对象
:param filename: 目标文件名
:return:
"""
if os.path.exists(self.DIR + filename):
print ("开始放送文件...")
conn.send('Ready!')
time.sleep(1)
with open(self.DIR + filename, 'r') as f:
while True:
data = f.read(4096)
print (data)
if not data:
break
conn.sendall(data)
time.sleep(1)
conn.send('exit')
print ("文件放送成功！")
else:
conn.send('False!')
@staticmethod
def config_read():
"""
配置文件全部读取
:return:
"""
if os.path.exists(PersonInfo.ConfigDir+'user_config'):
with open(PersonInfo.ConfigDir+'user_config','r') as f:
dict = pickle.load(f)
return dict
@staticmethod
def config_write(dict):
"""
配置文件全部写入
:param dict:
:return:
"""
with open(PersonInfo.ConfigDir + 'user_config', 'w') as f:
pickle.dump(dict,f)
return True
#!usr/bin/env python
# -*- coding:utf-8 -*-
# auther:Mr.chen
# 描述：
import socket,os
import time
TAG =True
DIR = os.path.dirname(os.path.abspath(__file__))
DIR = DIR+'/Folder/'
HOST = 'localhost'
PORT = 8888
def Recvfile(s,filename):
"""
接收文件方法函数
:param s: 套接字封装对象
:param filename: 目标文件名
:return:
"""
print ("开始下载文件...")
buffer = []
while TAG:
d = s.recv(4096)
if d == 'exit':
break
buffer.append(d)
data = ''.join(buffer)
with open(DIR+filename,'w') as f:
f.write(data)
print ("文件下载完毕！")
def Sendfile(s,filename):
"""
放送文件方法函数
:param s: 套接字封装对象
:param filename: 目标文件名
:return:
"""
print ("开始上传文件！")
if os.path.exists(DIR+filename):
with open(DIR+filename,'r') as f:
while TAG:
data = f.read(4096)
if not data:
break
s.sendall(data)
time.sleep(1)
s.send('exit')
print ("文件上传完毕")
else:
print ("你的目录里没有这个文件")
time.sleep(1)
s.send('exit')
def Confirm(s,command):
"""
验证与服务器连接是否正常；
把用户命令发过去，让服务器做好相应准备准备
:param s: 套接字封装对象
:param command: 用户输入的命令
:return:
"""
s.sendall(command)
re = s.recv(4096)
if re == 'Ready!':
return True
elif re == 'False!':
return False
else:
print ("与服务器连接出现异常！")
def File_transfer(s):
"""
用户指令函数
:param s:
:return:
"""
while TAG:
command = raw_input("请输入你想执行的命令>>")
if not command:
continue
if command.lower().strip() == 'help':
print ("请用'put'+'空格'+'文件名'的格式上传文件")
print ("请用'get'+'空格'+'文件名'的格式下载文件")
print ("输入'ls'查看用户服务器家目录")
continue
if command.lower().strip() == 'ls':
s.send('ls')
data = s.recv(4096)
print (data)
continue
try:
action,filename = command.strip().split()
action = action.lower()
except:
print ("您的输入有误!输入help查看帮助文档")
continue
if action == 'put':
re = Confirm(s,command)
if re == True:
Sendfile(s, filename)
else:
print ("对方服务器没有准备好！")
break
elif action == 'get':
re = Confirm(s,command)
if re == True:
Recvfile(s, filename)
elif re == False:
print ("服务器家目录没有这个文件")
else:
print ("对方服务器没有准备好！")
break
else:
print ("你输入的命令有误！输入help查看帮助文档")
def Login(s):
"""
用户登录
:param s:
:return:
"""
name = raw_input("请输入你的用户名：")
password = raw_input("请输入你的密码：")
command = 'login'+' '+ name + ',' + password
re = Confirm(s, command)
if re == True:
print ("登陆成功！")
File_transfer(s)
elif re == False:
print ("您的输入有误，请重新输入！")
Login(s)
else:
print ("与服务器连接出现异常！")
def Register(s):
"""
用户注册
:param s:
:return:
"""
name = raw_input("请输入你的用户名：")
password = raw_input("请输入你的密码：")
Password = raw_input("请再次输入密码：")
if password != Password:
print ("你的密码两次输入不一致，请重新输入！")
Register(s)
command = 'register' + ' ' + name + ',' + password
print (command)
re = Confirm(s,command)
if re == True:
File_transfer(s)
elif re == False:
print ("用户名重复，请重新输入！")
Register(s)
else:
print ("与服务器连接出现异常！")
def Main(s,log = '未联通主机...'):
"""
用户登陆界面
:param s:
:param log:
:return:
"""
text = """
用户登陆界面 {0}
1，用户登陆
2，用户注册
""".format(log)
print (text)
choose = raw_input("请输入索引进行选择：")
if choose == '':
Login(s)
elif choose == '':
Register(s)
else:
print ("你的选择有误！")
if __name__ == "__main__":
s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
try:
s.connect((HOST,PORT))
Main(s,s.recv(1024))
except Exception,e:
print "服务器连接不上....",e
finally:
s.close()
