diff --git a/lofar_stingray/extract.py b/lofar_stingray/extract.py index 065cb7f6066da0a83b6336d3fafde75bdb32dc1b..47938fd21aac74f73b9f7d2b9db9907a6c66d462 100644 --- a/lofar_stingray/extract.py +++ b/lofar_stingray/extract.py @@ -1,4 +1,4 @@ -# Copyright (C) 2024 ASTRON (Netherlands Institute for Radio Astronomy) +# Copyright (C) 2025 ASTRON (Netherlands Institute for Radio Astronomy) # SPDX-License-Identifier: Apache-2.0 """Entry point to extract statistic matrices & metadata to store them as HDF5""" @@ -6,6 +6,7 @@ import argparse import logging import os +import sys import tempfile from datetime import datetime, timedelta from urllib.parse import urlparse @@ -80,13 +81,13 @@ def _round_datetime_ms(timestamp: datetime) -> datetime: return timestamp - timedelta(microseconds=subtract_ms) -def main(sys_args) -> int: +def main(sys_args=None) -> int: """Parser main method""" logger.debug("Starting hdf5 converter") parser = _create_parser() args = parser.parse_args(sys_args) - minio_client = get_minio_client(args) + minio_client = get_minio_client(args or sys.argv[1:]) logger.info("Using source %s", args.source) # connect to storage @@ -155,6 +156,4 @@ def main(sys_args) -> int: if __name__ == "__main__": - import sys - sys.exit(main(sys.argv[1:]))