Source

indie-community / gamedb / views.py

Diff from to

gamedb/views.py

 from django.views.generic.detail import DetailView
 from gamedb.models import Game, Writer, Artist, Composer
 from django.views.generic.list import ListView
-from gamedb.forms import FilterForm
+from gamedb.forms import FilterForm, GameForm
+from django.views.generic.edit import CreateView
+from django.shortcuts import redirect
 
 class ViewGame(DetailView):
     model = Game
 
 class ViewComposer(DetailView):
     model = Composer
+
+class AddGame(CreateView):
+    model = Game
+    form_class = GameForm
+    def form_valid(self, form):
+        self.object = form.save(commit=False)
+        self.object.owner = self.request.user
+        self.object.save()
+        return redirect(self.object)