first commit

[osm/LW-UI.git] / sf_t3d / views.py

diff --git a/sf_t3d/views.py b/sf_t3d/views.py
new file mode 100644 (file)
index 0000000..91590f0
--- /dev/null
+++ b/sf_t3d/views.py
@@ -0,0 +1,22 @@
+from django.shortcuts import render
+from django.contrib.auth.decorators import login_required
+
+from projecthandler.models import Project
+from sf_user.models import CustomUser
+
+
+@login_required
+def home(request):
+    user = CustomUser.objects.get(id=request.user.id)
+    projects = Project.objects.filter(owner=user).select_subclasses()
+    result = {
+        'projects': len(projects) if projects else 0,
+    }
+    return render(request, 'home.html', result)
+
+
+def forbidden(request):
+    return render(request, 'forbidden.html')
+
+
+