Updating Attachment Model with __init__

fix from gophish#24
EuanKerr · Oct 4, 2022 · ffda10b · ffda10b
1 parent 1a8a6be
commit ffda10b
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/gophish/models.py b/gophish/models.py
@@ -395,6 +395,10 @@ def parse(cls, json):
 class Attachment(Model):
     _valid_properties = {'content': None, 'type': None, 'name': None}
 
+     def __init__(self, **kwargs):
+        for key, default in Attachment._valid_properties.items():
+            setattr(self, key, kwargs.get(key, default))
+
     @classmethod
     def parse(cls, json):
         attachment = cls()