la-chariotte/la_chariotte/lieu/forms.py

import datetime

from django import forms
from django.contrib.auth import get_user_model
from django.forms.utils import to_current_timezone
from django.utils import timezone

from la_chariotte.lieu.models import Lieu


class LieuForm(forms.ModelForm):
    class Meta:
        model = Lieu
        fields = [
            "name",
            "description",
            "url",
        ]
        widgets = {
            "name": forms.TextInput(
                attrs={"placeholder": "ex : Centre social Kropotkine"}
            ),
            "description": forms.Textarea(
                attrs={"placeholder": "Plus d'infos sur le lieu ? (facultatif)"}
            ),
            "url": forms.TextInput(attrs={"placeholder": "raccourci"}),
        }

    def __init__(self, *args, **kwargs):
        self.user = kwargs.pop("user")
        super().__init__(*args, **kwargs)

    def save(self, commit=True):
        self.instance.orga = get_user_model().objects.get(id=self.user.pk)
        return super().save(commit=commit)