add workplace option

work/views.py

@@ -1,12 +1,8 @@
 from django.http import HttpResponseRedirect
 from django.shortcuts import render
-from work.forms import HoursInputForm
+from work.forms import *
 from django.contrib.auth.decorators import login_required
 
-#
-# @login_required
-# def index(request):
-#     return render(request,'registration/login.html')
 
 @login_required
 def get_basic_workdata(request):
@@ -14,10 +10,25 @@ def get_basic_workdata(request):
     if request.method == 'POST':
         form = HoursInputForm(request.POST)
         if form.is_valid():
-            form.save()
+            fs = form.save(commit=False)
+            fs.user = request.user
+            fs.save()
             return HttpResponseRedirect('/')
     else:
         form = HoursInputForm()
 
     return render(request, 'inputhours.html', {'form': form, 'title': title})
 
+
+@login_required
+def addworkplace_view(request):
+    title = "Prace"
+    if request.method == 'POST':
+        form = WorkplaceInputForm(request.POST)
+        if form.is_valid():
+            form.save()
+            return HttpResponseRedirect('/')
+    else:
+        form = WorkplaceInputForm()
+
+    return render(request, 'addworkplace.html', {'form': form, 'title': title})