forms.py

from django import forms
from django.contrib.auth.models import User
from django.db.models import Q

from matches.models import Match, GameChoice, PlatformChoice, MapChoice, MapPoolChoice, SportChoice
from news.models import Post
from pages.models import StaticInfo, SocialInfo, Partner, FrontPageSlide
from profiles.models import UserProfile
from singletournaments.models import SingleEliminationTournament, SingleTournamentRuleset, SingleTournamentRound
from store.models import Product
from support.models import TicketComment, Ticket, TicketCategory, QuestionAnswerCategory, QuestionAnswer
from teams.models import Team, TeamInvite
from leagues.models import League, LeagueDivision, LeagueSettings, LeagueTeam


class StaticInfoForm(forms.ModelForm):
    class Meta:
        model = StaticInfo
        fields = '__all__'
        # fields = ('about_us', 'terms', 'stream', 'slide1link', 'slide1_img',
        #          'slide2link', 'slide2_img', 'slide3link', 'slide3_img', 'welcomeln1',
        #          'welcomeln2')

    # def __init__(self, request, *args, **kwargs):
    #    if request.tenant == 'binge':
    #        newfields = ('bingetop1', 'bingetop2', 'bingetop3', 'bingetop1link', 'bingetop2link',
    #                     'bingetop3link', 'bingeslide1big', 'bingeslide2big', 'bingeslide3big',
    #                    'bingeslide1small', 'bingeslide2small', 'bingeslide3small', 'bingeslide1link',
    #                     'bingeslide1link', 'bingeslide2link', 'bingeslide3link')
    #    super().__init__(request, *args, **kwargs)
    #    self.fields.update(newfields)


class SocialInfoForm(forms.ModelForm):
    twitchchannel = forms.URLField(required=False)
    youtubechannel = forms.URLField(required=False)
    twitterprofile = forms.URLField(required=False)
    facebookpage = forms.URLField(required=False)
    instagrampage = forms.URLField(required=False)

    class Meta:
        model = SocialInfo
        fields = '__all__'


class PartnerForm(forms.ModelForm):
    class Meta:
        model = Partner
        fields = '__all__'


class SingleRulesetCreateForm(forms.ModelForm):
    class Meta:
        model = SingleTournamentRuleset
        fields = ('name', 'text')


class TicketSearchForm(forms.Form):
    showClosed = forms.BooleanField(required=False, label='Show closed')
    searchQuery = forms.CharField(required=False, label='Search')


class TicketCategoryCreateForm(forms.ModelForm):
    class Meta:
        model = TicketCategory
        fields = ('name', 'priority',)


class TicketCommentCreateForm(forms.ModelForm):
    class Meta:
        model = TicketComment
        fields = ('comment',)


class CreateTournamentForm(forms.ModelForm):
    class Meta:
        model = SingleEliminationTournament
        fields = '__all__'
        widgets = {
            'open_register': forms.DateTimeInput(
                attrs={'class': 'form-control datetimepicker-input', 'id': 'datetimepicker1',
                       'data-toggle': 'datetimepicker', 'data-target': '#datetimepicker1'}),
            'close_register': forms.DateTimeInput(
                attrs={'class': 'form-control datetimepicker-input', 'id': 'datetimepicker2',
                       'data-toggle': 'datetimepicker', 'data-target': '#datetimepicker2'}),
            'start': forms.DateTimeInput(attrs={'class': 'form-control datetimepicker-input', 'id': 'datetimepicker3',
                                                'data-toggle': 'datetimepicker', 'data-target': '#datetimepicker3'})
        }

    def __init__(self, *args, **kwargs):
        super(CreateTournamentForm, self).__init__(*args, **kwargs)
        self.fields['twitch'].required = False


