Chewing_Bever/didier
0
0
Fork 0
mirror of https://github.com/stijndcl/didier.git synced 2026-04-19 05:25:48 +02:00
Code Issues Projects Releases Wiki Activity

Use importlib for db scripts

Browse source
This commit is contained in:
stijndcl 2022-09-20 01:07:36 +02:00
parent 181118aa1d
commit 1fe04b3687
2 changed files with 9 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

0
database/scripts/db01_initial_easter_eggs.py Normal file
View file

19
run_db_scripts.py
View file

@ -4,14 +4,10 @@ This is slightly ugly, but running the scripts directly isn't possible because o
This could be cleaned up a bit using importlib but this is safer
"""
import asyncio
import importlib
import sys
from typing import Callable
from database.scripts.db00_example import main as debug_add_courses
script_mapping: dict[str, Callable] = {"debug_add_courses.py": debug_add_courses}
if __name__ == "__main__":
scripts = sys.argv[1:]
if not scripts:
@ -19,10 +15,13 @@ if __name__ == "__main__":
exit(1)
for script in scripts:
script_main = script_mapping.get(script.removeprefix("database/scripts/"), None)
if script_main is None:
script = script.replace("/", ".").removesuffix(".py")
module = importlib.import_module(script)
try:
script_main: Callable = module.main
asyncio.run(script_main())
print(f"Successfully ran {script}")
except AttributeError:
print(f'Script "{script}" not found.', file=sys.stderr)
exit(1)
asyncio.run(script_main())
print(f"Successfully ran {script}")