dictature 0.10.0__py3-none-any.whl → 0.11.0__py3-none-any.whl

dictature/backend/misp.py

@@ -3,7 +3,7 @@ from typing import Iterable, Optional
 from .mock import DictatureTableMock, DictatureBackendMock, Value, ValueMode, ValueSerializer, ValueSerializerMode
 
 try:
-    from pymisp import PyMISP, MISPEvent, MISPAttribute
+    from pymisp import PyMISP, MISPEvent, MISPAttribute, MISPTag
 except ImportError as e:
     raise ImportError("Please install the 'pymisp' package to use the 'DictatureBackendMISP' backend.") from e
 
@@ -92,6 +92,15 @@ class DictatureTableMISP(DictatureTableMock):
                     self.__event = event
                     break
             else:
+                # Check if tag exists, create if not
+                existing_tags = self.__misp.search_tags(self.__tag, strict_tagname=True, pythonify=True)
+                if not type(existing_tags) is list or not existing_tags:
+                    tag = MISPTag()
+                    tag.name = self.__tag
+                    tag.colour = '#000000'
+                    tag.exportable = True
+                    self.__misp.add_tag(tag)
+
                 event = MISPEvent()
                 event.info = self.__event_description
                 event.distribution = 0

dictature/backend/mock.py

@@ -74,7 +74,7 @@ class ValueSerializer:
         if self.__mode == ValueSerializerMode.hex_only:
             allowed_chars = hexdigits
         elif self.__mode == ValueSerializerMode.filename_only:
-            allowed_chars = ascii_letters + digits + '_.'
+            allowed_chars = ascii_letters + digits + ' -_.'
         elif self.__mode in (ValueSerializerMode.any_string, ValueSerializerMode.ascii_only):
             allowed_chars = None
         else:

dictature/dictature.py

@@ -16,6 +16,7 @@ class Dictature:
             name_transformer: MockTransformer = PassthroughTransformer(),
             value_transformer: MockTransformer = PassthroughTransformer(),
             table_name_transformer: Optional[MockTransformer] = None,
+            allow_pickle: bool = True,
     ) -> None:
         """
         Create a new Dictature object
@@ -23,6 +24,7 @@ class Dictature:
         :param name_transformer: transformer to use for table and key names
         :param value_transformer: transformer to use for values
         :param table_name_transformer: transformer to use for table names, if None, name_transformer is used
+        :param allow_pickle: if True, pickle is allowed for values (warning: this may be a security risk if the data is not trusted)
         """
         self.__backend = backend
         self.__table_cache: Dict[str, "DictatureTable"] = {}
@@ -30,6 +32,7 @@ class Dictature:
         self.__value_transformer = value_transformer
         self.__table_name_transformer = table_name_transformer or name_transformer
         self.__cache_size = 4096
+        self.__allow_pickle = allow_pickle
 
     def keys(self) -> Set[str]:
         """
@@ -80,7 +83,8 @@ class Dictature:
                 self.__backend,
                 item,
                 name_transformer=self.__name_transformer,
-                value_transformer=self.__value_transformer
+                value_transformer=self.__value_transformer,
+                allow_pickle=self.__allow_pickle,
             )
         return self.__table_cache[item]
 
@@ -114,7 +118,8 @@ class DictatureTable:
             backend: DictatureBackendMock,
             table_name: str,
             name_transformer: MockTransformer = PassthroughTransformer(),
-            value_transformer: MockTransformer = PassthroughTransformer()
+            value_transformer: MockTransformer = PassthroughTransformer(),
+            allow_pickle: bool = True
     ):
         """
         Create a new DictatureTable object
@@ -122,12 +127,14 @@ class DictatureTable:
         :param table_name: name of the table
         :param name_transformer:  transformer to use for key names
         :param value_transformer: transformer to use for values
+        :param allow_pickle: if True, pickle is allowed for values (warning: this may be a security risk if the data is not trusted)
         """
         self.__backend = backend
         self.__name_transformer = name_transformer
         self.__value_transformer = value_transformer
         self.__table = self.__backend.table(self.__table_key(table_name))
         self.__table_created = False
+        self.__allow_pickle = allow_pickle
 
     def get(self, item: str, default: Optional[Any] = None) -> Any:
         """
@@ -210,6 +217,8 @@ class DictatureTable:
         elif mode == ValueMode.json:
             return json.loads(value)
         elif mode == ValueMode.pickle:
+            if not self.__allow_pickle:
+                raise ValueError("Pickle is not allowed")
             return pickle.loads(decompress(b64decode(value.encode('ascii'))))
         raise ValueError(f"Unknown mode '{mode}'")
 
