fix: api endpoint for save and reset action

resilient-tech · Sep 24, 2024 · 322cc4b · 322cc4b
1 parent 65a2a70
commit 322cc4b
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/india_compliance/gst_india/api_classes/taxpayer_returns.py b/india_compliance/gst_india/api_classes/taxpayer_returns.py
@@ -123,3 +123,15 @@ def get_files(self, gstin, token, otp=None):
             return response
 
         return FilesAPI().get_all(response)
+
+    def save_or_reset_action(self, action, gstin, data, otp=None):
+        ACTION_MAP = {"RESET": "RESETIMS", "SAVE": "SAVE"}
+
+        gov_data = {"rtin": gstin, "reqtyp": action, "invdata": data}
+
+        return self.get(
+            action=ACTION_MAP[action],
+            endpoint=self.END_POINT,
+            json=gov_data,
+            otp=otp,
+        )