From b97ba8b5c86c9a6ca9c8f4303d0e87c02ae5deec Mon Sep 17 00:00:00 2001
From: Mikko Ahlroth <mikko.ahlroth@gmail.com>
Date: Tue, 25 Jun 2019 12:55:43 +0300
Subject: [PATCH] Go to main view after creating or joining

---
 frontend/src/components/week-budget.ts | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/frontend/src/components/week-budget.ts b/frontend/src/components/week-budget.ts
index 4012d86..ee8c3be 100644
--- a/frontend/src/components/week-budget.ts
+++ b/frontend/src/components/week-budget.ts
@@ -92,6 +92,7 @@ class WeekBudgetComponent extends LitElement {
       budgetObj.makeImmutable()
       this.budgets = [...this.budgets, budgetObj]
       storeToLocal(this.budgets)
+      this.setDisplayMode(DisplayMode.BudgetView)
     }
     catch (err) {
       if (err instanceof CreationError) {
@@ -112,6 +113,7 @@ class WeekBudgetComponent extends LitElement {
       const budget = await this._getBudget(e.detail.uuid)
       this.budgets = [...this.budgets, budget]
       storeToLocal(this.budgets)
+      this.setDisplayMode(DisplayMode.BudgetView)
     }
     catch (err) {
       if (err instanceof GetError) {