python练习-用30行代码开发一个上传、下载文件的接口
分享一段代码,开发了3个接口:
1、上传文件
2、查看所有文件
3、下载文件
使用python开发,需要安装flask模块,使用pip install flask安装即可,代码量特别少,就能实现,加上注释总共35行代码,特别的简单,代码如下:
import flask,os,time from flask import request,send_from_directory,jsonify app = flask.Flask(__name__)#创建一个app,代表这个web服务 @app.route('/get_file',methods=['get']) def get_file(): #下载文件接口 filename = request.values.get('fname',None) #获取需要下载的文件名 if filename:#如果获取到的文件名话 if os.path.isfile(filename):#判断是否是一个文件 #返回要下载的文件 return send_from_directory('.',filename,as_attachment=True) else: return jsonify({"msg":"文件不存在!"}) else: return jsonify({'msg':'文件名不能为空'}) @app.route('/files',methods=['get']) def file_list(): #获取文件列表接口 files = os.listdir('.')#获取当前目录下所有文件 new_files = [f for f in files if os.path.isfile(f)] #三元运算符,把是文件的放到list中 return jsonify({"files":new_files}) @app.route('/upload',methods=['post']) def upload(): #上传文件接口 f = request.files.get('file_name',None) if f: t = time.strftime('%Y%m%d%H%M%S')#获取当前时间 new_file_name = t+f.filename#给文件重命名,防止有重复文件覆盖 f.save(new_file_name)#保存文件 return jsonify({"code":"ok"}) else: return jsonify({"msg":"请上传文件!"}) app.run(debug=True,port=8888)#启动这个web服务