Source code for tcms.management.forms

# -*- coding: utf-8 -*-

from django import forms

from tcms.management.models import (
    Build,
    Classification,
    Component,
    Priority,
    Product,
    Tag,
    Version,
)




class BuildForm(forms.ModelForm):


    class Meta:
        model = Build
        fields = "__all__"






class ClassificationForm(forms.ModelForm):


    class Meta:
        model = Classification
        fields = "__all__"






class ComponentForm(forms.ModelForm):


    class Meta:
        model = Component
        fields = "__all__"


    def __init__(self, *args, **kwargs):
        request = kwargs.pop("request", None)
        super().__init__(*args, **kwargs)
        if (
            request
            and hasattr(request, "tenant")
            and request.tenant.schema_name != "public"
        ):
            self.fields["initial_owner"].queryset = (
                request.tenant.authorized_users.all()
            )





class PriorityForm(forms.ModelForm):


    class Meta:
        model = Priority
        fields = "__all__"






class ProductForm(forms.ModelForm):


    class Meta:
        model = Product
        fields = "__all__"






class TagForm(forms.ModelForm):


    class Meta:
        model = Tag
        fields = "__all__"






class VersionForm(forms.ModelForm):


    class Meta:
        model = Version
        fields = ["product", "value"]