from .enforcement_asset import EnforcementAsset
from .constants import Capability, EntityType, JobName, Status


class RegionalOutlookBots(EnforcementAsset):
    def __init__(self):
        super().__init__(
            name=JobName.REGIONAL_OUTLOOK_BOTS,
            description="Apply gen restrictions to regional outlook bots",
            entity_type=EntityType.USER,
            capability=Capability.GENERATE,
            status=Status.SILENTLY_REVOKED,
            reason="Bots currently spamming us from regional outlook domains",
            schedule="15 * * * *",
            query_file="queries/regional_outlook_bots.sql",
            max_entities_per_run=1000,
        )


# Create the asset
asset = RegionalOutlookBots()