add time partitioning API support

jgarman · jgarman · commit d9755d25dca5 · 2017-04-11T15:48:43.000-04:00
diff --git a/src/cbapi/response/__init__.py b/src/cbapi/response/__init__.py
@@ -4,6 +4,6 @@
 
 from .models import (
     BannedHash, Site, ThrottleRule, Alert, Feed, Sensor, User, Watchlist, Investigation, ThreatReport, Binary, Process,
-    SensorGroup, FeedAction, WatchlistAction, TaggedEvent
+    SensorGroup, FeedAction, WatchlistAction, TaggedEvent, StoragePartition
 )
 from .rest_api import CbEnterpriseResponseAPI, CbResponseAPI
diff --git a/src/cbapi/response/models.py b/src/cbapi/response/models.py
@@ -70,6 +70,43 @@
 #     urlobject = '/api/v1/binary'
 
 
+class StoragePartitionQuery(SimpleQuery):
+    @property
+    def results(self):
+        if not self._full_init:
+            self._results = []
+            for k,v in iteritems(self._cb.get_object(self._urlobject, default={})):
+                t = self._doc_class.new_object(self._cb, v, full_doc=True)
+                if self._match_query(t):
+                    self._results.append(t)
+            self._results = self._sort(self._results)
+            self._full_init = True
+
+        return self._results
+
+
+class StoragePartition(NewBaseModel):
+    urlobject = "/api/v1/storage/events/partition"
+    primary_key = "name"
+
+    @classmethod
+    def _query_implementation(cls, cb):
+        return StoragePartitionQuery(cls, cb)
+
+    def _refresh(self):
+        # there is no GET method for a StoragePartition.
+        return True
+
+    def delete(self):
+        self._cb.delete_object("/api/v1/storage/events/{0}".format(self._model_unique_id))
+
+    def unmount(self):
+        self._cb.post_object("/api/v1/storage/events/{0}/unmount".format(self._model_unique_id), None)
+
+    def mount(self):
+        self._cb.post_object("/api/v1/storage/events/{0}/mount".format(self._model_unique_id), None)
+
+
 class BannedHash(MutableBaseModel, CreatableModelMixin):
     urlobject = "/api/v1/banning/blacklist"
     swagger_meta_file = "response/models/hash_blacklist.yaml"
diff --git a/src/cbapi/response/rest_api.py b/src/cbapi/response/rest_api.py
@@ -174,8 +174,17 @@ def from_ui(self, uri):
     def _request_lr_session(self, sensor_id):
         return self.live_response.request_session(sensor_id)
 
-    def _close_lr_session(self, sensor_id):
-        return self.live_response.close_session(sensor_id)
+    def create_new_partition(self):
+        """Create a new Solr time partition for event storage. Available in Cb Response 6.1 and above.
+        This will force roll-over current hot partition into warm partition (by renaming it to a time-stamped name)
+        and create a new hot partition ("writer").
+
+        :returns: Nothing if successful.
+        :raises ApiError: if there was an error creating the new partition.
+        :raises ServerError: if there was an error creating the new partition.
+        """
+        self.post_object("/api/v1/storage/events/new_partition", None)
+
 
 class CbEnterpriseResponseAPI(CbResponseAPI):
     """

Original file line number	Diff line number	Diff line change
`@@ -4,6 +4,6 @@`
`4`	`4`
`5`	`5`	`from .models import (`
`6`	`6`	`BannedHash, Site, ThrottleRule, Alert, Feed, Sensor, User, Watchlist, Investigation, ThreatReport, Binary, Process,`
`7`		`- SensorGroup, FeedAction, WatchlistAction, TaggedEvent`
	`7`	`+ SensorGroup, FeedAction, WatchlistAction, TaggedEvent, StoragePartition`
`8`	`8`	`)`
`9`	`9`	`from .rest_api import CbEnterpriseResponseAPI, CbResponseAPI`