__init__.py

# ./swan/__init__.py

# from swan.api.orchestrator import Orchestrator
# from swan.api_client import APIClient
# from swan.contract.swan_contract import SwanContract
from swan.session import Session
from swan.api.orchestrator import Orchestrator

from swan.api.bucket_api import BucketAPI

DEFAULT_SESSION = None

def setup_default_session(api_key=None, network='mainnet', login_url=None, **kwargs):
    """
    Set up a default session, passing through any parameters to the session constructor.
    """
    global DEFAULT_SESSION
    session = Session(api_key=api_key, network=network, login_url=login_url, **kwargs)
    if session.login and session.token == None:
        return 
    DEFAULT_SESSION = session

def _get_default_session(api_key=None, network='mainnet', login_url=None):
    """
    Get the default session, creating one if needed.

    :return: The default session
    """
    global DEFAULT_SESSION
    if DEFAULT_SESSION is None or (api_key is not None and DEFAULT_SESSION.api_key != api_key):
        setup_default_session(api_key=api_key, network=network, login_url=login_url)

    return DEFAULT_SESSION

def resource(api_key=None, login_url=None, service_name=None, *args, **kwargs):
    """
    Create a resource service client by name using the default session for orchestrator, or create an mcs session
    """

    
    # for creating an orchestrator
    if service_name.lower() == 'orchestrator':
        network = kwargs.get('network', 'mainnet')
        session = _get_default_session(api_key, network, login_url)
        if session == None:
            raise ValueError(f"login failed, api key is incorrect")
        return session.resource(service_name='Orchestrator',*args, **kwargs)
    
    # for creating a mcs bucket storage object
    if service_name.lower() == 'storage':
        return BucketAPI(api_key=api_key, *args, **kwargs)
    
    else:
        raise Exception(f"{service_name} is not a valid service")