automation_actions.py

"""
Actions used for the automation rules.

Each function will receive the following args:

- version: The version object where the action will be applied
- action_arg: An additional argument to apply the action
"""

import structlog

from readthedocs.core.utils import trigger_build
from readthedocs.projects.constants import PRIVATE
from readthedocs.projects.constants import PUBLIC


log = structlog.get_logger(__name__)


def trigger_build_for_version(version, *args, **kwargs):
    """Trigger a build for this version."""
    if version.active:
        trigger_build(project=version.project, version=version, from_webhook=True)


def activate_version(version, *args, **kwargs):
    """
    Sets version as active.

    It triggers a build if the version isn't built.
    """
    version.active = True
    version.save()
    if not version.built:
        trigger_build(project=version.project, version=version)


def set_default_version(version, *args, **kwargs):
    """
    Sets version as the project's default version.

    The version is activated first.
    """
    activate_version(version)
    project = version.project
    project.default_version = version.slug
    project.save()


def hide_version(version, *args, **kwargs):
    """
    Sets version as hidden.

    It also activates the version and triggers a build.
    """
    version.hidden = True
    version.save()

    if not version.active:
        activate_version(version)


def set_public_privacy_level(version, *args, **kwargs):
    """Sets the privacy_level of the version to public."""
    version.privacy_level = PUBLIC
    version.save()


def set_private_privacy_level(version, *args, **kwargs):
    """Sets the privacy_level of the version to private."""
    version.privacy_level = PRIVATE
    version.save()


def delete_version(version, *args, **kwargs):
    """Delete a version if isn't marked as the default version."""
    if version.project.default_version == version.slug:
        log.info(
            "Skipping deleting default version.",
            project_slug=version.project.slug,
            version_slug=version.slug,
        )
        return
    version.delete()