skyflow-flowvault-python 1.0.0__py3-none-any.whl

skyflow/vault/data/_get_response_record.py

@@ -0,0 +1,17 @@
+class GetResponseRecord:
+    def __init__(self, skyflow_id=None, table_name=None, tokens=None, data=None,
+                 hashed_data=None, http_code=None, error=None, request_id=None):
+        self.skyflow_id = skyflow_id
+        self.table_name = table_name
+        self.tokens = tokens
+        self.data = data
+        self.hashed_data = hashed_data
+        self.http_code = http_code
+        self.error = error
+        self.request_id = request_id
+
+    def __repr__(self):
+        return ("GetResponseRecord(skyflow_id={}, table_name={}, tokens={}, data={}, hashed_data={}, "
+                "http_code={}, error={}, request_id={})").format(
+            self.skyflow_id, self.table_name, self.tokens, self.data, self.hashed_data,
+            self.http_code, self.error, self.request_id)

skyflow/vault/data/_insert_options.py

@@ -0,0 +1,8 @@
+from typing import Callable, Optional
+
+from ._request_context import RequestContext
+
+
+class InsertOptions:
+    def __init__(self, interceptor: Optional[Callable[[RequestContext], None]] = None):
+        self.interceptor = interceptor

skyflow/vault/data/_insert_request.py

@@ -0,0 +1,11 @@
+from typing import List
+
+from ._insert_request_record import InsertRequestRecord
+from ._upsert_options import UpsertOptions
+
+
+class InsertRequest:
+    def __init__(self, records: List[InsertRequestRecord], table_name: str = None, upsert: UpsertOptions = None):
+        self.records = records
+        self.table_name = table_name
+        self.upsert = upsert

skyflow/vault/data/_insert_request_record.py

@@ -0,0 +1,9 @@
+from ._upsert_options import UpsertOptions
+
+
+class InsertRequestRecord:
+    def __init__(self, data: dict, table_name: str = None, tokens: dict = None, upsert: UpsertOptions = None):
+        self.data = data
+        self.table_name = table_name
+        self.tokens = tokens
+        self.upsert = upsert

skyflow/vault/data/_insert_response.py

@@ -0,0 +1,9 @@
+class InsertResponse:
+    def __init__(self, records=None):
+        self.records = records
+
+    def __repr__(self):
+        return f"InsertResponse(records={self.records})"
+
+    def __str__(self):
+        return self.__repr__()

skyflow/vault/data/_insert_response_record.py

@@ -0,0 +1,17 @@
+class InsertResponseRecord:
+    def __init__(self, skyflow_id=None, table_name=None, tokens=None, data=None,
+                 hashed_data=None, http_code=None, error=None, request_id=None):
+        self.skyflow_id = skyflow_id
+        self.table_name = table_name
+        self.tokens = tokens
+        self.data = data
+        self.hashed_data = hashed_data
+        self.http_code = http_code
+        self.error = error
+        self.request_id = request_id
+
+    def __repr__(self):
+        return ("InsertResponseRecord(skyflow_id={}, table_name={}, tokens={}, data={}, hashed_data={}, "
+                "http_code={}, error={}, request_id={})").format(
+            self.skyflow_id, self.table_name, self.tokens, self.data, self.hashed_data,
+            self.http_code, self.error, self.request_id)

skyflow/vault/data/_request_context.py

@@ -0,0 +1,16 @@
+NOT_BATCHED = -1
+
+
+class RequestContext:
+    def __init__(self, operation, batch_index=NOT_BATCHED, total_batches=NOT_BATCHED):
+        self.operation = operation
+        self.batch_index = batch_index
+        self.total_batches = total_batches
+        self._headers = {}
+
+    def add_header(self, key, value):
+        self._headers[key] = value
+
+    @property
+    def headers(self):
+        return dict(self._headers)

skyflow/vault/data/_token.py

@@ -0,0 +1,8 @@
+class Token:
+    def __init__(self, token: str = None, token_group_name: str = None, path: str = None):
+        self.token = token
+        self.token_group_name = token_group_name
+        self.path = path
+
+    def __repr__(self):
+        return f"Token(token={self.token}, token_group_name={self.token_group_name}, path={self.path})"

skyflow/vault/data/_token_group_redactions.py

@@ -0,0 +1,4 @@
+class TokenGroupRedactions:
+    def __init__(self, token_group_name: str = None, redaction: str = None):
+        self.token_group_name = token_group_name
+        self.redaction = redaction

skyflow/vault/data/_update_options.py

@@ -0,0 +1,8 @@
+from typing import Callable, Optional
+
+from ._request_context import RequestContext
+
+
+class UpdateOptions:
+    def __init__(self, interceptor: Optional[Callable[[RequestContext], None]] = None):
+        self.interceptor = interceptor

skyflow/vault/data/_update_request.py

@@ -0,0 +1,11 @@
+from typing import List
+
+from skyflow.utils.enums import UpsertType
+from ._update_request_record import UpdateRequestRecord
+
+
+class UpdateRequest:
+    def __init__(self, records: List[UpdateRequestRecord], table_name: str = None, update_type: UpsertType = None):
+        self.records = records
+        self.table_name = table_name
+        self.update_type = update_type

skyflow/vault/data/_update_request_record.py

@@ -0,0 +1,6 @@
+class UpdateRequestRecord:
+    def __init__(self, skyflow_id: str = None, data: dict = None, tokens: dict = None, table_name: str = None):
+        self.skyflow_id = skyflow_id
+        self.data = data
+        self.tokens = tokens
+        self.table_name = table_name

skyflow/vault/data/_update_response.py

@@ -0,0 +1,9 @@
+class UpdateResponse:
+    def __init__(self, records=None):
+        self.records = records
+
+    def __repr__(self):
+        return f"UpdateResponse(records={self.records})"
+
+    def __str__(self):
+        return self.__repr__()

skyflow/vault/data/_update_response_record.py

@@ -0,0 +1,17 @@
+class UpdateResponseRecord:
+    def __init__(self, skyflow_id=None, table_name=None, tokens=None, data=None,
+                 hashed_data=None, http_code=None, error=None, request_id=None):
+        self.skyflow_id = skyflow_id
+        self.table_name = table_name
+        self.tokens = tokens
+        self.data = data
+        self.hashed_data = hashed_data
+        self.http_code = http_code
+        self.error = error
+        self.request_id = request_id
+
+    def __repr__(self):
+        return ("UpdateResponseRecord(skyflow_id={}, table_name={}, tokens={}, data={}, hashed_data={}, "
+                "http_code={}, error={}, request_id={})").format(
+            self.skyflow_id, self.table_name, self.tokens, self.data, self.hashed_data,
+            self.http_code, self.error, self.request_id)

skyflow/vault/data/_upsert_options.py

@@ -0,0 +1,7 @@
+from skyflow.utils.enums import UpsertType
+
+
+class UpsertOptions:
+    def __init__(self, unique_columns: list = None, update_type: UpsertType = None):
+        self.unique_columns = unique_columns
+        self.update_type = update_type