class EditTournamentForm(forms.ModelForm):
    class Meta:
        model = SingleEliminationTournament
        fields = '__all__'
        widgets = {
            'open_register': forms.DateTimeInput(
                attrs={'class': 'form-control datetimepicker-input', 'id': 'datetimepicker1',
                       'data-toggle': 'datetimepicker', 'data-target': '#datetimepicker1'}),
            'close_register': forms.DateTimeInput(
                attrs={'class': 'form-control datetimepicker-input', 'id': 'datetimepicker2',
                       'data-toggle': 'datetimepicker', 'data-target': '#datetimepicker2'}),
            'start': forms.DateTimeInput(attrs={'class': 'form-control datetimepicker-input', 'id': 'datetimepicker3',
                                                'data-toggle': 'datetimepicker', 'data-target': '#datetimepicker3'})
        }

    def __init__(self, *args, **kwargs):
        super(EditTournamentForm, self).__init__(*args, **kwargs)
        self.fields['twitch'].required = False


class AddTournamentTeamForm(forms.ModelForm):
    teams = forms.IntegerField()

    class Meta:
        model = SingleEliminationTournament
        fields = ()


class DeclareMatchWinnerForm(forms.ModelForm):
    winner = forms.ModelChoiceField(queryset=None)

    class Meta:
        model = Match
        # fields = ('winner',)
        fields = ()

    def __init__(self, request, pk, *args, **kwargs):
        match = Match.objects.filter(id=pk)
        team1 = Team.objects.filter(id__in=match.values_list('hometeam', flat=True))
        team2 = Team.objects.filter(id__in=match.values_list('awayteam', flat=True))
        super().__init__(*args, **kwargs)
        self.fields['winner'].queryset = team1 | team2


class DeclareMatchWinnerPost(forms.ModelForm):
    class Meta:
        model = Match
        fields = ('winner', 'completed')


class ArticleCreateForm(forms.ModelForm):
    class Meta:
        model = Post
        fields = ('title', 'image', 'slug', 'body', 'publish', 'status')
        widgets = {
            'publish': forms.DateTimeInput(
                attrs={'class': 'form-control datetimepicker-input', 'id': 'datetimepicker1',
                       'data-toggle': 'datetimepicker', 'data-target': '#datetimepicker1'}),

        }


class TicketStatusChangeForm(forms.ModelForm):
    class Meta:
        model = Ticket
        fields = ('status', 'assignee')

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        assignees = User.objects.filter(Q(user__user_type='admin') | Q(user__user_type='superadmin'))
        self.fields['assignee'].queryset = assignees


class DeclareTournamentWinnerForm(forms.ModelForm):
    class Meta:
        model = SingleEliminationTournament
        fields = ('winner', 'second', 'third')


class CreateNewsPostForm(forms.ModelForm):
    class Meta:
        model = Post
        fields = '__all__'


class EditNewsPostForm(forms.ModelForm):
    class Meta:
        model = Post
        fields = '__all__'
        widgets = {
            'publish': forms.DateTimeInput(
                attrs={'class': 'form-control datetimepicker-input', 'id': 'datetimepicker1',
                       'data-toggle': 'datetimepicker', 'data-target': '#datetimepicker1'}),
        }


class RemovePlayerForm(forms.Form):
    remove = forms.ModelChoiceField(queryset=None)

    def __init__(self, request, pk, *args, **kwargs):
        team = Team.objects.get(id=pk)
        players = team.players.all() | team.captain.all()
        super().__init__(*args, **kwargs)
        self.fields['remove'].queryset = players


class RemovePlayerFormPost(forms.Form):
    remove = forms.ModelChoiceField(queryset=None)


class ModifyUserForm(forms.ModelForm):
    class Meta:
        model = UserProfile
        fields = (
        'num_bronze', 'num_silver', 'num_gold', 'current_earning', 'xp', 'credits', 'user_type', 'alternate_name',
        'steamid64', 'discord')


class CreateProductForm(forms.ModelForm):
    class Meta:
        model = Product
        fields = ('amount', 'item_name', 'name', 'active')


