Merge pull request #7 from cape-ph/capepy2

fix: update to capepy v2
cape-ph · Jan 28, 2025 · 5d45e79 · 5d45e79
2 parents 7f8bcb7 + 5ce868e
commit 5d45e79
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/main.py b/main.py
@@ -16,7 +16,7 @@
 
 # the response should contain a StreamingBody object that needs to be converted
 # to a file like object to make the pdf libraries happy
-f = io.BytesIO(etl_job.get_raw_file())
+f = io.BytesIO(etl_job.get_src_file())
 
 try:
     # get the report date from the 4th line of the pdf
@@ -65,4 +65,4 @@
 # write out the transformed data
 with io.StringIO() as csv_buff:
     interim.to_csv(csv_buff, index=False)
-    etl_job.write_clean_file(csv_buff.getvalue(), clean_obj_key)
+    etl_job.write_sink_file(csv_buff.getvalue(), clean_obj_key)
diff --git a/requirements.txt b/requirements.txt
@@ -1,6 +1,6 @@
 aws-glue-libs @ git+https://github.com/awslabs/aws-glue-libs@9d8293962e6ffc607e5dc328e246f40b24010fa8
 boto3==1.34.103
-capepy>=1.0.0,<2.0.0
+capepy>=2.0.0,<3.0.0
 pandas==2.2.2
 pyspark==3.5.1
 python-docx==1.1.2