Model Save method returns a Thingy object

numberly · Mar 14, 2018 · 4fa343f · 4fa343f
1 parent d9869af
commit 4fa343f
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/appnexus/model.py b/appnexus/model.py
@@ -74,7 +74,7 @@ def save(self, **kwargs):
             result = self.create(payload, **kwargs)
         else:
             result = self.modify(payload, id=self.id, **kwargs)
-        return result
+        return type(self)(result)
 
 
 class Campaign(Model):
@@ -86,9 +86,8 @@ def profile(self):
 
 class Report(Model):
 
-    @classmethod
-    def download(cls, **kwargs):
-        return cls.client.get("report-download", **kwargs)
+    def download(self, **kwargs):
+        return self.client.get("report-download", id=self.report_id)
 
 
 def create_models(services_list):