mirror of
https://github.com/spiral-project/ihatemoney.git
synced 2025-05-05 12:41:49 +02:00
fix error redirection on import
This commit is contained in:
parent
99a78f86fb
commit
946fa3e8e9
2 changed files with 4 additions and 8 deletions
|
@ -74,13 +74,13 @@ class BaseTestCase(TestCase):
|
|||
follow_redirects=follow_redirects,
|
||||
)
|
||||
|
||||
def import_project(self, id, data, expected_code=200):
|
||||
def import_project(self, id, data, success=True):
|
||||
resp = self.client.post(
|
||||
f"/{id}/import",
|
||||
data=data,
|
||||
follow_redirects=True,
|
||||
# follow_redirects=True,
|
||||
)
|
||||
self.assertEqual(resp.status_code, expected_code)
|
||||
self.assertEqual("/{id}/edit" in str(resp.response), not success)
|
||||
|
||||
def create_project(self, id, default_currency="XXX", name=None, password=None):
|
||||
name = name or str(id)
|
||||
|
|
|
@ -444,7 +444,6 @@ def edit_project():
|
|||
@main.route("/<project_id>/import", methods=["POST"])
|
||||
def import_project():
|
||||
form = ImportProjectForm()
|
||||
result = 200
|
||||
if form.validate():
|
||||
try:
|
||||
data = form.file.data
|
||||
|
@ -497,13 +496,10 @@ def import_project():
|
|||
return redirect(url_for("main.list_bills"))
|
||||
except ValueError as b:
|
||||
flash(b.args[0], category="danger")
|
||||
result = 400
|
||||
else:
|
||||
for component, errors in form.errors.items():
|
||||
flash(_(component + ": ") + ", ".join(errors), category="danger")
|
||||
result = 400
|
||||
|
||||
return redirect(url_for(".edit_project"), code=result)
|
||||
return redirect(request.headers.get("Referer") or url_for(".edit_project"))
|
||||
|
||||
|
||||
@main.route("/<project_id>/delete", methods=["POST"])
|
||||
|
|
Loading…
Reference in a new issue