Files
crawlab/routes/files.py
2019-02-17 11:14:22 +08:00

36 lines
852 B
Python

import os
from flask_restful import reqparse, Resource
from app import api
from utils import jsonify
class FileApi(Resource):
parser = reqparse.RequestParser()
arguments = []
def __init__(self):
super(FileApi).__init__()
self.parser.add_argument('path', type=str)
def get(self):
args = self.parser.parse_args()
path = args.get('path')
folders = []
files = []
for _path in os.listdir(path):
if os.path.isfile(os.path.join(path, _path)):
files.append(_path)
elif os.path.isdir(os.path.join(_path)):
folders.append(_path)
return jsonify({
'status': 'ok',
'files': sorted(files),
'folders': sorted(folders),
})
api.add_resource(FileApi,
'/api/files')