ninjajiraiya
/
devine
forked from DRMTalks/devine
1
0
Fork 0
Code Pull Requests Activity

Efficiently store data in auth list, sort the data

This commit is contained in:
rlaphoenix 2023-02-07 21:07:58 +00:00
parent dd441bcd85
commit 794de8b516
1 changed files with 9 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
devine/commands/auth.py
View File

@ -1,5 +1,6 @@
import logging import logging
import tkinter.filedialog import tkinter.filedialog
from collections import defaultdict
from pathlib import Path from pathlib import Path
from typing import Optional from typing import Optional
@ -36,27 +37,23 @@ def list_(ctx: click.Context, service: Optional[str] = None) -> None:
log = ctx.obj log = ctx.obj
service_f = service service_f = service
profiles: dict[str, dict[str, list]] = {} auth_data: dict[str, dict[str, list]] = defaultdict(lambda: defaultdict(list))
for cookie_dir in config.directories.cookies.iterdir(): for cookie_dir in config.directories.cookies.iterdir():
service = cookie_dir.name service = cookie_dir.name
profiles[service] = {}
for cookie in cookie_dir.glob("*.txt"): for cookie in cookie_dir.glob("*.txt"):
if cookie.stem not in profiles[service]: if cookie.stem not in auth_data[service]:
profiles[service][cookie.stem] = ["Cookie"] auth_data[service][cookie.stem].append("Cookie")
for service, credentials in config.credentials.items(): for service, credentials in config.credentials.items():
if service not in profiles: for profile in credentials:
profiles[service] = {} auth_data[service][profile].append("Credential")
for profile, credential in credentials.items():
if profile not in profiles[service]:
profiles[service][profile] = []
profiles[service][profile].append("Credential")
for service, profiles in profiles.items(): for service, profiles in dict(sorted(auth_data.items())).items(): # type:ignore
if service_f and service != service_f.upper(): if service_f and service != service_f.upper():
continue continue
log.info(service) log.info(service)
for profile, authorizations in profiles.items(): for profile, authorizations in dict(sorted(profiles.items())).items():
log.info(f' "{profile}": {", ".join(authorizations)}') log.info(f' "{profile}": {", ".join(authorizations)}')