mag_daskhub

mag_daskhub module

cli(ctx, verbose)

magasin client is the glue between magasin components, it makes easier common tasks

Source code in mag/mag.py

@click.group(cls=ClickAliasedGroup)
@click.option('-v', '--verbose', count=True)
@click.option('--version', is_flag=True, callback=print_version,
              expose_value=False, is_eager=True)
@click.pass_context
def cli(ctx, verbose):
    """magasin client is the glue between magasin components, it makes easier common tasks"""
    ctx.ensure_object(dict)

daskhub(realm)

Daskhub/Jupyterhub commands

Source code in mag_daskhub/daskhub.py

@cli.group('daskhub', cls=ClickAliasedGroup, aliases=['dh'])
@options.realm
def daskhub(realm):
  """Daskhub/Jupyterhub commands"""
  namespace = get_namespace(component_name=COMPONENT, realm=realm)
  click.echo("namespace: " + namespace)

get_namespace(component_name, realm='magasin')

Generate a namespace based on the component name and realm.

Parameters:

Name	Type	Description	Default
component_name	str	The magasin component name (superset, daskhub, drill, ...)	required
realm	str	The realm. Defaults to 'magasin'.	'magasin'

Example

• get_namespace("superset", "magasin") -> "magasin-superset"
• get_namespace("superset", "magasin-postfix") -> "magasin-superset-postfix"

Reference

For more information about magasin realms, please see the magasin realms documentation.

Source code in mag/mag_core/realm.py

def get_namespace(component_name: str, realm='magasin') -> str:
    """
    Generate a namespace based on the component name and realm.


    Args:
        component_name (str): The magasin component name (superset, daskhub, drill, ...)
        realm (str, optional): The realm. Defaults to 'magasin'.

    Example:
        * `get_namespace("superset", "magasin")` -> "magasin-superset"
        * `get_namespace("superset", "magasin-postfix")` -> "magasin-superset-postfix"

    Reference:
        For more information about magasin realms, please see the [magasin realms documentation](https://unicef.github.io/magasin/install/advanced.html#magasin-realms).
    """
    prefix, suffix = split_realm(realm)
    namespace = ""
    if prefix:
        namespace = prefix + '-' + component_name
    if suffix:
        namespace = namespace + "-" + suffix
    return namespace

launch_ui(realm, component, service_name, ports, protocol='http', verbose=False)

Launches the user interface for a given realm, component, and service.

Parameters:

Name	Type	Description	Default
realm	str	The realm of the magasin instance.	required
component	str	The magasin component (f.i superset, daskhub, drill, ...)	required
service_name	str	The name of the kubernetes service to forward.	required
ports	str	The ports to forward, using the format "local_port:remote_port".	required
protocol	str	The protocol to use (default is "http").	'http'
verbose	bool	Whether to display verbose output (default is False).	False

Returns:

Name	Type	Description
None	None	Nothing

Source code in mag/mag_core/launch.py

def launch_ui(realm: str, component: str, service_name: str, ports: str, protocol: str = "http", verbose=False) -> None:
    """
    Launches the user interface for a given realm, component, and service.

    Args:
        realm (str): The realm of the magasin instance.
        component (str): The magasin component (f.i superset, daskhub, drill, ...)
        service_name (str): The name of the kubernetes service to forward.
        ports (str): The ports to forward, using the format "local_port:remote_port".
        protocol (str, optional): The protocol to use (default is "http").
        verbose (bool, optional): Whether to display verbose output (default is False).

    Returns:
        None: Nothing
    """    
    forward_port(realm=realm, component=component,
                 service_name=service_name, ports=ports, verbose=verbose)

    localhost_port, _ = split_ports(ports)
    url = f"{protocol}://localhost:{localhost_port}"
    click.echo(f"Open browser at: {url}")
    click.launch(url)
    click.echo("launch ui")

    try:
        # Wait for user to press Ctrl+C
        signal.pause()
    except KeyboardInterrupt:
        # Handle Ctrl+C: terminate the server and clean up
        process.terminate()
        os.waitpid(process.pid, 0)
        click.echo("\nServer terminated. Exiting.")

ui(realm, ports)

Launch jupyterhub user interface

Source code in mag_daskhub/daskhub.py

@daskhub.command()
@options.realm
@options.ports(default="8001:80")
def ui(realm, ports):
  """Launch jupyterhub user interface"""
  launch_ui(realm=realm,component=COMPONENT, service_name='service/proxy-public', ports=ports)