pep8 cleanup

This commit is contained in:
2014-10-17 09:33:16 +02:00
parent 3ac7b75f52
commit d95a9f1d70
2 changed files with 15 additions and 11 deletions

View File

@@ -25,6 +25,7 @@ app = Flask(__name__)
api = Api(app) api = Api(app)
auth = HTTPBasicAuth() auth = HTTPBasicAuth()
@api.representation('application/octet-stream') @api.representation('application/octet-stream')
def output_file(data, code, headers=None): def output_file(data, code, headers=None):
"""Makes a Flask response to return a file.""" """Makes a Flask response to return a file."""
@@ -32,16 +33,18 @@ def output_file(data, code, headers=None):
resp.headers.extend(headers or {}) resp.headers.extend(headers or {})
return resp return resp
@auth.get_password @auth.get_password
def get_password(username): def get_password(username):
if username == 'bam': if username == 'bam':
return 'bam' return 'bam'
return None return None
@auth.error_handler @auth.error_handler
def unauthorized(): def unauthorized():
return make_response(jsonify( { 'message': 'Unauthorized access' } ), 403) return make_response(jsonify({'message': 'Unauthorized access'}), 403)
# return 403 instead of 401 to prevent browsers from displaying # return 403 instead of 401 to prevent browsers from displaying
# the default auth dialog # the default auth dialog
@@ -51,19 +54,20 @@ class FilesListAPI(Resource):
def __init__(self): def __init__(self):
super(FilesListAPI, self).__init__() super(FilesListAPI, self).__init__()
def get(self): def get(self):
return { 'message': 'Display files list' } return {'message': 'Display files list'}
class FileAPI(Resource): class FileAPI(Resource):
"""Downloads a file.""" """Downloads a file."""
decorators = [auth.login_required] decorators = [auth.login_required]
def __init__(self): def __init__(self):
# self.reqparse = reqparse.RequestParser() # self.reqparse = reqparse.RequestParser()
# self.reqparse.add_argument('path', # self.reqparse.add_argument('path',
# type = str, # type = str,
# location = 'json') # location = 'json')
super(FileAPI, self).__init__() super(FileAPI, self).__init__()
@@ -71,8 +75,8 @@ class FileAPI(Resource):
with open(path, 'r') as f: with open(path, 'r') as f:
body = f.read() body = f.read()
return output_file(body, 200) return output_file(body, 200)
#return { 'path': path } # return {'path': path}
api.add_resource(FilesListAPI, '/files', endpoint = 'files') api.add_resource(FilesListAPI, '/files', endpoint='files')
api.add_resource(FileAPI, '/file/<path:path>', endpoint = 'file') api.add_resource(FileAPI, '/file/<path:path>', endpoint='file')

View File

@@ -1,2 +1,2 @@
from application import app from application import app
app.run(debug = True) app.run(debug=True)