ob-metaflow-extensions 1.1.151__py2.py3-none-any.whl → 1.6.2__py2.py3-none-any.whl

metaflow_extensions/outerbounds/plugins/apps/core/_vendor/spinner/spinners.py

@@ -0,0 +1,478 @@
+# -*- coding: utf-8 -*-
+"""
+Python wrapper for beautiful terminal spinner library.
+
+Spinners are from:
+* cli-spinners:
+    MIT License
+
+    Copyright (c) Sindre Sorhus <sindresorhus@gmail.com> (sindresorhus.com)
+
+    Permission is hereby granted, free of charge, to any person obtaining a copy
+    of this software and associated documentation files (the "Software"), to deal
+    in the Software without restriction, including without limitation the rights to
+    use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
+    the Software, and to permit persons to whom the Software is furnished to do so,
+    subject to the following conditions:
+
+    The above copyright notice and this permission notice shall be included
+    in all copies or substantial portions of the Software.
+
+    THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED,
+    INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR
+    PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE
+    FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
+    ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
+    IN THE SOFTWARE.
+"""
+from __future__ import unicode_literals
+
+from enum import Enum
+
+Spinners = Enum(
+    "Spinners",
+    {
+        "dots": {
+            "interval": 80,
+            "frames": ["⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"],
+        },
+        "dots2": {"interval": 80, "frames": ["⣾", "⣽", "⣻", "⢿", "⡿", "⣟", "⣯", "⣷"]},
+        "dots3": {
+            "interval": 80,
+            "frames": ["⠋", "⠙", "⠚", "⠞", "⠖", "⠦", "⠴", "⠲", "⠳", "⠓"],
+        },
+        "dots4": {
+            "interval": 80,
+            "frames": [
+                "⠄",
+                "⠆",
+                "⠇",
+                "⠋",
+                "⠙",
+                "⠸",
+                "⠰",
+                "⠠",
+                "⠰",
+                "⠸",
+                "⠙",
+                "⠋",
+                "⠇",
+                "⠆",
+            ],
+        },
+        "dots5": {
+            "interval": 80,
+            "frames": [
+                "⠋",
+                "⠙",
+                "⠚",
+                "⠒",
+                "⠂",
+                "⠂",
+                "⠒",
+                "⠲",
+                "⠴",
+                "⠦",
+                "⠖",
+                "⠒",
+                "⠐",
+                "⠐",
+                "⠒",
+                "⠓",
+                "⠋",
+            ],
+        },
+        "dots6": {
+            "interval": 80,
+            "frames": [
+                "⠁",
+                "⠉",
+                "⠙",
+                "⠚",
+                "⠒",
+                "⠂",
+                "⠂",
+                "⠒",
+                "⠲",
+                "⠴",
+                "⠤",
+                "⠄",
+                "⠄",
+                "⠤",
+                "⠴",
+                "⠲",
+                "⠒",
+                "⠂",
+                "⠂",
+                "⠒",
+                "⠚",
+                "⠙",
+                "⠉",
+                "⠁",
+            ],
+        },
+        "dots7": {
+            "interval": 80,
+            "frames": [
+                "⠈",
+                "⠉",
+                "⠋",
+                "⠓",
+                "⠒",
+                "⠐",
+                "⠐",
+                "⠒",
+                "⠖",
+                "⠦",
+                "⠤",
+                "⠠",
+                "⠠",
+                "⠤",
+                "⠦",
+                "⠖",
+                "⠒",
+                "⠐",
+                "⠐",
+                "⠒",
+                "⠓",
+                "⠋",
+                "⠉",
+                "⠈",
+            ],
+        },
+        "dots8": {
+            "interval": 80,
+            "frames": [
+                "⠁",
+                "⠁",
+                "⠉",
+                "⠙",
+                "⠚",
+                "⠒",
+                "⠂",
+                "⠂",
+                "⠒",
+                "⠲",
+                "⠴",
+                "⠤",
+                "⠄",
+                "⠄",
+                "⠤",
+                "⠠",
+                "⠠",
+                "⠤",
+                "⠦",
+                "⠖",
+                "⠒",
+                "⠐",
+                "⠐",
+                "⠒",
+                "⠓",
+                "⠋",
+                "⠉",
+                "⠈",
+                "⠈",
+            ],
+        },
+        "dots9": {"interval": 80, "frames": ["⢹", "⢺", "⢼", "⣸", "⣇", "⡧", "⡗", "⡏"]},
+        "dots10": {"interval": 80, "frames": ["⢄", "⢂", "⢁", "⡁", "⡈", "⡐", "⡠"]},
+        "dots11": {"interval": 100, "frames": ["⠁", "⠂", "⠄", "⡀", "⢀", "⠠", "⠐", "⠈"]},
+        "dots12": {
+            "interval": 80,
+            "frames": [
+                "⢀⠀",
+                "⡀⠀",
+                "⠄⠀",
+                "⢂⠀",
+                "⡂⠀",
+                "⠅⠀",
+                "⢃⠀",
+                "⡃⠀",
+                "⠍⠀",
+                "⢋⠀",
+                "⡋⠀",
+                "⠍⠁",
+                "⢋⠁",
+                "⡋⠁",
+                "⠍⠉",
+                "⠋⠉",
+                "⠋⠉",
+                "⠉⠙",
+                "⠉⠙",
+                "⠉⠩",
+                "⠈⢙",
+                "⠈⡙",
+                "⢈⠩",
+                "⡀⢙",
+                "⠄⡙",
+                "⢂⠩",
+                "⡂⢘",
+                "⠅⡘",
+                "⢃⠨",
+                "⡃⢐",
+                "⠍⡐",
+                "⢋⠠",
+                "⡋⢀",
+                "⠍⡁",
+                "⢋⠁",
+                "⡋⠁",
+                "⠍⠉",
+                "⠋⠉",
+                "⠋⠉",
+                "⠉⠙",
+                "⠉⠙",
+                "⠉⠩",
+                "⠈⢙",
+                "⠈⡙",
+                "⠈⠩",
+                "⠀⢙",
+                "⠀⡙",
+                "⠀⠩",
+                "⠀⢘",
+                "⠀⡘",
+                "⠀⠨",
+                "⠀⢐",
+                "⠀⡐",
+                "⠀⠠",
+                "⠀⢀",
+                "⠀⡀",
+            ],
+        },
+        "line": {"interval": 130, "frames": ["-", "\\", "|", "/"]},
+        "line2": {"interval": 100, "frames": ["⠂", "-", "–", "—", "–", "-"]},
+        "pipe": {"interval": 100, "frames": ["┤", "┘", "┴", "└", "├", "┌", "┬", "┐"]},
+        "simpleDots": {"interval": 400, "frames": [".  ", ".. ", "...", "   "]},
+        "simpleDotsScrolling": {
+            "interval": 200,
+            "frames": [".  ", ".. ", "...", " ..", "  .", "   "],
+        },
+        "star": {"interval": 70, "frames": ["✶", "✸", "✹", "✺", "✹", "✷"]},
+        "star2": {"interval": 80, "frames": ["+", "x", "*"]},
+        "flip": {
+            "interval": 70,
+            "frames": ["_", "_", "_", "-", "`", "`", "'", "´", "-", "_", "_", "_"],
+        },
+        "hamburger": {"interval": 100, "frames": ["☱", "☲", "☴"]},
+        "growVertical": {
+            "interval": 120,
+            "frames": ["▁", "▃", "▄", "▅", "▆", "▇", "▆", "▅", "▄", "▃"],
+        },
+        "growHorizontal": {
+            "interval": 120,
+            "frames": ["▏", "▎", "▍", "▌", "▋", "▊", "▉", "▊", "▋", "▌", "▍", "▎"],
+        },
+        "balloon": {"interval": 140, "frames": [" ", ".", "o", "O", "@", "*", " "]},
+        "balloon2": {"interval": 120, "frames": [".", "o", "O", "°", "O", "o", "."]},
+        "noise": {"interval": 100, "frames": ["▓", "▒", "░"]},
+        "bounce": {"interval": 120, "frames": ["⠁", "⠂", "⠄", "⠂"]},
+        "boxBounce": {"interval": 120, "frames": ["▖", "▘", "▝", "▗"]},
+        "boxBounce2": {"interval": 100, "frames": ["▌", "▀", "▐", "▄"]},
+        "triangle": {"interval": 50, "frames": ["◢", "◣", "◤", "◥"]},
+        "arc": {"interval": 100, "frames": ["◜", "◠", "◝", "◞", "◡", "◟"]},
+        "circle": {"interval": 120, "frames": ["◡", "⊙", "◠"]},
+        "squareCorners": {"interval": 180, "frames": ["◰", "◳", "◲", "◱"]},
+        "circleQuarters": {"interval": 120, "frames": ["◴", "◷", "◶", "◵"]},
+        "circleHalves": {"interval": 50, "frames": ["◐", "◓", "◑", "◒"]},
+        "squish": {"interval": 100, "frames": ["╫", "╪"]},
+        "toggle": {"interval": 250, "frames": ["⊶", "⊷"]},
+        "toggle2": {"interval": 80, "frames": ["▫", "▪"]},
+        "toggle3": {"interval": 120, "frames": ["□", "■"]},
+        "toggle4": {"interval": 100, "frames": ["■", "□", "▪", "▫"]},
+        "toggle5": {"interval": 100, "frames": ["▮", "▯"]},
+        "toggle6": {"interval": 300, "frames": ["ဝ", "၀"]},
+        "toggle7": {"interval": 80, "frames": ["⦾", "⦿"]},
+        "toggle8": {"interval": 100, "frames": ["◍", "◌"]},
+        "toggle9": {"interval": 100, "frames": ["◉", "◎"]},
+        "toggle10": {"interval": 100, "frames": ["㊂", "㊀", "㊁"]},
+        "toggle11": {"interval": 50, "frames": ["⧇", "⧆"]},
+        "toggle12": {"interval": 120, "frames": ["☗", "☖"]},
+        "toggle13": {"interval": 80, "frames": ["=", "*", "-"]},
+        "arrow": {"interval": 100, "frames": ["←", "↖", "↑", "↗", "→", "↘", "↓", "↙"]},
+        "arrow2": {
+            "interval": 80,
+            "frames": ["⬆️ ", "↗️ ", "➡️ ", "↘️ ", "⬇️ ", "↙️ ", "⬅️ ", "↖️ "],
+        },
+        "arrow3": {
+            "interval": 120,
+            "frames": ["▹▹▹▹▹", "▸▹▹▹▹", "▹▸▹▹▹", "▹▹▸▹▹", "▹▹▹▸▹", "▹▹▹▹▸"],
+        },
+        "bouncingBar": {
+            "interval": 80,
+            "frames": [
+                "[    ]",
+                "[=   ]",
+                "[==  ]",
+                "[=== ]",
+                "[ ===]",
+                "[  ==]",
+                "[   =]",
+                "[    ]",
+                "[   =]",
+                "[  ==]",
+                "[ ===]",
+                "[====]",
+                "[=== ]",
+                "[==  ]",
+                "[=   ]",
+            ],
+        },
+        "bouncingBall": {
+            "interval": 80,
+            "frames": [
+                "( ●    )",
+                "(  ●   )",
+                "(   ●  )",
+                "(    ● )",
+                "(     ●)",
+                "(    ● )",
+                "(   ●  )",
+                "(  ●   )",
+                "( ●    )",
+                "(●     )",
+            ],
+        },
+        "smiley": {"interval": 200, "frames": ["😄 ", "😝 "]},
+        "monkey": {"interval": 300, "frames": ["🙈 ", "🙈 ", "🙉 ", "🙊 "]},
+        "hearts": {"interval": 100, "frames": ["💛 ", "💙 ", "💜 ", "💚 ", "❤️ "]},
+        "clock": {
+            "interval": 100,
+            "frames": [
+                "🕛 ",
+                "🕐 ",
+                "🕑 ",
+                "🕒 ",
+                "🕓 ",
+                "🕔 ",
+                "🕕 ",
+                "🕖 ",
+                "🕗 ",
+                "🕘 ",
+                "🕙 ",
+                "🕚 ",
+            ],
+        },
+        "earth": {"interval": 180, "frames": ["🌍 ", "🌎 ", "🌏 "]},
+        "moon": {
+            "interval": 80,
+            "frames": ["🌑 ", "🌒 ", "🌓 ", "🌔 ", "🌕 ", "🌖 ", "🌗 ", "🌘 "],
+        },
+        "runner": {"interval": 140, "frames": ["🚶 ", "🏃 "]},
+        "pong": {
+            "interval": 80,
+            "frames": [
+                "▐⠂       ▌",
+                "▐⠈       ▌",
+                "▐ ⠂      ▌",
+                "▐ ⠠      ▌",
+                "▐  ⡀     ▌",
+                "▐  ⠠     ▌",
+                "▐   ⠂    ▌",
+                "▐   ⠈    ▌",
+                "▐    ⠂   ▌",
+                "▐    ⠠   ▌",
+                "▐     ⡀  ▌",
+                "▐     ⠠  ▌",
+                "▐      ⠂ ▌",
+                "▐      ⠈ ▌",
+                "▐       ⠂▌",
+                "▐       ⠠▌",
+                "▐       ⡀▌",
+                "▐      ⠠ ▌",
+                "▐      ⠂ ▌",
+                "▐     ⠈  ▌",
+                "▐     ⠂  ▌",
+                "▐    ⠠   ▌",
+                "▐    ⡀   ▌",
+                "▐   ⠠    ▌",
+                "▐   ⠂    ▌",
+                "▐  ⠈     ▌",
+                "▐  ⠂     ▌",
+                "▐ ⠠      ▌",
+                "▐ ⡀      ▌",
+                "▐⠠       ▌",
+            ],
+        },
+        "shark": {
+            "interval": 120,
+            "frames": [
+                "▐|\\____________▌",
+                "▐_|\\___________▌",
+                "▐__|\\__________▌",
+                "▐___|\\_________▌",
+                "▐____|\\________▌",
+                "▐_____|\\_______▌",
+                "▐______|\\______▌",
+                "▐_______|\\_____▌",
+                "▐________|\\____▌",
+                "▐_________|\\___▌",
+                "▐__________|\\__▌",
+                "▐___________|\\_▌",
+                "▐____________|\\▌",
+                "▐____________/|▌",
+                "▐___________/|_▌",
+                "▐__________/|__▌",
+                "▐_________/|___▌",
+                "▐________/|____▌",
+                "▐_______/|_____▌",
+                "▐______/|______▌",
+                "▐_____/|_______▌",
+                "▐____/|________▌",
+                "▐___/|_________▌",
+                "▐__/|__________▌",
+                "▐_/|___________▌",
+                "▐/|____________▌",
+            ],
+        },
+        "dqpb": {"interval": 100, "frames": ["d", "q", "p", "b"]},
+        "weather": {
+            "interval": 100,
+            "frames": [
+                "☀️ ",
+                "☀️ ",
+                "☀️ ",
+                "🌤 ",
+                "⛅️ ",
+                "🌥 ",
+                "☁️ ",
+                "🌧 ",
+                "🌨 ",
+                "🌧 ",
+                "🌨 ",
+                "🌧 ",
+                "🌨 ",
+                "⛈ ",
+                "🌨 ",
+                "🌧 ",
+                "🌨 ",
+                "☁️ ",
+                "🌥 ",
+                "⛅️ ",
+                "🌤 ",
+                "☀️ ",
+                "☀️ ",
+            ],
+        },
+        "christmas": {"interval": 400, "frames": ["🌲", "🎄"]},
+        "grenade": {
+            "interval": 80,
+            "frames": [
+                "،   ",
+                "′   ",
+                " ´ ",
+                " ‾ ",
+                "  ⸌",
+                "  ⸊",
+                "  |",
+                "  ⁎",
+                "  ⁕",
+                " ෴ ",
+                "  ⁓",
+                "   ",
+                "   ",
+                "   ",
+            ],
+        },
+        "point": {"interval": 125, "frames": ["∙∙∙", "●∙∙", "∙●∙", "∙∙●", "∙∙∙"]},
+        "layer": {"interval": 150, "frames": ["-", "=", "≡"]},
+    },
+)

