main.py

import curses

import mongoengine

from model.LoginCredentials import NoSuchUserException, IncorrectPasswordException, login_user, register_user
from model.User import *
from ui import choice_window, deck_edit_screen, study_deck, get_user_input, get_login_credentials, \
    show_message


def show_menu(stdscr, deck_i: int, user: User):
    deck = list(user.decks)[deck_i]
    options = list(
        filter(lambda x: x is not None, ['study', 'edit', 'share' if deck.size > 0 else None, 'delete', 'cancel']))
    choice = options[choice_window(deck.name, options, stdscr)]

    if choice == 'delete':
        user.drop_deck(deck_i)
    elif choice == 'edit':
        deck_edit_screen(stdscr, DeckCreationWizard(user, deck))
        user.save()
    elif choice == 'study':
        study_deck(stdscr, deck, user)
    elif choice == 'cancel':
        pass
    elif choice == 'share':
        description = get_user_input(stdscr, "{} deck description".format(deck.name))
        deck.share(user, description)


def import_deck(stdscr, user):
    search_term = get_user_input(stdscr, 'Search term (ESC to accept)')
    objects = SharedDeck.objects(name__icontains=search_term)
    choice = choice_window('Which one do you choose?',
                           [(d.name, d.description) for d in objects], stdscr)
    if choice >= 0:
        chosen = objects[choice]
        user.import_deck(chosen)


def handle_user(stdscr):
    while True:
        credentials = get_login_credentials(stdscr)
        email = credentials[0]
        password = credentials[1]
        try:
            return login_user(email, password)
        except NoSuchUserException:
            try:
                username = get_user_input(stdscr, 'You are creating a new account. Choose your new username')
                return register_user(username, email, password)
            except mongoengine.errors.ValidationError:
                show_message(stdscr, "{} is not a correct email address".format(email))
                continue
        except IncorrectPasswordException:
            show_message(stdscr, "Incorrect password")
            continue


def main(stdscr):
    stdscr.keypad(True)
    curses.cbreak()
    curses.start_color()
    curses.init_pair(1, curses.COLOR_BLUE, curses.COLOR_BLACK)
    curses.init_pair(2, curses.COLOR_BLUE, curses.COLOR_BLACK)
    curses.init_pair(5, curses.COLOR_WHITE, curses.COLOR_BLACK)
    curses.init_pair(3, curses.COLOR_RED, curses.COLOR_BLACK)
    curses.init_pair(4, curses.COLOR_GREEN, curses.COLOR_BLACK)
    curses.noecho()

    user = handle_user(stdscr)
    print(user.username)

    while True:
        choice = choice_window('Your decks:', user.get_decks_names(), stdscr, {'n': -10, 'i': -11},
                               ['\'n\' to create new deck, \'i\' to search for shared decks', '\'q\' to exit'])
        if choice >= 0:
            show_menu(stdscr, choice, user)
        elif choice == -10:
            name = get_user_input(stdscr, 'Nazwa nowej talii')
            deck_edit_screen(stdscr, user.create_new_deck(name))
            user.save()
        elif choice == -11:
            import_deck(stdscr, user)
        elif choice == -1:
            break
        else:
            break

    user.save()


curses.wrapper(main)