import datetime
from typing import Optional
import strawberry
from strawberry.relay import GlobalID


@strawberry.input
class RevenueInput:
    account_id: GlobalID
    amount: float
    start_date: datetime.date
    end_date: Optional[datetime.date] = None
    wave_service_id: Optional[str] = None


@strawberry.input
class RevenueUpdateInput:
    id: GlobalID
    account_id: Optional[GlobalID] = None
    amount: Optional[float] = None
    start_date: Optional[datetime.date] = None
    end_date: Optional[datetime.date] = None
    wave_service_id: Optional[str] = None