Django en passant des paramètres de formulaire personnalisés à ModelFormset
-
06-07-2019 - |
Question
Mon problème est similaire à Django en transmettant des paramètres de formulaire personnalisés à Formset
J'ai ces classes
class Game(models.Model):
home_team = models.ForeignKey(Team, related_name='home_team')
away_team = models.ForeignKey(Team, related_name='away_team')
round = models.ForeignKey(Round)
TEAM_CHOICES = ((1, '1'), (2, 'X'), (3, '2'),)
class Odds(models.Model):
game = models.ForeignKey(Game, unique=False)
team = models.IntegerField(choices = TEAM_CHOICES)
odds = models.FloatField()
class Meta:
verbose_name_plural = "Odds"
unique_together = (
("game", "team"),
)
class Vote(models.Model):
user = models.ForeignKey(User, unique=False)
game = models.ForeignKey(Game)
score = models.ForeignKey(Odds)
class Meta:
unique_together = (
("game", "user"),)
Et j'ai défini mon propre modelformset_factory:
def mymodelformset_factory(ins):
class VoteForm(forms.ModelForm):
score = forms.ModelChoiceField(queryset=Odds.objects.filter(game=ins), widget=forms.RadioSelect(), empty_label=None)
def __init__(self, *args, **kwargs):
super(VoteForm, self).__init__(*args, **kwargs)
class Meta:
model = Vote
exclude = ['user']
return VoteForm
Et je l'utilise comme ça:
VoteFormSet = modelformset_factory(Vote, form=mymodelformset_factory(v), extra=0)
formset = VoteFormSet(request.POST, queryset=Vote.objects.filter(game__round=round, user=user))
Ceci affiche le formulaire:
La liste déroulante du ou des jeux de la série spécifiée est supposée afficher 3 boutons radio pour les cotes, mais je ne sais pas quoi transmettre en tant que paramètre à mymodelformset_factory. Si v = jeu.objets. get (pk = 1) évidemment, il affiche uniquement Game avec pk = 1 pour TOUS les jeux. Ce dont j'ai besoin est v = Game.objects.get (pk = "jeu qui est connecté à la cote concernant") si vous voyez ce que je veux dire. ..
La solution
Je pense que vous souhaitez apporter des modifications à votre fonction d'usine personnalisée. Il devrait renvoyer la classe formset, pas la forme. Que diriez-vous de ceci:
def make_vote_formset(game_obj, extra=0):
class _VoteForm(forms.ModelForm):
score = forms.ModelChoiceField(
queryset=Odds.objects.filter(game=game_obj),
widget=forms.RadioSelect(),
empty_label=None)
class Meta:
model = Vote
exclude = ['user',]
return modelformset_factory(Vote, form=_VoteForm, extra=extra)
Ensuite, dans votre code d'affichage:
current_game = Game.objects.filter(id=current_game_id)
VoteFormSet = make_vote_formset(current_game)
formset = VoteFormSet(
request.POST,
queryset=Vote.objects.filter(game__round=round, user=user))
Autres conseils
Une autre solution consiste à sous-classer BaseModelFormSet et à remplacer la méthode _construct_forms . Par défaut, la méthode _construct_form BaseFormSet appelle dans _construct_forms avec un seul argument, i:
# django/forms/formsets.py
def _construct_forms(self):
# instantiate all the forms and put them in self.forms
self.forms = []
for i in xrange(self.total_form_count()):
self.forms.append(self._construct_form(i))
mais il peut y avoir un nombre quelconque d'arguments de mots clés:
# django/forms/formsets.py
def _construct_form(self, i, **kwargs):
Voici donc ma méthode d'affichage et mon formulaire qui reçoit un paramètre supplémentaire dans init à partir de _construct_form:
# view
def edit(request, id):
class ActionsFormSet(BaseModelFormSet):
department = request.user.userdata.department.pk
def _construct_forms(self):
self.forms = []
for i in range(self.total_form_count()):
self.forms.append(self._construct_form(i, dep=self.department))
actions_formset = modelformset_factory(Action, form=ActionForm, extra=0, can_delete=True, formset=ActionsFormSet)
...
# form
class ActionForm(forms.ModelForm):
def __init__(self, dep=0, *args, **kwargs):
super(ActionForm, self).__init__(*args, **kwargs)
self.fields['user'].choices = [(u.pk, u.first_name) for u in User.objects.filter(userdata__department=dep)]
class Meta:
model = Action
exclude = ('req',)