boto3-assist 0.1.3__tar.gz → 0.1.5__tar.gz

{boto3_assist-0.1.3 → boto3_assist-0.1.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: boto3_assist
-Version: 0.1.3
+Version: 0.1.5
 Summary: Additional boto3 wrappers to make your life a little easier
 Author-email: Eric Wilson <boto3-assist@geekcafe.com>
 License-File: LICENSE-EXPLAINED.txt

{boto3_assist-0.1.3 → boto3_assist-0.1.5}/pyproject.toml

@@ -7,7 +7,7 @@ packages = ["src/boto3_assist"]
 
 [project]
 name = "boto3_assist"
-version = "0.1.3"
+version = "0.1.5"
 authors = [
   { name="Eric Wilson", email="boto3-assist@geekcafe.com" }
 ]

{boto3_assist-0.1.3 → boto3_assist-0.1.5}/src/boto3_assist/dynamodb/dynamodb.py

@@ -372,6 +372,7 @@ class DynamoDB(DynamoDBConnection):
         do_projections: bool = False,
         ascending: bool = False,
         strongly_consistent: bool = False,
+        limit: Optional[int] = None,
     ) -> dict:
         """Helper function to list by criteria"""
 
@@ -391,6 +392,7 @@ class DynamoDB(DynamoDBConnection):
             expression_attribute_names=expression_attribute_names,
             ascending=ascending,
             strongly_consistent=strongly_consistent,
+            limit=limit,
         )
 
         return response

{boto3_assist-0.1.3 → boto3_assist-0.1.5}/src/boto3_assist/dynamodb/dynamodb_importer.py

@@ -38,9 +38,11 @@ class DynamoDBImporter:
             raise FileNotFoundError(f"File not found: {json_file_path}")
         if json_file_path.endswith(".gz"):
             data = self.read_gzip_file(json_file_path)
-        else:
+        elif json_file_path.endswith(".json"):
             with open(json_file_path, "r", encoding="utf-8") as json_file:
                 data = json.load(json_file)
+        else:
+            raise ValueError(f"Unsupported file type: {json_file_path}")
 
         # table = self.db.dynamodb_resource.Table(self.table_name)
         # with table.batch_writer() as batch:
@@ -58,7 +60,22 @@ class DynamoDBImporter:
     def import_json_files(self, json_file_paths: list[str]) -> None:
         """Import multiple json files into the database"""
         for json_file_path in json_file_paths:
-            self.import_json_file(json_file_path)
+            if os.path.exists(json_file_path) is False:
+                raise FileNotFoundError(f"File not found: {json_file_path}")
+            else:
+                if json_file_path.endswith(".gz") or json_file_path.endswith(".json"):
+                    self.import_json_file(json_file_path)
+                else:
+                    if os.path.isdir(json_file_path):
+                        logger.warning(
+                            f"Unsupported sub directory import {json_file_path}. "
+                            "Skipping import on this file."
+                        )
+                    else:
+                        logger.warning(
+                            f"Unsupported file type: {json_file_path}. "
+                            "Skipping import on this file.  Files should end with .gz or .json"
+                        )
 
     def read_gzip_file(self, file_path: str) -> List[dict]:
         """

boto3_assist-0.1.5/src/boto3_assist/version.py

@@ -0,0 +1 @@
+__version__ = '0.1.5'

boto3_assist-0.1.3/src/boto3_assist/version.py

@@ -1 +0,0 @@
-__version__ = '0.1.3'