Only add custom attribute names to the mapping. (#826)

pynamodb · Sep 3, 2020 · ef704dc · ef704dc
1 parent cc6bafa
commit ef704dc
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pynamodb/attributes.py b/pynamodb/attributes.py
@@ -237,10 +237,10 @@ def _initialize_attributes(cls):
                 initialized = attribute._make_attribute()
 
             cls._attributes[name] = attribute
-            if attribute.attr_name is not None:
-                cls._dynamo_to_python_attrs[attribute.attr_name] = name
-            else:
+            if attribute.attr_name is None:
                 attribute.attr_name = name
+            if attribute.attr_name != name:
+                cls._dynamo_to_python_attrs[attribute.attr_name] = name
 
             if initialized and isinstance(attribute, MapAttribute):
                 # To support creating expressions from nested attributes, MapAttribute instances