python第八周测试_Python学习笔记第八周

目录：

一、基础概念

1、动态导入模块

2、断言机制

3、SocketServer

二、作业 FTP服务器

三、单独注明

1、类中装饰器

2、进度条

一、基础概念

1、动态导入模块

有时在编程过程中，如果知道模块对应的字符串，可以通过动态导入的方式将该字符对应的模块进行动态导入

导入的方式有两种：

1、python解释器默认的方式导入

2、官方推荐动态导入方式

下面来分别介绍以下两种方式：

lib模块中包括aa文件，aa文件内容如下：

classC(object):def __init__(self):

self.name= 'gavin'

通过方法一导入：

modname = 'lib' #知道模块名字对应字符串

mod= __import__('lib.aa') #导入lib同时导入lib模块下的aa,但是这个mod只是lib,而不是lib.aa 要记住

obj= mod.aa.C()#因为mod只表示lib,所以需要写成mod.aa表示lib.aa

print(obj.name)

通过方法二导入：

importimportlib

mod= importlib.import_module('lib.aa') #官方建议这种方法更有效,mod直接表示了lib.aa

obj =mod.C()print(obj.name)

2、断言机制

断言的作用：

当后续的程序很重要，不允许出错，可以在该程序前添加断言机制，当断言不正确就不执行，断言正确会继续执行后续的程序

例子：

借用上面的lib模块引用，当引用该模块后，发现其中的类型为str类型后，打印，然后将类型断言变为int类型，此时会发现原来正常打印执行的语句不再执行

1、正确打印：

importimportlib

mod_name= 'lib.aa'mod=importlib.import_module(mod_name)

obj=mod.C()assert type(obj.name) isstrprint('断言正确执行')#输出

断言正确执行

2、忽略打印动作

importimportlib

mod_name= 'lib.aa'mod=importlib.import_module(mod_name)

obj=mod.C()assert type(obj.name) isintprint('断言正确执行')#输出

raceback (most recent call last):

File"/Users/Gavin/PycharmProjects/python/day8/复现/2-断言/断言.py", line 7, in

assert type(obj.name) isint

AssertionError

3、socketserver

socketserver是一个模块，该模块简化了编写网络服务器的任务复杂度

socketserver一共有这么几种类型

1、tcp类型

class socketserver.TCPServer(server_address, RequestHandlerClass,bind_and_activate=True)

2、udp类型

class socketserver.UDPServer(server_address, RequestHandlerClass, bind_and_activate=True)

3、使用Unix domain方式socket类型

class socketserver.UnixStreamServer(server_address, RequestHandlerClass, bind_and_activate=True)class socketserver.UnixDatagramServer(server_address, RequestHandlerClass,bind_and_activate=True)

这几种类的继承关系如下：

+------------+

| BaseServer |

+------------+

|

v

+-----------+ +------------------+

| TCPServer |------->| UnixStreamServer |

+-----------+ +------------------+

|

v

+-----------+ +--------------------+

| UDPServer |------->| UnixDatagramServer |

+-----------+ +--------------------+

创建一个socketserver至少分以下几个步骤：

1、必须创建一个请求处理类，这个要继承BaseRequestHandler类并且要重写父类里的handle方法

2、必须要实例化一个其中任意类型的类，同时传递Server IP和上面创建的请求处理类给这个任意类

3、调用类方法

类方法包含以下两种：

1、server.serve_request() #只处理一个请求，不会使用这个

2、server.serve_forever()#处理多个请求，永远执行着调用

4、关闭server.serve_forever()

注意：跟客户端所有交互，都是在handle()方法中完成的，每个请求过来以后，都是通过handle来规定处理行为

基本的sockerserver代码

importsocketserverclassMyTCPHandler(socketserver.BaseRequestHandler):"""The request handler class for our server.

It is instantiated once per connection to the server, and must

override the handle() method to implement communication to the

client."""

defhandle(self):#self.request is the TCP socket connected to the client

self.data = self.request.recv(1024).strip()print("{} wrote:".format(self.client_address[0]))print(self.data)#just send back the same data, but upper-cased

self.request.sendall(self.data.upper())if __name__ == "__main__":

HOST, PORT= "localhost", 9999

#Create the server, binding to localhost on port 9999

server =socketserver.TCPServer((HOST, PORT), MyTCPHandler)#Activate the server; this will keep running until you

