diff --git a/flangr/circles/views.py b/flangr/circles/views.py index 590f5ed..3a77d30 100644 --- a/flangr/circles/views.py +++ b/flangr/circles/views.py @@ -34,6 +34,7 @@ class CircleCreateView(LoginRequiredMixin, CreateView): circle = form.save(commit=False) circle.owner = self.request.user circle.save() + form.save_m2m() return HttpResponseRedirect(self.get_success_url()) diff --git a/flangr/posts/views.py b/flangr/posts/views.py index f3685e3..ffbd1ba 100644 --- a/flangr/posts/views.py +++ b/flangr/posts/views.py @@ -4,6 +4,7 @@ from django.http import HttpResponseRedirect from django.urls import reverse from django.views.generic import CreateView from django.views.generic import DetailView +from django.views.generic import ListView from django.views.generic.edit import ModelFormMixin from .models import Comment @@ -48,5 +49,11 @@ class PostCreateView(LoginRequiredMixin, CreateView): self.object = form.save(commit=False) self.object.user = self.request.user self.object.save() + form.save_m2m() return HttpResponseRedirect(self.get_success_url()) + + +class PostListView(LoginRequiredMixin, ListView): + def get_queryset(self): + return Post.objects.filter(circles__members=self.request.user)