metaflow_extensions/outerbounds/plugins/apps/core/app_config.py

@@ -0,0 +1,128 @@
+import json
+import os
+
+from typing import Dict, Any
+from .config import (
+    CoreConfig,
+    AuthType,
+    MergingNotAllowedFieldsException,
+    ConfigValidationFailedException,
+    RequiredFieldMissingException,
+)
+
+CODE_PACKAGE_PREFIX = "mf.obp-apps"
+
+CAPSULE_DEBUG = os.environ.get("OUTERBOUNDS_CAPSULE_DEBUG", False)
+
+
+class AppConfigError(Exception):
+    """Exception raised when app configuration is invalid."""
+
+    pass
+
+
+def _try_loading_yaml(file):
+    try:
+        import yaml
+
+        return yaml.safe_load(file)
+    except ImportError:
+        pass
+
+    try:
+        from outerbounds._vendor import yaml
+
+        return yaml.safe_load(file)
+    except ImportError:
+        pass
+    return None
+
+
+class AppConfig:
+    """Class representing an Outerbounds App configuration."""
+
+    def __init__(self, core_config: CoreConfig):
+        """Initialize configuration from a dictionary."""
+        self._core_config = core_config
+        self._final_state: Dict[str, Any] = {}
+        self.config = {}
+
+    def set_state(self, key, value):
+        self._final_state[key] = value
+        return self
+
+    def get_state(self, key, default=None):
+        return self._final_state.get(key, self.get(key, default))
+
+    def dump_state(self):
+        x = {k: v for k, v in self.config.items()}
+        for k, v in self._final_state.items():
+            x[k] = v
+        return x
+
+    def commit(self):
+        try:
+            self._core_config.commit()
+            self.config = self._core_config.to_dict()
+        except RequiredFieldMissingException as e:
+            raise AppConfigError(
+                "The configuration is missing the following required fields: %s. \n\tException: %s"
+                % (e.field_name, e.message)
+            )
+        except ConfigValidationFailedException as e:
+            raise AppConfigError(
+                "The configuration is invalid. \n\n\tException: %s" % (e.message)
+            )
+
+    def get(self, key: str, default: Any = None) -> Any:
+        """Get a configuration value by key."""
+        config_value = self.config.get(key, default)
+        if config_value is None:
+            return default
+        return config_value
+
+    def to_json(self):
+        return json.dumps(self.config, indent=2)
+
+    def to_yaml(self):
+        return self.to_json()
+
+    def to_dict(self):
+        return self.config
+
+    @classmethod
+    def from_file(cls, file_path: str) -> "AppConfig":
+        """Create a configuration from a file."""
+        if not os.path.exists(file_path):
+            raise AppConfigError(f"Configuration file '{file_path}' does not exist.")
+
+        with open(file_path, "r") as f:
+            try:
+                config_dict = _try_loading_yaml(f)
+                if config_dict is None:
+                    config_dict = json.load(f)
+            except json.JSONDecodeError as e:
+                raise AppConfigError(
+                    "The PyYAML package is not available as a dependency and JSON parsing of the configuration file also failed %s: \n%s"
+                    % (file_path, str(e))
+                )
+            except Exception as e:
+                raise AppConfigError(f"Failed to parse configuration file: {e}")
+
+        return cls(CoreConfig.from_dict(config_dict))
+
+    @classmethod
+    def from_cli(cls, options: Dict[str, Any]):
+        return cls(CoreConfig.from_cli(options))
+
+    def update_from_cli_options(self, options):
+        cli_options_config = CoreConfig.from_cli(options)
+        try:
+            self._core_config = CoreConfig.merge_configs(
+                self._core_config, cli_options_config
+            )
+        except MergingNotAllowedFieldsException as e:
+            raise AppConfigError(
+                "CLI Overrides are not allowed for the following fields: %s. \n\tException: %s"
+                % (e.field_name, e.message)
+            )