#interrupt the program with Ctrl-C

server.serve_forever()

View Code

在上面代码，仍然不能同时处理多个链接，如果要处理多个链接，需要将socketserver.TCPServer变为以下类：

classsocketserver.ForkingTCPServer

classsocketserver.ForkingUDPServer

classsocketserver.ThreadingTCPServer

classsocketserver.ThreadingUDPServer

所以只需要替换

server = socketserver.TCPServer((HOST, PORT), MyTCPHandler)

变为下面这个，就可以多并发了

server = socketserver.ThreadingTCPServer((HOST, PORT), MyTCPHandler)

classsocketserver.BaseServer(server_address, RequestHandlerClass) 主要有以下方法

classsocketserver.BaseServer(server_address, RequestHandlerClass)

Thisis the superclass of all Server objects in the module. It defines the interface, given below, but does not implement most of the methods, which is done in subclasses. The two parameters are stored in the respective server_address andRequestHandlerClass attributes.

fileno()

Return an integer file descriptorfor the socket on which the server is listening. This function is most commonly passed to selectors, to allow monitoring multiple servers inthe same process.

handle_request()

Process a single request. This function calls the following methodsin order: get_request(), verify_request(), and process_request(). If the user-provided handle() method of the handler class raises an exception, the server’s handle_error() method will be called. If no request is received within timeout seconds, handle_timeout() will be called and handle_request() will return.

serve_forever(poll_interval=0.5)

Handle requests until an explicit shutdown() request. Pollfor shutdown every poll_interval seconds. Ignores the timeout attribute. It also calls service_actions(), which may be used by a subclass or mixin to provide actions specific to a given service. For example, the ForkingMixIn classuses service_actions() to clean up zombie child processes.

Changedin version 3.3: Added service_actions call to the serve_forever method.

service_actions()

Thisis called in the serve_forever() loop. This method can be overridden by subclasses ormixin classes to perform actions specific to a given service, such as cleanup actions.

Newin version 3.3.

shutdown()

Tell the serve_forever() loop to stopandwait until it does.

server_close()

Clean up the server. May be overridden.

address_family

The family of protocols to which the server’s socket belongs. Common examples are socket.AF_INETandsocket.AF_UNIX.

RequestHandlerClass

The user-provided request handler class; an instance of this class is created foreach request.

server_address

The address on which the serveris listening. The format of addresses varies depending on the protocol family; see the documentation for the socket module for details. For Internet protocols, this is a tuple containing a string giving the address, and an integer port number: ('127.0.0.1', 80), forexample.

socket

The socket object on which the server will listenforincoming requests.

The server classes support the followingclassvariables:

allow_reuse_address

Whether the server will allow the reuse of an address. This defaults to False,and can be set insubclasses to change the policy.

request_queue_size

The size of the request queue. If it takes a long time to process a single request, any requests that arrivewhile the server is busy are placed into a queue, up to request_queue_size requests. Once the queue is full, further requests from clients will get a “Connection denied” error. The default value is usually 5, but this can be overridden by subclasses.

socket_type

The type of socket used by the server; socket.SOCK_STREAMandsocket.SOCK_DGRAM are two common values.

timeout

Timeout duration, measuredin seconds, or None if no timeout is desired. If handle_request() receives no incoming requests within the timeout period, the handle_timeout() method iscalled.

There are various server methods that can be overridden by subclasses of base server classes like TCPServer; these methods aren’t useful to external users of the server object.

finish_request()

Actually processes the request by instantiating RequestHandlerClassandcalling its handle() method.

get_request()

Must accept a requestfrom the socket, and return a 2-tuple containing the new socket object to be used to communicate with the client, andthe client’s address.

handle_error(request, client_address)

This functionis called if the handle() method of a RequestHandlerClass instance raises an exception. The default action is to print the traceback to standard output and continuehandling further requests.

handle_timeout()

This functionis called when the timeout attribute has been set to a value other than None and the timeout period has passed with no requests being received. The default action for forking servers is to collect the status of any child processes that have exited, while inthreading servers this method does nothing.

process_request(request, client_address)

Calls finish_request() to create an instance of the RequestHandlerClass. If desired, this function can create a new processor thread to handle the request; the ForkingMixIn andThreadingMixIn classes do this.

server_activate()

