notes: auto add user when saving new note

1 files changed, 5 insertions, 0 deletions

diff --git a/app/notes/forms.py b/app/notes/forms.py
index 5cde108..4342b89 100644
--- a/app/notes/forms.py
+++ b/app/notes/forms.py
@@ -7,9 +7,14 @@ class NoteCreateForm(ModelForm):
 
     def __init__(self,*args,**kwargs):
         user = kwargs.pop('user')
+        self.user = user
         super(NoteCreateForm,self).__init__(*args,**kwargs)
         self.fields['post'].queryset = Post.objects.filter(user=user).order_by("title")
 
+    def save(self, commit=True):
+        self.instance.user = self.user
+        return super().save(commit=commit)
+
     class Meta:
         model = Note 
         fields = ['title', 'url', 'body_markdown', 'post']