Add paths in argparser for relative github action paths

mitodl · Jan 30, 2024 · ffd940a · ffd940a
1 parent 632c5dc
commit ffd940a
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 5 deletions.
diff --git a/ocw_oer_export/cli.py b/ocw_oer_export/cli.py
@@ -23,11 +23,22 @@ def main():
         default="api",
         help="Specify data source for CSV creation (default: api)",
     )
-
+    parser.add_argument(
+        "--input_path",
+        default="/private/output/ocw_api_data.json",
+        help="Output path for the CSV file",
+    )
+    parser.add_argument(
+        "--output_path",
+        default="/private/output/ocw_oer_export.csv",
+        help="Output path for the CSV file",
+    )
     args = parser.parse_args()
 
     if args.create_csv:
-        create_csv(source=args.source)
+        create_csv(
+            source=args.source, input_path=args.input_path, output_path=args.output_path
+        )
     elif args.create_json:
         create_json()
     else:

diff --git a/ocw_oer_export/create_csv.py b/ocw_oer_export/create_csv.py
@@ -186,7 +186,7 @@ def transform_data(data):
 
 def create_csv(
     source="api",
-    input_file="/private/output/ocw_api_data.json",
+    input_path="/private/output/ocw_api_data.json",
     output_path="/private/output/ocw_oer_export.csv",
 ):
     """
@@ -199,7 +199,7 @@ def create_csv(
         api_data_json = extract_data_from_api(api_url=API_URL)
 
     elif source == "json":
-        api_data_json = extract_data_from_json(input_file)
+        api_data_json = extract_data_from_json(input_path)
 
     else:
         raise ValueError("Invalid source. Use 'api' or 'json'.")

diff --git a/tests/test_csv_creation_from_json.py b/tests/test_csv_creation_from_json.py
@@ -20,7 +20,7 @@ def test_csv_creation_from_json(self):
         """Test the CSV generated from JSON matches the expected CSV content."""
         create_csv(
             source="json",
-            input_file=self.sample_json_path,
+            input_path=self.sample_json_path,
             output_path=self.generated_csv_path,
         )
         generated_csv_data = extract_data_from_file(self.generated_csv_path)