Skip to content

Commit ac5297e

Browse files
committed
CU-869azeyvz: Add logging to download scripts
1 parent dfe9fe2 commit ac5297e

File tree

2 files changed

+14
-3
lines changed

2 files changed

+14
-3
lines changed

medcat-v2/medcat/__main__.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,13 +3,16 @@
33

44
def main(*args: str):
55
if not args:
6-
print("Usage: python -m medcat download-scripts [DEST]",
6+
print("Usage: python -m medcat download-scripts [DEST] [log_level]",
77
file=sys.stderr)
88
sys.exit(1)
99
if len(args) >= 1 and args[0] == "download-scripts":
10-
from medcat.utils.download_scripts import fetch_scripts
10+
from medcat.utils.download_scripts import main
1111
dest = args[1] if len(args) > 1 else "."
12-
fetch_scripts(dest)
12+
kwargs = {}
13+
if len(args) > 2:
14+
kwargs["log_level"] = args[2].upper()
15+
main(dest, **kwargs)
1316
else:
1417
print("Usage: python -m medcat download-scripts [DEST]",
1518
file=sys.stderr)

medcat-v2/medcat/utils/download_scripts.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -90,3 +90,11 @@ def fetch_scripts(destination: str | Path = ".") -> Path:
9090

9191
logger.info("Scripts extracted to: %s", dest / SCRIPTS_PATH)
9292
return dest / SCRIPTS_PATH
93+
94+
95+
def main(destination: str = ".",
96+
log_level: int | str = logging.INFO):
97+
logger.setLevel(log_level)
98+
if not logger.handlers:
99+
logger.addHandler(logging.StreamHandler())
100+
fetch_scripts(destination)

0 commit comments

Comments
 (0)