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

dictature/backend/webdav.py

@@ -0,0 +1,141 @@
+import io
+import posixpath
+from typing import Iterable
+
+try:
+    from webdav3.client import Client as WebdavClient
+    from webdav3.exceptions import (
+        WebDavException,
+        RemoteResourceNotFound,
+        RemoteParentNotFound,
+        MethodNotSupported,
+    )
+except ImportError:
+    raise ImportError('Requires: pip install webdavclient3')
+
+from .mock import DictatureTableMock, DictatureBackendMock, Value, ValueMode, ValueSerializer, ValueSerializerMode
+
+
+class DictatureBackendWebdav(DictatureBackendMock):
+    def __init__(
+            self,
+            client: WebdavClient,
+            dir_prefix: str = 'db_',
+            item_prefix: str = 'item_'
+    ) -> None:
+        self.__client = client
+        self.__dir_prefix = dir_prefix
+        self.__item_prefix = item_prefix
+
+    def keys(self) -> Iterable[str]:
+        try:
+            resources = self.__client.list(get_info=True)
+
+            for resource_info in resources:
+                name = posixpath.basename(posixpath.dirname(resource_info.get('path')))
+                is_dir = resource_info.get('isdir', False)
+
+                if not name:
+                    continue
+
+                if is_dir and name.startswith(self.__dir_prefix):
+                    table_name_encoded = name[len(self.__dir_prefix):]
+                    # noinspection PyProtectedMember
+                    yield DictatureTableWebdav._filename_decode(table_name_encoded, suffix='')
+        except RemoteResourceNotFound:
+            pass
+
+    def table(self, name: str) -> 'DictatureTableMock':
+        return DictatureTableWebdav(self.__client, name, self.__dir_prefix, self.__item_prefix)
+
+
+class DictatureTableWebdav(DictatureTableMock):
+    def __init__(self, client: WebdavClient, name: str, db_prefix: str, prefix: str) -> None:
+        self.__client = client
+        self.__encoded_name = self._filename_encode(name, suffix='')
+        self.__path = posixpath.join(db_prefix + self.__encoded_name)
+        self.__prefix = prefix
+        self.__name_serializer = ValueSerializer(mode=ValueSerializerMode.filename_only)
+        self.__value_serializer = ValueSerializer(mode=ValueSerializerMode.any_string)
+
+    def keys(self) -> Iterable[str]:
+        try:
+            resources = self.__client.list(self.__path, get_info=True)
+
+            for resource_info in resources:
+                name = posixpath.basename(resource_info.get('path'))
+                is_dir = resource_info.get('isdir', False)
+
+                if not name:
+                    continue
+
+                if not is_dir and name.startswith(self.__prefix):
+                    item_name_encoded = name[len(self.__prefix):]
+                    yield self._filename_decode(item_name_encoded, suffix='.txt')
+
+        except RemoteResourceNotFound:
+            pass
+
+    def drop(self) -> None:
+        try:
+            self.__client.clean(self.__path)
+        except RemoteResourceNotFound:
+            pass
+
+    def create(self) -> None:
+        try:
+            if not self.__client.check(self.__path):
+                self.__client.mkdir(self.__path)
+        except MethodNotSupported as e:
+            if not self.__client.check(self.__path):
+                raise e
+
+    def set(self, item: str, value: Value) -> None:
+        item_path = self.__item_path(item)
+        self.create()
+
+        save_data: str = self.__value_serializer.serialize(value)
+        data_bytes = save_data.encode('utf-8')
+
+        with io.BytesIO(data_bytes) as buffer:
+            self.__client.upload_to(buffer, item_path)
+
+    def get(self, item: str) -> Value:
+        item_path = self.__item_path(item)
+        buffer = io.BytesIO()
+        try:
+            self.__client.download_from(buffer, item_path)
+            buffer.seek(0)
+            save_data = buffer.read().decode('utf-8')
+        except RemoteResourceNotFound:
+            raise KeyError(item)
+        return self.__value_serializer.deserialize(save_data)
+
+    def delete(self, item: str) -> None:
+        """
+        Deletes the file corresponding to the item. No error if not found.
+        Raises WebDavException for other deletion errors.
+        """
+        item_path = self.__item_path(item)
+        try:
+            self.__client.clean(item_path)
+        except RemoteResourceNotFound:
+            pass
+
+    def __item_path(self, item: str) -> str:
+        encoded_item_name = self.__prefix + self._filename_encode(item, suffix='.txt')
+        full_path = posixpath.join(self.__path, encoded_item_name)
+        return full_path
+
+    @staticmethod
+    def _filename_encode(name: str, suffix: str = '.txt') -> str:
+        return ValueSerializer(mode=ValueSerializerMode.filename_only).serialize(Value(
+            value=name,
+            mode=ValueMode.string.value
+        )) + suffix
+
+    @staticmethod
+    def _filename_decode(name: str, suffix: str = '.txt') -> str:
+        if suffix and name.endswith(suffix):
+            name = name[:-len(suffix)]
+        return ValueSerializer(mode=ValueSerializerMode.filename_only).deserialize(name).value

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dictature
-Version: 0.9.7
+Version: 0.10.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
@@ -54,6 +54,7 @@ Currently, the following backends are supported:
 - `DictatureBackendDirectory`: stores the data in a directory as json files
 - `DictatureBackendSQLite`: stores the data in a SQLite database
 - `DictatureBackendMISP`: stores the data in a MISP instance
+- `DictatureBackendWebdav`: stores data in a WebDav share as files
 
 ### Transformers
 

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

@@ -5,6 +5,7 @@ dictature/backend/directory.py,sha256=nPNJ-wRZOsuzaCOLxgGnNB093qID5LWw7VELAqi7kZ
 dictature/backend/misp.py,sha256=iPjvgnJg6WveNP2wvgN7OK2vkX-SC9qYPrdoa9ahRT0,4411
 dictature/backend/mock.py,sha256=BzfLstxkTIjk6mcMTdFKj8rSaFgIqn9-2Cyelslj8bY,5889
 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
 dictature/transformer/aes.py,sha256=ZhC1dT9QpnziErkDLriWLgXDEFNGQW0KG4aqSN2AZpA,1926
 dictature/transformer/gzip.py,sha256=pngvJQeALa-lv98VBeJ1Pl6_gaAfGcPXD9UR7PexrYA,1921
@@ -12,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.9.7.dist-info/LICENSE,sha256=n1U9DKr8sM5EY2QHcvxSGiKTDWUT8MyXsOC79w94MT0,1072
-dictature-0.9.7.dist-info/METADATA,sha256=oyBWco6Qtzo3C6MsHl_tdco6Oqo0mTyIsyXeui7sfjo,2869
-dictature-0.9.7.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
-dictature-0.9.7.dist-info/top_level.txt,sha256=-RO39WWCF44lqiXhSUcACVqbk6SkgReZTz7ZmHKH3-U,10
-dictature-0.9.7.dist-info/RECORD,,
+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,,