Support reserved word resource properties (#1178)

stripe · Dec 18, 2023 · 3a34544 · 3a34544
1 parent d2b8ed1
commit 3a34544
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/stripe/_stripe_object.py b/stripe/_stripe_object.py
@@ -146,6 +146,8 @@ def __getattr__(self, k):
                 raise AttributeError(k)
 
             try:
+                if k in self._field_remappings:
+                    k = self._field_remappings[k]
                 return self[k]
             except KeyError as err:
                 raise AttributeError(*err.args)
@@ -530,6 +532,8 @@ def __deepcopy__(self, memo: Dict[int, Any]) -> "StripeObject":
 
         return copied
 
+    _field_remappings: ClassVar[Dict[str, str]] = {}
+
     _inner_class_types: ClassVar[Dict[str, Type["StripeObject"]]] = {}
     _inner_class_dicts: ClassVar[List[str]] = []
 

diff --git a/tests/test_stripe_object.py b/tests/test_stripe_object.py
@@ -376,3 +376,10 @@ def serialize(self, previous):
         )
 
         assert obj.serialize(None) == {"nested": ""}
+
+    def test_field_name_remapping(self):
+        class Foo(stripe.stripe_object.StripeObject):
+            _field_remappings = {"getter_name": "data_name"}
+
+        obj = Foo.construct_from({"data_name": "foo"}, "mykey")
+        assert obj.getter_name == "foo"