@@ -228,6 +237,8 @@ class DictatureTable:
                 value = json.dumps(value)
                 value_mode = ValueMode.json.value
             except TypeError:
+                if not self.__allow_pickle:
+                    raise ValueError("Pickle is not allowed")
                 value = b64encode(compress(pickle.dumps(value))).decode('ascii')
                 value_mode = ValueMode.pickle.value
 

{dictature-0.10.0.dist-info → dictature-0.11.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dictature
-Version: 0.10.0
+Version: 0.11.0
 Summary: dictature -- A generic wrapper around dict-like interface with mulitple backends
 Author-email: Adam Hlavacek <git@adamhlavacek.com>
 Project-URL: Homepage, https://github.com/esoadamo/dictature

{dictature-0.10.0.dist-info → dictature-0.11.0.dist-info}/RECORD

@@ -1,9 +1,9 @@
 dictature/__init__.py,sha256=UCPJKHeyirRZ0pCYoyeat-rwXa8pDezOJ3UWCipDdyc,33
-dictature/dictature.py,sha256=SHwG_XvGwm6qpvMt5OjeS8BoU5wLgJi_4jIBKFmYFyI,9330
+dictature/dictature.py,sha256=q4QFQz91r2boViuhsf6y-C9uRntaMD-VagKFUGZjnpQ,10015
 dictature/backend/__init__.py,sha256=d5s6QCJOUzFglVNg8Cqqx_8b61S-AOTGjEUIF6FS69U,149
 dictature/backend/directory.py,sha256=nPNJ-wRZOsuzaCOLxgGnNB093qID5LWw7VELAqi7kZc,3384
-dictature/backend/misp.py,sha256=iPjvgnJg6WveNP2wvgN7OK2vkX-SC9qYPrdoa9ahRT0,4411
-dictature/backend/mock.py,sha256=BzfLstxkTIjk6mcMTdFKj8rSaFgIqn9-2Cyelslj8bY,5889
+dictature/backend/misp.py,sha256=R9osuEJa79SoXErwByIwhViJ26cEfw2nKhWu8XxkO7Y,4860
+dictature/backend/mock.py,sha256=o4YBl6Wk-q6IHFdykGeryTKcQFsJHvymlONPbbwVkak,5891
 dictature/backend/sqlite.py,sha256=zyphYEeLY4eGuBCor16i80_-brdipMpXZ3_kONwErsE,5237
 dictature/backend/webdav.py,sha256=Y-3_WTcMyKVUnsVjiUZAxuy10FK0Yr-7Mgn13clg3po,5039
 dictature/transformer/__init__.py,sha256=JIFJpXU6iB9hIUM8L7HL2o9Nqjm_YbMEuQBQC8ZJ6b4,124
@@ -13,8 +13,8 @@ dictature/transformer/hmac.py,sha256=vURsB0HlzRPn_Vkl7lGmZV9OKempQuds8AanmadDxIo
 dictature/transformer/mock.py,sha256=7zu65ZqUV_AVRaPSzNd73cVMXixXt31SeuX9OKZxaJQ,948
 dictature/transformer/passthrough.py,sha256=Pt3N6G_Qh6HJ_q75ETL5nfAwYHLB-SjkVwUwbbbMik8,344
 dictature/transformer/pipeline.py,sha256=OaQaJeJ5NpICetJe08r8ontqstsXGuW8jDbKw1zxYs4,842
-dictature-0.10.0.dist-info/LICENSE,sha256=n1U9DKr8sM5EY2QHcvxSGiKTDWUT8MyXsOC79w94MT0,1072
-dictature-0.10.0.dist-info/METADATA,sha256=rRiKQ1lciPHmU2f2iwhUxQGG_OOLPWBmp-njyl9pv2k,2937
-dictature-0.10.0.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
-dictature-0.10.0.dist-info/top_level.txt,sha256=-RO39WWCF44lqiXhSUcACVqbk6SkgReZTz7ZmHKH3-U,10
-dictature-0.10.0.dist-info/RECORD,,
+dictature-0.11.0.dist-info/LICENSE,sha256=n1U9DKr8sM5EY2QHcvxSGiKTDWUT8MyXsOC79w94MT0,1072
+dictature-0.11.0.dist-info/METADATA,sha256=cLMkq0S6mDBlPJXwitBNbZfpZ_KuqogCLoR4EGhb8zQ,2937
+dictature-0.11.0.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
+dictature-0.11.0.dist-info/top_level.txt,sha256=-RO39WWCF44lqiXhSUcACVqbk6SkgReZTz7ZmHKH3-U,10
+dictature-0.11.0.dist-info/RECORD,,