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


class FreeHotlookBots(EnforcementAsset):
    def __init__(self):
        super().__init__(
            name=JobName.FREE_HOTLOOK_BOTS,
            description="Apply gen restrictions to free hotlook bots",
            entity_type=EntityType.USER,
            capability=Capability.GENERATE,
            status=Status.REVOKED,
            reason="Free hotlook bot",
            schedule="15 * * * *",
            query_file="queries/free_hotlook_bots.sql",
            max_entities_per_run=100,  # dial this down for now
        )


# Create the asset
asset = FreeHotlookBots()