Compatibility with RavenPack Edge

Author: dariosky

CHANGELOG.md

@@ -1,5 +1,14 @@
 # Changelog
 
+## v1.0.46 (2021-04-05)
+Compatibility with RavenPack Edge
+
+* loosen validation for entity-types (in Edge we have various dynamical
+  EDETs)
+  
+* loosen validation for RT fields (in Edge we have several new fields)
+
+
 ## v1.0.43 (2020-11-04)
 
 * Entity-mapping - expose the matching score and the candidates

ravenpackapi/core.py

@@ -10,12 +10,11 @@
 from ravenpackapi.models.results import Results
 from ravenpackapi.upload.module import UploadApi
 from ravenpackapi.util import to_curl
-from ravenpackapi.utils.constants import ENTITY_TYPES
 from ravenpackapi.utils.date_formats import as_datetime_str
 from ravenpackapi.utils.dynamic_sessions import DynamicSession
 
 _VALID_METHODS = ('get', 'post', 'put', 'delete', 'patch')
-VERSION = '1.0.45'
+VERSION = '1.0.46'
 
 logger = logging.getLogger("ravenpack.core")
 
@@ -175,7 +174,6 @@ def get_entity_reference(self, rp_entity_id):
     def get_entity_type_reference(self, entity_type=None):
         if entity_type:
             entity_type = entity_type.upper()
-        assert entity_type in ENTITY_TYPES, "Please provide a valid entity type, one of %s" % ENTITY_TYPES
         response = self.request(
             endpoint="/entity-reference",
             method='get',

ravenpackapi/models/results.py

@@ -46,6 +46,8 @@ def __str__(self):
     def __getattr__(self, field):
         normalized_field = field.upper()
         analytics_field = FIELD_MAP.get(normalized_field)
+        if normalized_field not in FIELD_MAP:  # accessing unknown fields - i.e. Edge ones
+            return self.data[field]
         if analytics_field is None:
             return self.__getattribute__(field)
 

ravenpackapi/utils/constants.py

@@ -1,6 +1,6 @@
 from setuptools import setup, find_packages
 
-VERSION = '1.0.45'
+VERSION = '1.0.46'
 
 with open('README.rst') as readme_file:
     readme = readme_file.read()