Fix table creation

Author: pentest-gg

fastapi_users_db_dynamodb/_aioboto3_patch.py

@@ -3,8 +3,8 @@
 
 import aiobotocore.endpoint
 import aiobotocore.retryhandler
-from aiobotocore.endpoint import HttpxStreamingBody, StreamingBody
-from aiobotocore.retryhandler import ChecksumError, logger
+from aiobotocore.endpoint import HttpxStreamingBody, StreamingBody  # type: ignore
+from aiobotocore.retryhandler import ChecksumError, logger  # type: ignore
 
 try:
     import httpx
@@ -83,5 +83,5 @@ async def convert_to_response_dict(http_response, operation_model):
     return response_dict
 
 
-aiobotocore.retryhandler.AioCRC32Checker._check_response = _fixed_check_response
+aiobotocore.retryhandler.AioCRC32Checker._check_response = _fixed_check_response  # type: ignore
 aiobotocore.endpoint.convert_to_response_dict = convert_to_response_dict

tests/tables.py

@@ -1,11 +1,12 @@
 import aioboto3
+import botocore.exceptions
 
 
 async def ensure_table_exists(session: aioboto3.Session, table_name: str, region: str):
-    async with session.resource("dynamodb", region_name=region) as client:
+    async with session.client("dynamodb", region_name=region) as client:
         try:
             await client.describe_table(TableName=table_name)
-        except client.exceptions.ResourceNotFoundException:
+        except botocore.exceptions.ClientError:
             await client.create_table(
                 TableName=table_name,
                 KeySchema=[