32 lines
816 B
Python
32 lines
816 B
Python
|
import click
|
||
|
|
import os
|
||
|
|
from click_shell import shell
|
||
|
|
from codeforces_scraper import Scraper
|
||
|
|
from termforces.session_manager import load_session
|
||
|
|
|
||
|
|
SEARCH_LOCATIONS = ['.', '..', os.path.join(os.path.expanduser('~'), '.config', 'termforces')]
|
||
|
|
FILE_NAME = 'termforces_cookies.json'
|
||
|
|
|
||
|
|
|
||
|
|
class State:
|
||
|
|
def __init__(self):
|
||
|
|
self.scraper = Scraper()
|
||
|
|
|
||
|
|
|
||
|
|
def preload_session(state):
|
||
|
|
for loc in SEARCH_LOCATIONS:
|
||
|
|
file_path = os.path.join(loc, FILE_NAME)
|
||
|
|
if os.path.exists(file_path):
|
||
|
|
print(f'Loading session from {file_path}')
|
||
|
|
load_session(state.scraper, file_path)
|
||
|
|
return
|
||
|
|
|
||
|
|
|
||
|
|
@shell(prompt='termforces > ', intro='Entering termforces shell')
|
||
|
|
@click.pass_context
|
||
|
|
def termforces_shell(ctx):
|
||
|
|
state = State()
|
||
|
|
preload_session(state)
|
||
|
|
ctx.obj = state
|
||
|
|
pass
|