#!/usr/bin/env python

"""Demonstrates access to UMLS via UTS.

"""
__author__ = 'Paul Landes'

from typing import Dict, List
from dataclasses import dataclass, field
from pprint import pprint
from zensols.cli import CliHarness ; CliHarness.add_sys_path('src/python')
from zensols.cli import ProgramNameConfigurator
from zensols.mednlp import UTSClient


# the definition of the application class executed from the CLI glue code
@dataclass
class Application(object):
    """Demonstrates access to UTS.

    """
    # tell the application not mistake the `uts_client` as an option when
    # generating the online help with the -h option
    CLI_META = {'option_excludes': {'uts_client'}}

    uts_client: UTSClient = field()
    """Queries UMLS data."""

    def lookup(self, term: str = 'heart'):
        """Look up a term, then return the entry based on the CUI of the first found
        term.

        :param term: the term to search on

        """
        # terms are returned as a list of pages with dictionaries of data
        pages: List[Dict[str, str]] = self.uts_client.search_term(term)
        # get all term dictionaries from the first page
        terms: Dict[str, str] = pages[0]
        # get the concept unique identifier
        cui: str = terms['ui']

        # print atoms of this concept
        print('atoms:')
        pprint(self.uts_client.get_atoms(cui))

        # print all relationships from the concept to all others in the
        # ontology
        print('a few relations:')
        pprint(self.uts_client.get_relations(cui)[:2])


if (__name__ == '__main__'):
    CliHarness(
        app_config_resource='uts.conf',
        app_config_context=ProgramNameConfigurator(
            None, default='uts').create_section(),
        proto_args='',
    ).run()