Skip to content

Main

main(command=None) async

Main function to run the Peter Explains CLI.

This function is an async function that runs the main Peter Explains CLI. It uses the asyncio library to run the main Peter Explains CLI.

Source code in src/peter_explains/main.py 
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
async def main(command: str = None):
    """
    Main function to run the Peter Explains CLI.

    This function is an async function that runs the main Peter Explains CLI.
    It uses the asyncio library to run the main Peter Explains CLI.
    """

    try:
        # command = parse_arguments()
        peter_ai = PeterAi()
        cache = PeterCache()

        if command in cache:  # Check in cache
            result = cache.get(command)

        else:
            tasks = [
                asyncio.create_task(show_loading_message()),
                asyncio.create_task(peter_ai.explain_command(command)),
            ]
            done, pending = await asyncio.wait(
                tasks, return_when=asyncio.FIRST_COMPLETED
            )

            pending.pop().cancel()  # Cancel the loading message task
            result = done.pop().result()  # Get the result of the explain_command task

            cache.save(command, result)  # Save in cache

        pretty_print_result(result)

    except Exception as e:
        show_error_message(e)

peter(command, api, delete_api, delete_cache, help)

Function to run the Peter Explains CLI.

This function is the entry point for the Peter Explains CLI. It runs the main function using the asyncio library.

Source code in src/peter_explains/main.py 
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
@click.command()
@click.argument("command", required=False)
@click.option("--api", metavar="<API_KEY>", help="Set the Google AI platform API key")
@click.option(
    "--delete-api", is_flag=True, help="Deletes the Google AI platform API key"
)
@click.option("--delete-cache", is_flag=True, help="Delete the cache")
@click.option("--help", "-h", is_flag=True, help="Display the help message")
@click.version_option(__version__, prog_name="Peter Explains (peter)")
def peter(command, api, delete_api, delete_cache, help):
    """
    Function to run the Peter Explains CLI.

    This function is the entry point for the Peter Explains CLI.
    It runs the main function using the asyncio library.
    """
    # Handle API key operations
    google_api_key = GoogleApiKey()

    if api:
        google_api_key.set(api)
        click.echo("API key set successfully.")
        return

    if delete_api:
        google_api_key.clear()
        click.echo("API key deleted successfully.")
        return

    # Handle cache operations
    cache = PeterCache()

    if delete_cache:
        cache.clear()
        click.echo("Cache cleared successfully.")
        return

    if help:
        show_peter_help()
        return

    # If no command is provided, show help
    if not command:
        show_peter_help()
        return

    # Run the explanation function
    asyncio.run(main(command.strip().lower()))