Allow for json str as arg

CIROH-UA · Dec 4, 2023 · 36634c5 · 36634c5
1 parent a9b5a5b
commit 36634c5
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/forcingprocessor/src/forcingprocessor/forcingprocessor.py b/forcingprocessor/src/forcingprocessor/forcingprocessor.py
@@ -796,12 +796,15 @@ def prep_ngen_data(conf):
     )
     args = parser.parse_args()
 
-    if 's3' in args.infile:
-        os.system(f'wget {args.infile}')
-        filename = args.infile.split('/')[-1]
-        conf = json.load(open(filename))
+    if args.infile[0] == '{':
+        conf = json.loads(args.infile)
     else:
-        conf = json.load(open(args.infile))
+        if 's3' in args.infile:
+            os.system(f'wget {args.infile}')
+            filename = args.infile.split('/')[-1]
+            conf = json.load(open(filename))
+        else:
+            conf = json.load(open(args.infile))
 
     prep_ngen_data(conf)