1 files changed, 44 insertions, 0 deletions

diff --git a/restaurant_orders/settings/views.py b/restaurant_orders/settings/views.py
new file mode 100644
index 0000000..0921cd3
--- /dev/null
+++ b/restaurant_orders/settings/views.py
@@ -0,0 +1,44 @@
+from django.urls import reverse
+from django.shortcuts import render, redirect, Http404
+from django.views.generic.edit import UpdateView, CreateView, View
+from django.contrib.messages.views import SuccessMessageMixin
+from django.contrib.auth.mixins import LoginRequiredMixin, PermissionRequiredMixin
+
+from core.models import Restaurant
+from settings.forms import RestaurantForm
+
+
+class Home(LoginRequiredMixin, PermissionRequiredMixin, View):
+    permission_required = 'core.change_restaurant'
+
+    def get(self, request):
+        if not self.request.user.has_perm('settings.change_restaurant'):
+            raise Http404
+
+        redirect_url = 'settings:restaurant_settings'
+        restaurants = Restaurant.get_user_restaurants(request.user)
+
+        if len(restaurants) == 1:
+            return redirect(redirect_url, pk=restaurants[0].pk)
+
+        return render(request, template_name='restaurants_choice.html', context={
+            'title': 'Ustawienia',
+            'restaurants': restaurants,
+            'redirect_url': redirect_url
+        })
+
+class RestaurantSettings(LoginRequiredMixin, SuccessMessageMixin, PermissionRequiredMixin, UpdateView):
+    template_name = 'settings/restaurant_settings.html'
+    form_class = RestaurantForm
+    model = Restaurant
+    success_message = 'Zapisano!'
+    permission_required = 'core.change_restaurant'
+
+    def get_queryset(self, *args, **kwargs):
+        return super().get_queryset(*args, **kwargs).filter(
+            pk=self.kwargs['pk'],
+            users=self.request.user.pk
+        )
+
+    def get_success_url(self):
+        return reverse('settings:restaurant_settings', args=(self.kwargs['pk'], ))