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


class BotModelUserSync(EnforcementAsset):
    def __init__(self):
        super().__init__(
            name=JobName.BOT_MODEL_USER_SYNC,
            description="Apply degraded gens to bots",
            entity_type=EntityType.USER,
            capability=Capability.GENERATE,
            status=Status.DEGRADED,
            reason=Reason.BOT.value,
            schedule="0,30 * * * *",
            query_file="queries/bot_model_user_sync.sql",
            max_entities_per_run=250000,
        )


# Create the asset
asset = BotModelUserSync()