class EditProductForm(forms.ModelForm):
    class Meta:
        model = Product
        fields = ('amount', 'item_name', 'name', 'active')


class DeleteProductForm(forms.Form):
    price = forms.CharField(max_length=50)
    name = forms.CharField(max_length=50)

    class Meta:
        fields = ('price', 'name')


class EditMatchForm(forms.ModelForm):
    class Meta:
        model = Match
        fields = ('info', 'disable_userreport', 'bestof', 'server', 'datetime')
        widgets = {
            'datetime': forms.DateTimeInput(
                attrs={'class': 'form-control datetimepicker-input', 'id': 'datetimepicker1',
                       'data-toggle': 'datetimepicker', 'data-target': '#datetimepicker1'})}


class GameChoiceForm(forms.ModelForm):
    class Meta:
        model = GameChoice
        fields = '__all__'


class SportChoiceForm(forms.ModelForm):
    class Meta:
        model = SportChoice
        fields = '__all__'


class PlatformChoiceForm(forms.ModelForm):
    class Meta:
        model = PlatformChoice
        fields = '__all__'


class EditRoundInfoForm(forms.ModelForm):
    class Meta:
        model = SingleTournamentRound
        fields = ('info',)


class MapChoiceForm(forms.ModelForm):
    class Meta:
        model = MapChoice
        fields = '__all__'


class MapPoolChoiceForm(forms.ModelForm):
    class Meta:
        model = MapPoolChoice
        fields = '__all__'

    def __init__(self, *args, **kwargs):
        super(MapPoolChoiceForm, self).__init__(*args, **kwargs)
        self.fields['maps'].widget = forms.widgets.CheckboxSelectMultiple()
        if hasattr(self.instance, 'game'):
            self.fields['maps'].queryset = MapChoice.objects.filter(game_id=self.instance.game)


class CreateTeamForm(forms.ModelForm):
    class Meta:
        model = Team
        fields = ('name', 'founder', 'country', 'players')


class CreateQuestionAnswerCategory(forms.ModelForm):
    class Meta:
        model = QuestionAnswerCategory
        fields = ('name',)


class CreateQuestionAnswer(forms.ModelForm):
    class Meta:
        model = QuestionAnswer
        fields = ('question', 'answer', 'category')


class CreateSlide(forms.ModelForm):
    class Meta:
        model = FrontPageSlide
        fields = '__all__'


class CreateLeagueForm(forms.ModelForm):
    class Meta:
        model = League
        fields = '__all__'
        widgets = {
            'open_register': forms.DateTimeInput(
                attrs={'class': 'form-control datetimepicker-input', 'id': 'datetimepicker1',
                       'data-toggle': 'datetimepicker', 'data-target': '#datetimepicker1'}),
            'close_register': forms.DateTimeInput(
                attrs={'class': 'form-control datetimepicker-input', 'id': 'datetimepicker2',
                       'data-toggle': 'datetimepicker', 'data-target': '#datetimepicker2'}),
            'start': forms.DateTimeInput(attrs={'class': 'form-control datetimepicker-input', 'id': 'datetimepicker3',
                                                'data-toggle': 'datetimepicker', 'data-target': '#datetimepicker3'})
        }


class TeamForceAddUser(forms.Form):
    user = forms.CharField(required=True, max_length=50)


class CreateLeagueSettingsForm(forms.ModelForm):
    class Meta:
        model = LeagueSettings
        fields = '__all__'


class EditLeagueSettingsForm(forms.ModelForm):
    class Meta:
        model = LeagueSettings
        fields = '__all__'


class AddLeagueMatchForm(forms.Form):
    awayteam = forms.IntegerField(required=True)
    hometeam = forms.IntegerField(required=True)
    division = forms.IntegerField(required=False)

    class Meta:
        fields = ('hometeam', 'awayteam', 'division')


class DivisionAddTeamForm(forms.Form):
    teamid = forms.IntegerField(required=True)

    class Meta:
        fields = ('teamid')