from django import forms
from .models import Session, Purchase, Game


class SessionForm(forms.ModelForm):
    class Meta:
        model = Session
        fields = [
            "purchase",
            "timestamp_start",
            "timestamp_end",
            "duration_manual",
            "note",
        ]


class PurchaseForm(forms.ModelForm):
    class Meta:
        model = Purchase
        fields = ["game", "platform", "date_purchased", "date_refunded"]


class GameForm(forms.ModelForm):
    class Meta:
        model = Game
        fields = ["name", "wikidata"]