dahjah/grawlix
1
0
Fork 0
mirror of https://github.com/jo1gi/grawlix.git synced 2026-03-25 20:28:45 -06:00
Code Issues Actions Packages Projects Releases Wiki Activity

Ask for login credentials if missing

Browse Source
This commit is contained in:
Joakim Holm 2024-03-24 19:45:45 +01:00
parent f9be630241
commit e48babbd1d
1 changed files with 21 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
grawlix/__main__.py
View File

@ -6,6 +6,7 @@ from .output import download_book
from . import arguments, logging
from typing import Tuple, Optional
from rich.prompt import Prompt
from rich.progress import Progress
from functools import partial
import os
@ -13,6 +14,12 @@ import asyncio
import traceback
def get_or_ask(attr: str, hidden: bool, source_config: Optional[SourceConfig], options) -> str:
return getattr(options, attr, None) \
or getattr(source_config, attr, None) \
or Prompt.ask(attr.capitalize(), password=hidden)
def get_login(source: Source, config: Config, options) -> Tuple[str, str, Optional[str]]:
"""
Get login credentials for source
@ -23,14 +30,20 @@ def get_login(source: Source, config: Config, options) -> Tuple[str, str, Option
:returns: Login credentials
"""
source_name = source.name.lower().replace(" ", "")
if source_name in config.sources:
username = config.sources[source_name].username or options.username
password = config.sources[source_name].password or options.password
library = config.sources[source_name].library or options.library
else:
username = options.username
password = options.password
library = options.library
source_config = config.sources.get(source_name)
username = get_or_ask("username", False, source_config, options)
password = get_or_ask("password", True, source_config, options)
library = None # TODO
# if source_name in config.sources:
# username = config.sources[source_name].username or options.username
# password = config.sources[source_name].password or options.password
# library = config.sources[source_name].library or options.library
# else:
# username = options.username
# password = options.password
# library = options.library
return username, password, library