added IntegerForeignProperty and updated BatchSize on the SQSEvent

capless · Jun 28, 2018 · aca8eee · aca8eee
1 parent da41a25
commit aca8eee
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/sammy/__init__.py b/sammy/__init__.py
@@ -16,7 +16,8 @@
 from valley.contrib import Schema
 from valley.utils.json_utils import ValleyEncoderNoType
 
-from sammy.custom_properties import ForeignInstanceListProperty, CharForeignProperty
+from sammy.custom_properties import ForeignInstanceListProperty, \
+    CharForeignProperty, IntForeignProperty
 
 
 API_METHODS = {
@@ -156,7 +157,7 @@ class SQSEvent(EventSchema):
     _event_type = 'SQS'
 
     Queue = CharForeignProperty(Ref, required=True)
-    BatchSize = IntegerProperty()
+    BatchSize = IntForeignProperty(Ref)
 
 
 class KinesisEvent(EventSchema):

diff --git a/sammy/custom_properties.py b/sammy/custom_properties.py
@@ -4,7 +4,7 @@
 from valley.mixins import ListMixin
 from valley.properties import BaseProperty, MultiProperty
 from valley.utils.json_utils import ValleyEncoder
-from valley.validators import ForeignListValidator, StringValidator, ForeignValidator
+from valley.validators import ForeignListValidator, StringValidator, ForeignValidator, IntegerValidator
 
 
 class ForeignSubclassListValidator(ForeignListValidator):
@@ -58,3 +58,10 @@ def __init__(self,foreign_class,**kwargs):
                         StringValidator()],**kwargs)
 
 
+class IntForeignProperty(MultiProperty):
+
+    def __init__(self,foreign_class,**kwargs):
+
+        super(IntForeignProperty, self).__init__(
+            validators=[ForeignValidator(foreign_class),
+                        IntegerValidator()],**kwargs)