Source code for runhouse.resources.secrets.provider_secrets.lambda_secret

import copy
import os

from typing import Dict

from runhouse.resources.secrets.provider_secrets.provider_secret import ProviderSecret
from runhouse.resources.secrets.utils import _check_file_for_mismatches
from runhouse.utils import create_local_dir


[docs]class LambdaSecret(ProviderSecret): """ .. note:: To create a LambdaSecret, please use the factory method :func:`provider_secret` with ``provider="lambda"``. """ # values format: {"api_key": api_key} _DEFAULT_CREDENTIALS_PATH = "~/.lambda_cloud/lambda_keys" _PROVIDER = "lambda" @staticmethod def from_config(config: dict, dryrun: bool = False, _resolve_children: bool = True): return LambdaSecret(**config, dryrun=dryrun) def _write_to_file(self, path: str, values: Dict = None, overwrite: bool = False): new_secret = copy.deepcopy(self) if not _check_file_for_mismatches( path, self._from_path(path), values, overwrite ): data = f'api_key = {values["api_key"]}\n' full_path = create_local_dir(path) with open(full_path, "w+") as f: f.write(data) new_secret._add_to_rh_config(path) new_secret._values = None new_secret.path = path return new_secret def _from_path(self, path: str = None): lines = None if path and os.path.exists(os.path.expanduser(path)): with open(os.path.expanduser(path), "r") as f: lines = f.readlines() if lines: for line in lines: split = line.split() if split[0] == "api_key": api_key = split[-1] return {"api_key": api_key} return {}