diff options
Diffstat (limited to 'src/wikiget')
| -rw-r--r-- | src/wikiget/parse.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/src/wikiget/parse.py b/src/wikiget/parse.py index 9bad59a..6396b3f 100644 --- a/src/wikiget/parse.py +++ b/src/wikiget/parse.py @@ -15,6 +15,7 @@ # You should have received a copy of the GNU General Public License # along with Wikiget. If not, see <https://www.gnu.org/licenses/>. +import fileinput import logging from argparse import Namespace from typing import Dict @@ -65,9 +66,12 @@ def get_dest(dl: str, args: Namespace) -> File: def read_batch_file(batch_file: str) -> Dict[int, str]: dl_list = {} - logger.info(f"Using batch file '{batch_file}'.") + if batch_file == "-": + logger.info("Using stdin for batch download") + else: + logger.info(f"Using file '{batch_file}' for batch download") - with open(batch_file) as fd: + with fileinput.input(batch_file) as fd: # read the file into memory and process each line as we go for line_num, line in enumerate(fd, start=1): line_s = line.strip() |