test.py

import sys
import time
import curses

from utils.cli_utils import update_console
from utils.html_parser import get_parsed_html
from utils.constants import TOPICS, TOPIC_CLASS, URL, QUOTE_CLASS, PAGE_LINK_CLASS
from text_acquisition.preloading import update_preload, preload


def get_topics_less_2_pages():
    topics_html = get_parsed_html(TOPICS)
    links = topics_html.find_all('a', {'class': TOPIC_CLASS}, href=True)
    hrefs = [link['href'] for link in links]

    exceptions = []
    for href in hrefs:
        topic_url = URL + href
        quotes_html = get_parsed_html(topic_url)
        divs = quotes_html.find_all('div', {'class': QUOTE_CLASS})
        page_links = quotes_html.find_all('a', {'class': PAGE_LINK_CLASS}, href=True)

        try:
            max_page_number = int(page_links[-2].text)
        except: 
            print(page_links)
            exceptions.append(href)

    for e in exceptions:
        print(e)


def throw_exception():
    console = curses.initscr()
    console.clear()
    console.addstr(0, 0, 'test')
    console.refresh()

    time.sleep(1)

    try:
        x = 1 / 0
    except Exception as e:
        curses.endwin() 
        time.sleep(1)
        print('Ended window')
        time.sleep(1)
        raise e


def catch_keypress():
    console = curses.initscr() 
    update_console(0, 0, "Press any key", console)

    while True:
        key = console.getch()
        update_console(1, 0, str(key), console)
        if key == 3:
            update_console(2, 0, 'KeyboardInterrupt', console)
            sys.exit()


def test_update_cache():
    cache = [['habibi', 'sein sohn']]
    update_preload(cache)


def test_preload():
    cache = preload()
    print(type(cache[0]))
    print(len(cache))
    print(cache)


if __name__ == '__main__':
    #get_topics_less_2_pages()
    #throw_exception()
    catch_keypress()
    #quote_acquisition_time() 
    #test_update_cache()
    #test_preload()