From 2cddc482d6ea24d44aa19484d9124765e26b2b59 Mon Sep 17 00:00:00 2001
From: "E. Joshua Rigler" <erigler@usgs.gov>
Date: Wed, 18 May 2022 10:43:32 -0600
Subject: [PATCH] Attempt StringIO after failed BytesIO file read
This is a bandaid to allow the `--input-url` option to be used to
read in single files. It adds zero functionality beyond what already
existed when the `--input-file` option was used.
---
geomagio/Controller.py | 9 +++++++--
1 file changed, 7 insertions(+), 2 deletions(-)
diff --git a/geomagio/Controller.py b/geomagio/Controller.py
index d05f45ae..41bffe20 100644
--- a/geomagio/Controller.py
+++ b/geomagio/Controller.py
@@ -1,7 +1,7 @@
"""Controller class for geomag algorithms"""
import argparse
-from io import BytesIO
+from io import BytesIO, StringIO
import sys
from typing import List, Optional, Tuple, Union
@@ -506,7 +506,12 @@ def get_input_factory(args):
input_factory_args["urlInterval"] = args.input_url_interval
input_factory_args["urlTemplate"] = args.input_url
else:
- input_stream = BytesIO(Util.read_url(args.input_url))
+ try:
+ input_stream = BytesIO(Util.read_url(args.input_url))
+ except TypeError as e:
+ print(str(e), file=sys.stderr)
+ print("Warning: reading url as BytesIO failed; attempting StringIO", file=sys.stderr)
+ input_stream = StringIO(Util.read_url(args.input_url))
input_type = args.input
if input_type == "edge":
input_factory = edge.EdgeFactory(
--
GitLab