mirror of https://github.com/stijndcl/didier
Fix db script runner
parent
969c66e2bf
commit
405545a9b0
2
.flake8
2
.flake8
|
@ -32,7 +32,7 @@ max-line-length = 120
|
||||||
# Disable some rules for entire files
|
# Disable some rules for entire files
|
||||||
per-file-ignores =
|
per-file-ignores =
|
||||||
# DALL000: Missing __all__, main isn't supposed to be imported
|
# DALL000: Missing __all__, main isn't supposed to be imported
|
||||||
main.py: DALL000,
|
main.py: DALL000, run_db_scripts.py: DALL000,
|
||||||
# DALL000: Missing __all__, Cogs aren't modules
|
# DALL000: Missing __all__, Cogs aren't modules
|
||||||
./didier/cogs/*: DALL000,
|
./didier/cogs/*: DALL000,
|
||||||
# DALL000: Missing __all__, tests aren't supposed to be imported
|
# DALL000: Missing __all__, tests aren't supposed to be imported
|
||||||
|
|
|
@ -8,7 +8,9 @@ import importlib
|
||||||
import sys
|
import sys
|
||||||
from typing import Callable
|
from typing import Callable
|
||||||
|
|
||||||
if __name__ == "__main__":
|
|
||||||
|
async def main():
|
||||||
|
"""Try to parse all command-line arguments into modules and run them sequentially"""
|
||||||
scripts = sys.argv[1:]
|
scripts = sys.argv[1:]
|
||||||
if not scripts:
|
if not scripts:
|
||||||
print("No scripts provided.", file=sys.stderr)
|
print("No scripts provided.", file=sys.stderr)
|
||||||
|
@ -20,8 +22,12 @@ if __name__ == "__main__":
|
||||||
|
|
||||||
try:
|
try:
|
||||||
script_main: Callable = module.main
|
script_main: Callable = module.main
|
||||||
asyncio.run(script_main())
|
await script_main()
|
||||||
print(f"Successfully ran {script}")
|
print(f"Successfully ran {script}")
|
||||||
except AttributeError:
|
except AttributeError:
|
||||||
print(f'Script "{script}" not found.', file=sys.stderr)
|
print(f'Script "{script}" not found.', file=sys.stderr)
|
||||||
exit(1)
|
exit(1)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
asyncio.run(main())
|
||||||
|
|
Loading…
Reference in New Issue