Commit 8d690ae9 authored by Paul Brown's avatar Paul Brown

fix python 3 compatibility issue, trim trailing space

parent 72cce208
...@@ -122,7 +122,7 @@ class FileAdmin(BaseView, ActionsMixin): ...@@ -122,7 +122,7 @@ class FileAdmin(BaseView, ActionsMixin):
""" """
Base form class. Will be used to create the upload, rename, edit, and delete form. Base form class. Will be used to create the upload, rename, edit, and delete form.
Allows enabling CSRF validation and useful if you want to have custom Allows enabling CSRF validation and useful if you want to have custom
contructor or override some fields. contructor or override some fields.
Example:: Example::
...@@ -205,7 +205,7 @@ class FileAdmin(BaseView, ActionsMixin): ...@@ -205,7 +205,7 @@ class FileAdmin(BaseView, ActionsMixin):
directories, etc) directories, etc)
""" """
return self.base_url return self.base_url
def get_upload_form(self): def get_upload_form(self):
""" """
Upload form class for file upload view. Upload form class for file upload view.
...@@ -219,10 +219,10 @@ class FileAdmin(BaseView, ActionsMixin): ...@@ -219,10 +219,10 @@ class FileAdmin(BaseView, ActionsMixin):
""" """
upload = fields.FileField(lazy_gettext('File to upload')) upload = fields.FileField(lazy_gettext('File to upload'))
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
super(UploadForm, self).__init__(*args, **kwargs) super(UploadForm, self).__init__(*args, **kwargs)
self.admin = kwargs['admin'] self.admin = kwargs['admin']
def validate_upload(self, field): def validate_upload(self, field):
if not self.upload.data: if not self.upload.data:
raise validators.ValidationError(gettext('File required.')) raise validators.ValidationError(gettext('File required.'))
...@@ -245,7 +245,7 @@ class FileAdmin(BaseView, ActionsMixin): ...@@ -245,7 +245,7 @@ class FileAdmin(BaseView, ActionsMixin):
(validators.required(),)) (validators.required(),))
return EditForm return EditForm
def get_name_form(self): def get_name_form(self):
""" """
Create form class for renaming and mkdir views. Create form class for renaming and mkdir views.
...@@ -280,7 +280,7 @@ class FileAdmin(BaseView, ActionsMixin): ...@@ -280,7 +280,7 @@ class FileAdmin(BaseView, ActionsMixin):
path = fields.HiddenField(validators=[validators.Required()]) path = fields.HiddenField(validators=[validators.Required()])
return DeleteForm return DeleteForm
def upload_form(self): def upload_form(self):
""" """
Instantiate file upload form and return it. Instantiate file upload form and return it.
...@@ -293,7 +293,7 @@ class FileAdmin(BaseView, ActionsMixin): ...@@ -293,7 +293,7 @@ class FileAdmin(BaseView, ActionsMixin):
# https://bitbucket.org/danjac/flask-wtf/issue/12/fieldlist-filefield-does-not-follow # https://bitbucket.org/danjac/flask-wtf/issue/12/fieldlist-filefield-does-not-follow
formdata = request.form.copy() # as request.form is immutable formdata = request.form.copy() # as request.form is immutable
formdata.update(request.files) formdata.update(request.files)
# admin=self allows the form to use self.is_file_allowed # admin=self allows the form to use self.is_file_allowed
return upload_form_class(formdata, admin=self) return upload_form_class(formdata, admin=self)
elif request.files: elif request.files:
...@@ -314,7 +314,7 @@ class FileAdmin(BaseView, ActionsMixin): ...@@ -314,7 +314,7 @@ class FileAdmin(BaseView, ActionsMixin):
return name_form_class(request.args) return name_form_class(request.args)
else: else:
return name_form_class() return name_form_class()
def edit_form(self): def edit_form(self):
""" """
Instantiate file editing form and return it. Instantiate file editing form and return it.
...@@ -699,7 +699,7 @@ class FileAdmin(BaseView, ActionsMixin): ...@@ -699,7 +699,7 @@ class FileAdmin(BaseView, ActionsMixin):
flash(gettext('Failed to create directory: %(error)s', error=ex), 'error') flash(gettext('Failed to create directory: %(error)s', error=ex), 'error')
else: else:
helpers.flash_errors(form, message='Failed to create directory: %(error)s') helpers.flash_errors(form, message='Failed to create directory: %(error)s')
return self.render(self.mkdir_template, return self.render(self.mkdir_template,
form=form, form=form,
dir_url=dir_url) dir_url=dir_url)
...@@ -717,7 +717,7 @@ class FileAdmin(BaseView, ActionsMixin): ...@@ -717,7 +717,7 @@ class FileAdmin(BaseView, ActionsMixin):
else: else:
return_url = self.get_url('.index') return_url = self.get_url('.index')
if self.validate_form(form): if self.validate_form(form):
# Get path and verify if it is valid # Get path and verify if it is valid
base_path, full_path, path = self._normalize_path(path) base_path, full_path, path = self._normalize_path(path)
...@@ -847,7 +847,7 @@ class FileAdmin(BaseView, ActionsMixin): ...@@ -847,7 +847,7 @@ class FileAdmin(BaseView, ActionsMixin):
helpers.flash_errors(form, message='Failed to edit file. %(error)s') helpers.flash_errors(form, message='Failed to edit file. %(error)s')
try: try:
with open(full_path, 'r') as f: with open(full_path, 'rb') as f:
content = f.read() content = f.read()
except IOError: except IOError:
flash(gettext("Error reading %(name)s.", name=path), 'error') flash(gettext("Error reading %(name)s.", name=path), 'error')
......
...@@ -34,7 +34,7 @@ def test_file_admin(): ...@@ -34,7 +34,7 @@ def test_file_admin():
rv = client.get('/admin/myfileadmin/') rv = client.get('/admin/myfileadmin/')
eq_(rv.status_code, 200) eq_(rv.status_code, 200)
ok_('path=dummy.txt' in rv.data.decode('utf-8')) ok_('path=dummy.txt' in rv.data.decode('utf-8'))
# edit # edit
rv = client.get('/admin/myfileadmin/edit/?path=dummy.txt') rv = client.get('/admin/myfileadmin/edit/?path=dummy.txt')
eq_(rv.status_code, 200) eq_(rv.status_code, 200)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment