add validate_or_make_namedtuples

mailerlite/constants.py

@@ -51,3 +51,11 @@
 for nt in [Subscriber, Field, Group, Activity, Segment, Meta, Pagination,
            Campaign, Stats, Webhook]:
     nt.__new__.__defaults__ = (None,) * len(nt._fields)
+
+
+def validate_or_make_namedtuples(obj, keys):
+
+    if not set(keys).issubset(obj._fields):
+        diff = set(list(obj._fields)).symmetric_difference(set(keys))
+        return namedtuple(f'{obj.__name__}2', obj._fields + tuple(diff))
+    return obj

mailerlite/tests/test_constants.py

@@ -0,0 +1,23 @@
+import ipdb
+from mailerlite.constants import validate_or_make_namedtuples, Segment
+
+
+def test_validate_or_make_namedtuples():
+
+    new_segment = validate_or_make_namedtuples(Segment,
+                                               ['id', 'title', 'filter',
+                                                'total', 'sent', 'opened',
+                                                'clicked','created_at',
+                                                'updated_at', 'timed_out'])
+
+    assert new_segment == Segment
+
+    new_segment = validate_or_make_namedtuples(Segment,
+                                               ['id', 'title', 'filter',
+                                                'total', 'sent',
+                                                'opened', 'clicked',
+                                                'created_at', 'updated_at',
+                                                'timed_out', 'custom'])
+
+    assert new_segment != Segment
+    assert 'custom' in new_segment._fields