ducklet/unwind
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

remove imdb users from config, load from db

Browse source
This commit is contained in:
ducklet 2021-07-08 09:56:37 +02:00
parent 60f0152f9d
commit cd42b95e49
2 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

9
unwind/__main__.py
View file

@ -4,9 +4,10 @@ import logging
from pathlib import Path from pathlib import Path
from . import config from . import config
from .db import close_connection_pool, open_connection_pool from .db import close_connection_pool, get_all, open_connection_pool
from .imdb import load_imdb from .imdb import load_imdb
from .imdb_import import import_from_file from .imdb_import import import_from_file
from .models import User
from .request import session from .request import session
log = logging.getLogger(__name__) log = logging.getLogger(__name__)
@ -18,9 +19,9 @@ async def run_load_user_ratings_from_imdb():
with session() as s: with session() as s:
s.headers["Accept-Language"] = "en-GB, en;q=0.5" s.headers["Accept-Language"] = "en-GB, en;q=0.5"
for name, imdb_id in config.imdb.items(): for user in await get_all(User):
log.info("Loading data for %s ... ⚡️", name) log.info("Loading data for %s ... ⚡️", user.name)
await load_imdb(imdb_id) await load_imdb(user.imdb_id)
await close_connection_pool() await close_connection_pool()

1
unwind/config.py
View file

@ -13,5 +13,4 @@ config_path = os.getenv("UNWIND_CONFIG", "./data/config.toml")
_config = toml.load(config_path) _config = toml.load(config_path)
imdb = _config["imdb"]
api_credentials = _config["api"]["credentials"] api_credentials = _config["api"]["credentials"]