Module pachyderm_sdk.constants

from os import environ
from pathlib import Path

from . import __version__

DEFAULT_HOST = "localhost"
DEFAULT_PORT = 30650

AUTH_TOKEN_ENV = "PACH_PYTHON_AUTH_TOKEN"
OIDC_TOKEN_ENV = "PACH_PYTHON_OIDC_TOKEN"
CDR_CACHE_LOCATION_ENV = "PACH_CDR_CACHE_LOCATION"
ENTERPRISE_CODE_ENV = "PACH_PYTHON_ENTERPRISE_CODE"
PACH_CONFIG_ENV = "PACH_CONFIG"
PACHD_SERVICE_HOST_ENV = "PACHD_PEER_SERVICE_HOST"
PACHD_SERVICE_PORT_ENV = "PACHD_PEER_SERVICE_PORT"
WORKER_PORT_ENV = "PPS_WORKER_GRPC_PORT"

# `~` used below to make docs look better.
CONFIG_PATH_LOCAL = Path("~").joinpath(".pachyderm", "config.json")
CONFIG_PATH_SPOUT = Path("/").joinpath("pachctl", "config.json")
DEFAULT_CONFIG = environ.get(PACH_CONFIG_ENV, CONFIG_PATH_LOCAL)

# `~` used below to make docs look better.
DEFAULT_CDR_CACHE_LOCATION = Path("~").joinpath(".pachyderm", "cdr_cache")
CDR_CACHE_LOCATION = environ.get(CDR_CACHE_LOCATION_ENV, DEFAULT_CDR_CACHE_LOCATION)

DOTENV_PATH_WORKER = Path("/pfs/.env")

MAX_RECEIVE_MESSAGE_SIZE = 20 * 1024**2  # 20MB
PRIMARY_USER_AGENT = "pachyderm-sdk"
SECONDARY_USER_AGENT = f"v{__version__}" if __version__ else "unspecified"
GRPC_CHANNEL_OPTIONS = [
    ("grpc.max_receive_message_length", MAX_RECEIVE_MESSAGE_SIZE),
    ("grpc.primary_user_agent", PRIMARY_USER_AGENT),
    ("grpc.secondary_user_agent", SECONDARY_USER_AGENT),
]