my-aws-helpers 2.2.0__tar.gz → 2.3.0__tar.gz

{my_aws_helpers-2.2.0 → my_aws_helpers-2.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: my_aws_helpers
-Version: 2.2.0
+Version: 2.3.0
 Summary: AWS Helpers
 Home-page: https://github.com/JarrodMccarthy/aws_helpers.git
 Author: Jarrod McCarthy

{my_aws_helpers-2.2.0 → my_aws_helpers-2.3.0}/my_aws_helpers/event.py

@@ -1,10 +1,11 @@
 from __future__ import annotations
-from my_aws_helpers.dynamo import BaseTableObject, DynamoSerialiser
+from typing import Optional
 from datetime import datetime
 from enum import Enum
 from uuid import uuid4
 from copy import copy
-
+from boto3.dynamodb.conditions import Key
+from my_aws_helpers.dynamo import BaseTableObject, DynamoSerialiser, BaseQueries, Dynamo
 
 class EventDynamoKeys:
     @staticmethod
@@ -71,3 +72,26 @@ class Event(BaseTableObject):
         obj['sk'] = self._get_sk()
         return DynamoSerialiser.object_serialiser(obj=obj)
 
+
+class EventQueries(BaseQueries):
+    def __init__(self, table_name, client: Optional[Dynamo] = None):
+        super().__init__(table_name, client)
+
+    def put_event(self, event: Event) -> Event:
+        response = self.client.put_item(item = event._to_dynamo_representation())
+        if response["ResponseMetadata"]["HTTPStatusCode"] == 200:
+            return event
+        raise Exception(f"Failed to put object into dynamo table")
+    
+    def get_event_by_id(self, id: str):
+        pk = EventDynamoKeys.get_event_pk(id=id)
+        sk = EventDynamoKeys.get_event_sk(id=id)
+        results = self.client.table.query(
+            KeyConditionExpression=Key("pk").eq(pk) & Key("sk").eq(sk)
+        )
+        if len(results["items"]) < 1:
+            return None
+        if len(results["Items"]) > 1:
+            raise Exception(f"Should only be 1 event with id {id}")
+        return Event._from_dynamo_representation(obj = results["Items"][0])
+

{my_aws_helpers-2.2.0 → my_aws_helpers-2.3.0}/my_aws_helpers.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: my-aws-helpers
-Version: 2.2.0
+Version: 2.3.0
 Summary: AWS Helpers
 Home-page: https://github.com/JarrodMccarthy/aws_helpers.git
 Author: Jarrod McCarthy

{my_aws_helpers-2.2.0 → my_aws_helpers-2.3.0}/setup.py

@@ -3,7 +3,7 @@ from setuptools import find_namespace_packages, setup
 
 base_path = os.path.abspath(os.path.dirname(__file__))
 
-version = "2.2.0"
+version = "2.3.0"
 
 setup(
     name="my_aws_helpers",