Called by the server’s constructor to activate the server. The default behaviorfora TCP server just invokes listen() on the server’s socket. May be overridden.

server_bind()

Called by the server’s constructor to bind the socket to the desired address. May be overridden.

verify_request(request, client_address)

Mustreturn a Boolean value; if the value is True, the request will be processed, and if it’s False, the request will be denied. This function can be overridden to implement access controls for a server. The default implementation always returns True.

View Code

二、作业

开发一个支持多用户在线的FTP程序

要求：

用户加密认证

允许同时多用户登录

每个用户有自己的家目录 ，且只能访问自己的家目录

对用户进行磁盘配额，每个用户的可用空间不同

允许用户在ftp server上随意切换目录(切换目录命令参数只包括 '.'   '..' 和绝对路径)

允许用户查看当前目录下文件

允许上传和下载文件，保证文件一致性

文件传输过程中显示进度条

附加功能：支持文件的断点续传(没实现，可以通过抓取异常处理，然后通过tell记录字节数，下一次传递通过seek到记录的字节数开始继续传输来实现

目录结构：

代码实现：

FTP_Client:

importosimportsys

BASEDIR= os.path.dirname(os.path.dirname(os.path.abspath(__file__)))print(BASEDIR)

sys.path.insert(0,BASEDIR)from core importmainif __name__ == '__main__':#HOST = input('输入你链接的地址:').strip()

#PORT = int(input('输入你要链接的端口:').strip())

obj = main.FtpClient('localhost',9999)

obj.FtpManager()

ftp_client.py

importosimportsysimportlogging

BASEDIR= os.path.dirname(os.path.dirname(os.path.abspath(__file__)))

sys.path.insert(0,BASEDIR)

USER_HOME= os.path.join(BASEDIR,'user_data/')

settings

importsocketimportosimportsysimporthashlibimportjson

BASEDIR= os.path.dirname(os.path.dirname(os.path.abspath(__file__)))

sys.path.insert(0,BASEDIR)from config importsettingsclassFtpClient(object):def __init__(self,host,port): #构造函数,初始化时可以指定服务器IP与端口

self.host =host

self.port=port

self.client=socket.socket()def help(self): #认证成功后,显示该客户端功能菜单

msg ={'cd': 'cd .|cd .. |cd abspath 三种方式切换路径','ls': '显示目录内容','pwd': '显示当前路径','put filename': '上传文件','get filename': '下载文件','quit': '退出程序',

}for line inmsg:print(line, msg[line])defFtpManager(self):

self.client.connect((self.host,self.port))whileTrue:

username= input('\033[32;1m输入用户名:\033[0m').strip()if username != 'q':

password= input('\033[32;1m输入密码:\033[0m').strip().encode('utf-8')

password_md5=hashlib.md5()

password_md5.update(password)#对密码进行加密

user_dic ={"username": username,"password": password_md5.hexdigest()

}

self.client.send(json.dumps(user_dic).encode('utf-8'))

server_auth_ack= int(self.client.recv(1024).decode())if server_auth_ack == 200: #认证成功后进入菜单

whileTrue:

getattr(self,'help')()

cmd_list= ['ls', 'pwd', 'cd']

cmd= input('#').strip()if len(cmd) ==0:continue

elif cmd.split()[0] incmd_list:

func= getattr(self,'interactive')

break_flag=func(cmd)if break_flag == 202:break

elifhasattr(self,cmd.split()[0]):

func=getattr(self,cmd.split()[0])

break_flag=func(cmd)if break_flag == 202:break

else:print('\033[31;1m没有这个选项,请重新输入\033[0m')elif server_auth_ack == 400:print('\033[31;1m用户名错误,请重新输入\033[0m')elif server_auth_ack == 401:print('\033[31;1m密码错误,请重新输入\033[0m')else:

exit('\033[31;1m退出程序\033[0m')def quit(self,*args):whileTrue:

cmd_split=args[0].split()

msg_dic={'action': 'quit'}

self.client.send(json.dumps(msg_dic).encode('utf-8'))

quit_ack= self.client.recv(1024).decode()if quit_ack == '201':

exit('\033[31;1m退出程序\033[0m')else:continue

def put(self,*args):

cmd_split=args[0].split()if len(cmd_split) > 1:

filename= cmd_split[1]

File_Name= settings.USER_HOME +filenameifos.path.isfile(File_Name):

File_size=os.stat(File_Name).st_size

msg_dic={'action': 'put','filename': filename,'filesize': File_size,'overridden': True

}

self.client.send(json.dumps(msg_dic).encode('utf-8'))

server_response= int(self.client.recv(1024).decode())if server_response == 200:

with open(File_Name,'rb') as f:for line inf:

self.client.send(line)

put_done_ack= self.client.recv(1024).decode()print('\033[31;1m%s :传输完毕\033[0m' %put_done_ack)elif server_response == 500:print('\033[31;1m该用户没有上传权限\033[0m')elif server_response == 300:print('\033[31;1m该文件已存在\033[0m')elif server_response == 501:print('\033[31;1m磁盘配额已用完\033[0m')elif server_response == 202:print('\033[31;1m没有登录,请重新登录\033[0m')return 202

else:print('\033[31;1m该文件不存在\033[0m')else:print('\033[31;1m命令错误请重新输入\033[0m')def get(self,*args):

cmd_split=args[0].split()if len(cmd_split) > 1:

filename= cmd_split[1]

File_Name= settings.USER_HOME +filenameifos.path.isfile(File_Name):print('\033[31;1m该文件已存在\033[0m')else:

msg_dic={'action': 'get','filename': filename

}

self.client.send(json.dumps(msg_dic).encode('utf-8'))

server_response= int(self.client.recv(1024).decode())if server_response == 200:

self.client.send('200'.encode('utf-8'))

msg_recv= json.loads(self.client.recv(1024).decode())

self.client.send('200'.encode('utf-8'))

filesize= msg_recv['filesize']

recv_size=0

with open(File_Name,'wb') as f:while recv_size 1024:

size= 1024

else:

size= filesize -recv_size

data=self.client.recv(size)

f.write(data)

f.flush()

recv_size+=len(data)

pri_size= int(recv_size/filesize*100)

pri_fun= pri_size +1str= '>'*int(pri_size/2) + ' '*int((100-pri_fun)/2)

sys.stdout.write('\r' + str + '%s%%' %pri_size)

sys.stdout.flush()

self.client.send('200'.encode('utf-8'))print('')print('\033[31;1m传输完毕\033[0m')elif server_response == 300:print('\033[31;1m文件在服务器上不存在\033[0m')elif server_response == 202:print('\033[31;1m没有登录,请重新登录\033[0m')return 202

else:print('\033[31;1m命令错误请重新输入\033[0m')def interactive(self,*args):

cmd_split=args[0].split()if len(cmd_split) >1:if len(cmd_split) < 3:

msg_dic={'action': cmd_split[0],'dict': cmd_split[1]

}

self.client.send(json.dumps(msg_dic).encode('utf-8'))

ack_flag= int(self.client.recv(1024).decode())if ack_flag == 200:

self.client.send('200'.encode('utf-8'))

pwd_res= self.client.recv(1024).decode()print(pwd_res)elif ack_flag == 202:print('\033[31;1m用户没有登录\033[0m')return 202

else:print('\033[31;1m命令错误请重新输入\033[0m')elif len(cmd_split) == 1:

msg_dic={'action': cmd_split[0]

}

self.client.send(json.dumps(msg_dic).encode('utf-8'))

ack_flag= int(self.client.recv(1024).decode())if ack_flag == 200:

self.client.send('200'.encode('utf-8'))

pwd_res= json.loads(self.client.recv(1024).decode())ifisinstance(pwd_res, list):for line inpwd_res:print(line)elifisinstance(pwd_res,str):print(pwd_res)elif ack_flag == 202:print('\033[31;1m用户没有登录\033[0m')return 202

main.py

FTP_Server:

importosimportsysimportsocketserver

BASEDIR= os.path.dirname(os.path.dirname(os.path.abspath(__file__)))

sys.path.insert(0,BASEDIR)from core importmainif __name__ == "__main__":

HOST, PORT= "localhost", 9999

#Create the server, binding to localhost on port 9999

server =socketserver.ThreadingTCPServer((HOST, PORT), main.MyTCPHandler)

server.serve_forever()

ftp_server.py

importosimportsys

BASEDIR= os.path.dirname(os.path.dirname(os.path.abspath(__file__)))

sys.path.insert(0,BASEDIR)

DATABASE_USER={'path': '%s/db' %BASEDIR

}

USER_HOME={'path': '%s/home' %BASEDIR

}

USER_TEMP={'path': '%s/temp' %BASEDIR

}

settings.py

importosimportsysimportsocketserverimportjsonimporthashlibimportreimporttime

BASEDIR= os.path.dirname(os.path.dirname(os.path.abspath(__file__)))

sys.path.insert(0,BASEDIR)from config importsettingsclassMyTCPHandler(socketserver.BaseRequestHandler):def auth(self,*args):

user_auth= '%s/%s.json' %(settings.DATABASE_USER['path'], args[0]['username']) #导入用户存在目录

if os.path.isfile(user_auth): #判断该目录是否存在

user_passwd =hashlib.md5()

with open(user_auth,'r') as f:

user_dic=json.load(f)

user_passwd.update(user_dic['password'].encode('utf-8')) #hashlib加密需要提前做encode

if user_passwd.hexdigest() == args[0]['password']:

user_dic['authenticated'] = 1with open(user_auth,'w') as f:

json.dump(user_dic,f)return 200

else:return 401

else:return 400

def handle(self): #socketserver模块中处理请求的方法

whileTrue:try:

user_data= json.loads(self.request.recv(1024).decode()) #接受到用户发送过来的json用户名与密码

auth_code = getattr(self,'auth')(user_data) #通过auth方法处理

self.request.send(str(auth_code).encode('utf-8'))if auth_code != 200:continue #验证不成功重新验

whileTrue:

data= self.request.recv(1024) #当验证成功后接收客户端指令

cmd_dic =json.loads(data.decode())

action= cmd_dic['action'] #提取客户端发送过来的指令

ifhasattr(self,action):

func= getattr(self,action) #根据指令反射对应方法

func(cmd_dic,user_data)exceptConnectionResetError as e:print("err",e)break

def check_auth(func): #装饰器,用于验证用户是否登录,

def deco(self,*args):

user_auth= '%s/%s.json' %(settings.DATABASE_USER['path'], args[1]['username'])

with open(user_auth,'r') as f:

user_dic=json.load(f)if user_dic['authenticated'] == 1:#在装饰器中,如果用户已登录会在登录函数auth中将authenticated置为1

func(self,*args)else:print('not authed')

self.request.send('202'.encode('utf-8'))returndeco

@check_authdef put(self,*args):'''接受客户端文件'''cmd_dic=args[0]

filename= cmd_dic['filename']

filesize= cmd_dic['filesize']

user_dic= args[1]

user_db= self.user_home(user_dic) #在操作前判断用户当前所在目录

#BASE_HOME = '%s/%s/' %(settings.USER_HOME['path'], args[1]['username']) #确定客户端对应服务器所在目录

user_auth = '%s/%s.json' %(settings.DATABASE_USER['path'], args[1]['username']) #确定用户认证文件,里面包括用户名与磁盘配额信息

for key inuser_db:

user_db_value=user_db[key]

with open(user_auth,'r') as f:

user_dic=json.load(f)if user_dic['allow_update'] == True: #判断用户是否允许上传文件

quato = user_dic['free_disk']if filesize < quato: #需要查看用户磁盘配额

if os.path.isfile(user_db_value + filename): #判断文件是否存在

self.request.send('300'.encode('utf-8'))else:

self.request.send('200'.encode('utf-8'))

received_size=0

with open(user_db_value+ filename, 'wb') as f:'''根据文件大小准确写入文件内容'''

while received_size 1024:

size= 1024

else:

size= filesize -received_size

data=self.request.recv(size)

f.write(data)

f.flush()

received_size+=len(data)

pri_size= int(received_size/filesize*100)

pri_fun= pri_size +1str= '>'*int(pri_size/2) + ' '*int((100-pri_fun)/2)

send_result= str + '%s%%' %pri_size

sys.stdout.write('\r' + str + '%s%%' %pri_size)

sys.stdout.flush()

self.request.send('200'.encode('utf-8'))

user_dic['free_disk'] = quato - filesize #磁盘配额计算

with open(user_auth,'w') as f: #新磁盘配额重新写入

json.dump(user_dic,f)else:

self.request.send('501'.encode('utf-8'))else:

self.request.send('500'.encode('utf-8'))

@check_authdef get(self,*args):'''发生文件给客户端'''cmd_dic=args[0]

filename= cmd_dic['filename']

user_dic= args[1]

user_db= self.user_home(user_dic) #判断当前所在目录

#BASE_HOME = '%s/%s/' %(settings.USER_HOME['path'], args[1]['username'])

user_auth = '%s/%s.json' %(settings.DATABASE_USER['path'], args[1]['username'])for key inuser_db:

user_db_value=user_db[key]if not os.path.isfile(user_db_value + '/'+ filename): #如果文件不存在,返回参数

self.request.send('300'.encode('utf-8'))else:

self.request.send('200'.encode('utf-8'))

self.request.recv(1024)

filesize= int(os.stat(user_db_value + '/' +filename).st_size)

msg_dic={'filename': filename,'filesize': filesize

}

self.request.send(json.dumps(msg_dic).encode('utf-8')) #发送文件和大小

self.request.recv(1024)

with open(BASE_HOME+ filename,'rb') as f:for line inf:

self.request.send(line)

get_done_ack= self.request.recv(1024)print('\033[31;1m%s: 传输完毕\033[0m' %get_done_ack)def user_home(self,*args): #判断当前所在目录函数

'''获取用户base_home信息'''user_auth= '%s/%s.json' %(settings.DATABASE_USER['path'], args[0]['username'])

user_temp= '%s/%s.json' %(settings.USER_TEMP['path'], args[0]['username'])if os.path.isfile(user_temp): #当user_temp文件存在时,会先读取该文件所在内容

with open(user_temp,'r') as f:

temp_dic=json.load(f)else:

with open(user_auth,'r') as f:

user=json.load(f)

temp_dic= user['Base_Home']returntemp_dicdef quit(self,*args):'''在程序退出时,会将装饰器中查看字段修改回0,同时删除用户跳转目录的记录文件'''user_auth= '%s/%s.json' %(settings.DATABASE_USER['path'], args[1]['username'])

with open(user_auth,'r') as f:

user_dic=json.load(f)

user_dic['authenticated'] = 0 #装饰器置为0

with open(user_auth,'w') as f:

json.dump(user_dic,f)

self.request.send('201'.encode('utf-8'))

user_temp= '%s/%s.json' %(settings.USER_TEMP['path'], args[1]['username'])ifos.path.isfile(user_temp):

os.remove(user_temp)#删除用户操作记录文件

@check_authdef pwd(self,*args):'''查看当前目录'''cmd_dic=args[0]

user_dic= args[1]

user_db= self.user_home(user_dic) #先查看当前目录在哪个位置

print(user_db)

user_auth= '%s/%s.json' %(settings.DATABASE_USER['path'], args[1]['username'])

user_temp= '%s/%s.json' %(settings.USER_TEMP['path'], args[1]['username'])

self.request.send('200'.encode('utf-8'))

client_ack= int(self.request.recv(1024).decode())if client_ack == 200:for key inuser_db:

user_db_key=key

user_db_key=json.dumps(user_db_key)

self.request.send(user_db_key.encode('utf-8'))else:

self.request.send('203'.encode('utf-8'))

@check_authdef ls(self,*args):'''查看当前目录'''cmd_dic=args[0]

user_dic= args[1]

user_db=self.user_home(user_dic)

user_auth= '%s/%s.json' %(settings.DATABASE_USER['path'], args[1]['username'])

user_temp= '%s/%s.json' %(settings.USER_TEMP['path'], args[1]['username'])

self.request.send('200'.encode('utf-8'))

client_ack= int(self.request.recv(1024).decode())if client_ack == 200:for key inuser_db:

user_db_value=user_db[key]

res=os.listdir(user_db_value)

res=json.dumps(res)

self.request.send(res.encode('utf-8'))else:

self.request.send('203'.encode('utf-8'))

@check_authdef cd(self,*args):

cmd_dic=args[0]

user_dic= args[1]

user_db=self.user_home(user_dic)

user_auth= '%s/%s.json' %(settings.DATABASE_USER['path'], args[1]['username'])

user_temp= '%s/%s.json' %(settings.USER_TEMP['path'], args[1]['username'])

self.request.send('200'.encode('utf-8'))

client_ack= int(self.request.recv(1024).decode())if client_ack == 200:

with open(user_auth,'r') as f:

user=json.load(f)

user_home= user['Base_Home']if user_db == user_home: #当前目录为用户家目录时操作

for key inuser_db:

user_db_key=key

user_db_value=user_db[key]if cmd_dic['dict'] == '.' or cmd_dic['dict'] == '..': #当前目录为家目录,.和..均返回家目录

self.request.send(user_db_key.encode('utf-8'))else:

r_match= '^' + '(' + user_db_key + ')' + '{1}' + '(.*)' #定义re,内容必须包括家目录字段,不然认为目录不存在

user_dict = re.search(r_match , cmd_dic['dict'])if user_dict.groups() is not None:#当条件匹配时通过拼接方式更新服务器真正目录

user_db_value_new = user_db_value + user_dict.groups()[1] # if os.path.isdir(user_db_value_new):#判断该目录是否存在

os.chdir(user_db_value_new) #如果存在,更改目录

self.request.send(cmd_dic['dict'].encode('utf-8'))del user_db[user_db_key] #删除用户目录对应的字典

user_db[cmd_dic['dict']] = user_db_value_new #重新填写更新内容

with open(user_temp, 'w') as f:

json.dump(user_db,f)#将用户操作字典写入文件

else:

self.request.send('dictory is not exist!'.encode('utf-8'))else:'''当前目录不是用户家目录时操作'''

for key inuser_db:

user_db_key=key

user_db_value=user_db[key]

with open(user_auth,'r') as f:

user=json.load(f)

user_home= user['Base_Home']for key inuser_home:

user_home_value=user_home[key]

user_home_key=keyif cmd_dic['dict'] == '.':

self.request.send(user_db_key.encode('utf-8'))elif cmd_dic['dict'] == '..':

user_db_value_new=os.path.split(user_db_value)[0]

user_db_key_new=os.path.split(user_db_key)[0]

r_match= '^' + '(' + user_home_key + ')' + '{1}' + '(.*)' #re对比的源目录为用户家目录

user_dict =re.search(r_match , user_db_value_new)if user_dict.groups() is notNone:

user_db_value_new= user_home_value + user_dict.groups()[1]ifos.path.isdir(user_db_value_new):

os.chdir(user_db_value_new)

self.request.send(user_db_key_new.encode('utf-8'))deluser_db[user_db_key]

user_db[user_db_key_new]=user_db_value_new

with open(user_temp,'w') as f:

json.dump(user_db,f)else:

self.request.send('已超过根目录'.encode('utf-8'))else:

r_match= '^' + '(' + user_home_key + ')' + '{1}' + '(.*)'user_dict= re.search(r_match , cmd_dic['dict'])if user_dict.groups() is notNone:

user_db_value_new= user_home_value + user_dict.groups()[1]ifos.path.isdir(user_db_value_new):

os.chdir(user_db_value_new)

self.request.send(cmd_dic['dict'].encode('utf-8'))deluser_db[user_db_key]

user_db[cmd_dic['dict']] =user_db_value_new

with open(user_temp,'w') as f:

json.dump(user_db,f)else:

self.request.send('dictory is not exist!'.encode('utf-8'))

mainpy

{"name": "chenronghua", "allow_update": true, "password": "12345", "authenticated": 1, "Base_Home": {"/home/chenronghua": "/Users/Gavin/PycharmProjects/python/day8/Ftp/Ftp-server/home/chenronghua"}, "free_disk": 9267344720}

XXX.json

三、单独标注：

1、类中装饰器

def check_auth(func): #装饰器,用于验证用户是否登录,

def deco(self,*args):

user_auth= '%s/%s.json' %(settings.DATABASE_USER['path'], args[1]['username'])

with open(user_auth,'r') as f:

user_dic=json.load(f)if user_dic['authenticated'] == 1:#在装饰器中,如果用户已登录会在登录函数auth中将authenticated置为1

func(self,*args)else:print('not authed')

self.request.send('202'.encode('utf-8'))return deco

View Code

2、进度条

recv_size =0

with open(File_Name,'wb') as f:while recv_size 1024:

size= 1024

else:

size= filesize -recv_size

data=self.client.recv(size)

f.write(data)

f.flush()

recv_size+=len(data)

pri_size= int(recv_size/filesize*100)

pri_fun= pri_size +1str= '>'*int(pri_size/2) + ' '*int((100-pri_fun)/2)

sys.stdout.write('\r' + str + '%s%%' %pri_size)

sys.stdout.flush()

View Code