mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-28 13:01:38 +00:00
109 lines
2.9 KiB
Python
109 lines
2.9 KiB
Python
from typing import Tuple
|
|
import json
|
|
import matplotlib.pyplot as plt
|
|
import openbb_terminal.session.local_model as Local
|
|
import openbb_terminal.session.hub_model as Hub
|
|
from openbb_terminal.session.user import User
|
|
from openbb_terminal.core.config.paths import PACKAGE_DIRECTORY
|
|
from openbb_terminal.helper_funcs import system_clear
|
|
from openbb_terminal.rich_config import console
|
|
from openbb_terminal import terminal_controller
|
|
|
|
|
|
def display_welcome_message():
|
|
"""Display welcome message"""
|
|
with open(PACKAGE_DIRECTORY / "session" / "banner.txt") as f:
|
|
console.print(f.read(), style="blue")
|
|
|
|
|
|
def get_user_input() -> Tuple[str, str, bool]:
|
|
"""Get user input
|
|
|
|
Returns
|
|
-------
|
|
Tuple[str, str, bool]
|
|
The user email, password and save login option.
|
|
"""
|
|
console.print("\nPlease enter your credentials:", style="info")
|
|
email = console.input("> Email: ", style="blue")
|
|
password = console.getpass("> Password: ", style="blue")
|
|
save_str = console.input("> Keep me logged in (y/n): ", style="blue").lower()
|
|
save = False
|
|
if save_str == "y":
|
|
save = True
|
|
|
|
return email, password, save
|
|
|
|
|
|
def create_session(email: str, password: str, save: bool) -> dict:
|
|
"""Create a session."""
|
|
|
|
session = Hub.get_session(email, password)
|
|
if session and save:
|
|
Local.save_session(session)
|
|
return session
|
|
|
|
|
|
def login_prompt(welcome=True, guest_allowed=True):
|
|
"""Login prompt and launch terminal if login is successful.
|
|
|
|
Parameters
|
|
----------
|
|
welcome : bool, optional
|
|
Display welcome message, by default True
|
|
"""
|
|
if welcome:
|
|
display_welcome_message()
|
|
|
|
while True:
|
|
email, password, save = get_user_input()
|
|
if not email and not password and guest_allowed:
|
|
return terminal_controller.parse_args_and_run()
|
|
session = create_session(email, password, save)
|
|
if isinstance(session, dict) and session:
|
|
break
|
|
|
|
login(session=session)
|
|
|
|
|
|
def login(session: dict):
|
|
"""Login and launch terminal.
|
|
|
|
Parameters
|
|
----------
|
|
session : dict
|
|
The session info.
|
|
"""
|
|
User.load_user_info(session)
|
|
response = Hub.fetch_user_configs(session)
|
|
if response:
|
|
if response.status_code == 200:
|
|
Local.apply_configs(configs=json.loads(response.content))
|
|
terminal_controller.parse_args_and_run()
|
|
else:
|
|
login_prompt(welcome=False)
|
|
else:
|
|
login_prompt(welcome=True)
|
|
|
|
|
|
def logout():
|
|
"""Logout and clear session."""
|
|
system_clear()
|
|
User.clean()
|
|
Hub.delete_session()
|
|
Local.remove_session_file()
|
|
plt.close("all")
|
|
|
|
|
|
def main(guest_allowed: bool = True):
|
|
"""Main function"""
|
|
local_session = Local.get_session()
|
|
if not local_session:
|
|
login_prompt(guest_allowed=guest_allowed)
|
|
else:
|
|
login(session=local_session)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|