Added campaign completion feature. (gophish#3)

martindube · jordan-wright · commit 714207fe730c · 2017-02-06T22:30:35.000-05:00
diff --git a/gophish/api/api.py b/gophish/api/api.py
@@ -25,12 +25,13 @@ def __init__(self, api, endpoint=None, cls=None):
         self.endpoint = endpoint
         self._cls = cls
 
-    def get(self, resource_id=None):
+    def get(self, resource_id=None, resource_action=None):
         """ Gets the details for one or more resources by ID
         
         Args:
             cls - gophish.models.Model - The resource class
             resource_id - str - The endpoint (URL path) for the resource
+            resource_action - str - An action to perform on the resource
 
         Returns:
             One or more instances of cls parsed from the returned JSON
@@ -41,6 +42,9 @@ def get(self, resource_id=None):
         if resource_id:
             endpoint = '{}/{}'.format(endpoint, resource_id)
 
+        if resource_action:
+            endpoint = '{}/{}'.format(endpoint, resource_action)
+
         response = self.api.execute("GET", endpoint)
         if not response.ok:
             return Error.parse(response.json())
diff --git a/gophish/api/campaigns.py b/gophish/api/campaigns.py
@@ -30,6 +30,12 @@ def delete(self, campaign_id):
 
         return super(API, self).delete(campaign_id)
 
+    def complete(self, campaign_id):
+        """ Complete an existing campaign (Stop processing events) """
+
+        return super(API, self).get(resource_id=campaign_id, 
+                                    resource_action='complete')
+
     def summary(campaign_id=None):
         """ Returns the summary of one or more campaigns. """
         raise NotImplementedError
