How to use _load_from_env method in lisa

Best Python code snippet using lisa_python

loaders.py

Source:loaders.py

...12    is returned as a second return value, it can be used to13    provide help to the caller.14    """15    missing = []16    uri = _load_from_env(17        "SPYGLASS_KAFKA_URI",18        "URI used to connect on kafka",19        missing,20    )21    cafile = _load_from_env(22        "SPYGLASS_KAFKA_SSL_CA",23        "Path to CA file used to sign certificate",24        missing,25    )26    cert = _load_from_env(27        "SPYGLASS_KAFKA_SSL_CERT",28        "Path to signed certificate",29        missing,30    )31    privkey = _load_from_env(32        "SPYGLASS_KAFKA_SSL_KEY",33        "Path to private key file",34        missing,35    )36    if missing != []:37        errmsg = "\nMissing environment variables for Kafka config:"38        return None, errmsg + "\n\n" + "\n".join(missing)39    return KafkaConfig(40        uri=uri, ssl_cafile=cafile, ssl_cert=cert, ssl_keyfile=privkey), None41def load_health_check_config():42    missing = []43    cfgpath = _load_from_env(44        "SPYGLASS_HEALTH_CHECKS_CONFIG",45        "Path to health checks config file (JSON)",46        missing,47    )48    if missing != []:49        errmsg = "\nMissing environment variables for health checks config:"50        return None, errmsg + "\n\n" + "\n".join(missing)51    try:52        with open(cfgpath, "r") as cfgfile:53            parsed_cfg = json.loads(cfgfile.read())54            checks = []55            for probe in parsed_cfg["probes"]:56                checks.append(HealthCheck(57                    url=probe["url"],58                    period_sec=probe["period_sec"],59                    patterns=probe.get("patterns", []),60                    )61                )62            return checks, None63    except Exception as err:64        m = f"\nError reading/parsing configuration:'{cfgpath}':\n{err}\n"65        return None, m66def load_postgresql_config():67    """68    Loads postgresql related config from the environment.69    If any configuration is missing an informational string70    is returned as a second return value, it can be used to71    provide help to the caller.72    """73    missing = []74    uri = _load_from_env(75        "SPYGLASS_POSTGRESQL_URI",76        "URI used to connect on PostgreSQL",77        missing,78    )79    if missing != []:80        errmsg = "\nMissing environment variables for postgresql config:"81        return None, errmsg + "\n\n" + "\n".join(missing)82    return PostgreSQLConfig(uri=uri), None83def load_log_level():84    val = os.environ.get("SPYGLASS_LOG_LEVEL", "debug")85    return val.upper()86def _load_from_env(envvar, about, missing):87    val = os.environ.get(envvar)88    if val is None:89        missing.append(envvar + " : " + about)...

settings.py

Source:settings.py

...5BOT_DEFAULTS = {'polling_interval': 3, 'router_polling_period': 3}6RESIDENTS = ['json']7ALLOWED = ['users']8def get_router_settings():9    return _load_from_env('router', ROUTER_SETTINGS)10def get_bot_settings():11    return _load_from_env('bot', BOT_SETTINGS, **BOT_DEFAULTS)12def get_residents():13    data = _load_from_env('residents', RESIDENTS)14    return json.loads(data.get('json'))15def get_allowed_users():16    data = _load_from_env('allowed', ALLOWED)17    return json.loads(data.get('users'))18def get_log_format():19    return '%(asctime)s (%(filename)s:%(lineno)d %(threadName)s) %(levelname)s - %(name)s: "%(message)s"'20def _load_from_env(prefix, variable_names, **defaults):21    defaults = {} if defaults is None else defaults22    settings = {v: os.getenv(f'{prefix}_{v}'.upper()) for v in variable_names}23    for field in settings.keys():24        if settings.get(field) is None:25            default_value = defaults.get(field)26            if default_value is not None:27                settings[field] = default_value28            else:29                raise Exception(f'The required variable {field} is not set in environment')...

config.py

Source:config.py

...5        self.server_IP : str = "127.0.0.1"6        self.server_port : int = 67427        self.device_name : Optional[str] = None8        self.color_hue : int = 3509        self._load_from_env()10    def _load_from_env(self):11        # Get IP 12        ip = os.environ.get("IP")13        self.server_IP = self.server_IP if ip is None else ip14        # Get port15        port = os.environ.get("PORT")16        self.server_port = self.server_port if port is None else int(port)17        # Get device name18        device_name = os.environ.get("DEVICE_NAME")19        self.device_name = device_name20        # Get color Hue21        color_hue = os.environ.get("COLOR_HUE")...

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.