Source code for pollination_apps.cli.context

import os
from pathlib import Path
from typing import Optional

import click
import pollination_sdk as sdk
from pydantic import BaseModel, Field

from ..client import APIClient

DEFAULT_CONFIG_DIR = Path.home() / '.pollination'
DEFAULT_CONFIG_PATH = DEFAULT_CONFIG_DIR / 'apps.config.json'




class Context(BaseModel):

    api_token: Optional[str] = Field(
        default_factory=lambda: os.getenv('POLLINATION_TOKEN'),
        description='The API token to use to authenticate a user'
    )



    @classmethod
    def from_file(cls):
        os.makedirs(DEFAULT_CONFIG_DIR, exist_ok=True)
        try:
            return cls.model_validate_json(DEFAULT_CONFIG_PATH.read_text())
        except:
            ctx = cls()
            ctx.save()
            return ctx




    def save(self):
        DEFAULT_CONFIG_PATH.write_text(self.model_dump_json())


    @property
    def client(self) -> APIClient:
        return APIClient(api_token=self.api_token)



    def try_login(self) -> sdk.UserPrivate:
        try:
            return self.client.get_account()
        except:
            raise click.ClickException(
                f'Not authenticated: {self.client.config.host}'
            )