from typing import Optional
from pydantic import BaseModel
from models.ec2 import EC2
from models.loadbalancer import LoadBalancer
from models.vpn import VPN
from models.ebs import EBS
from models.waf import WAF


class Services(BaseModel):
    EC2: Optional[EC2]
    LoadBalancer: Optional[LoadBalancer]
    VPN: Optional[VPN]
    EBS: Optional[EBS]
    WAF: Optional[WAF]