import argparse
import operations

parser = argparse.ArgumentParser(prog="Półka z książkami", description="Aplikacja do zarządzania zbiorem książek.")
parser.add_argument("--install", action="store_true")
parser.add_argument('--add-author', help="Imię i nazwisko autora rozdzielone spacją.")
parser.add_argument('--list-authors', action="store_true", help="Wypisuje wszystkich autorów.")
parser.add_argument('--add-book', nargs=2, metavar=("TYTUL", "AUTOR"), help="Dodaje książkę i autora")
parser.add_argument('--list-books', action="store_true", help="Wypisuje wszystkie książki.")
arguments = parser.parse_args()

if arguments.install:
    operations.create_tables()

if arguments.add_author:
    operations.add_author(arguments.add_author)

if arguments.list_authors:
    for author in operations.list_authors():
        print(author)

if arguments.add_book:
    title, author = arguments.add_book
    operations.add_book(
        title=title,
        author=author
    )

if arguments.list_books:
    books = {}
    for row in operations.list_books():
        if (row['first_name'], row['last_name']) not in books:
            books[(row['first_name'], row['last_name'])] = []

        books[(row['first_name'], row['last_name'])].append(row['title'])

    for first_name, last_name in books:
        print(first_name, last_name)

        for title in books[(first_name, last_name)]:
            print(f" - {title}")


    #
# Dodaj do aplikacji pobieranie książek wraz z autorami
# Wykorzystaj zapytanie SQL, które znajduje się na chacie.
# Format wyniku powinien wygądać tak:
# Jan Kochowski:
# - Treny
# - Fraszki
# Henryk Sienkiewicz:
# - W pusty...
# - Quo Vadis..