import pendulum
from airflow.models import DagRun


__all__ = ["get_timestamp", "get_run_key"]


def _get_run_datetime(dag_run: DagRun) -> pendulum.DateTime:
    return pendulum.instance(dag_run.logical_date).astimezone(tz=pendulum.UTC).replace(tzinfo=None)


def get_timestamp(dag_run: DagRun) -> str:
    return str(int(_get_run_datetime(dag_run).timestamp()))


def get_run_key(dag_run: DagRun) -> str:
    return f"{dag_run.run_type}__{get_timestamp(dag_run)}"