fdc-shared-kernel 0.0.80__tar.gz → 0.0.83__tar.gz

{fdc_shared_kernel-0.0.80 → fdc_shared_kernel-0.0.83}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: fdc_shared_kernel
-Version: 0.0.80
+Version: 0.0.83
 Summary: Shared library for microservice
 Author-email: Shikhil S <shikhil.s@dbizsolution.com>, Ahammed Akdham N <ahammedakdham.n@dbizsolution.com>
 Classifier: Programming Language :: Python :: 3

{fdc_shared_kernel-0.0.80 → fdc_shared_kernel-0.0.83}/fdc_shared_kernel.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: fdc_shared_kernel
-Version: 0.0.80
+Version: 0.0.83
 Summary: Shared library for microservice
 Author-email: Shikhil S <shikhil.s@dbizsolution.com>, Ahammed Akdham N <ahammedakdham.n@dbizsolution.com>
 Classifier: Programming Language :: Python :: 3

{fdc_shared_kernel-0.0.80 → fdc_shared_kernel-0.0.83}/fdc_shared_kernel.egg-info/SOURCES.txt

@@ -13,8 +13,23 @@ shared_kernel/async_task_executor/async_task_executor.py
 shared_kernel/auth/__init__.py
 shared_kernel/auth/jwt_helper.py
 shared_kernel/auth/token_handler.py
+shared_kernel/auth/workbook_permission_handler.py
 shared_kernel/config/__init__.py
 shared_kernel/database/__init__.py
+shared_kernel/datatype_mappings/__init__.py
+shared_kernel/datatype_mappings/connectors_to_system/__init__.py
+shared_kernel/datatype_mappings/connectors_to_system/csv.py
+shared_kernel/datatype_mappings/connectors_to_system/db2.py
+shared_kernel/datatype_mappings/connectors_to_system/jira.py
+shared_kernel/datatype_mappings/connectors_to_system/mssql.py
+shared_kernel/datatype_mappings/connectors_to_system/mysql.py
+shared_kernel/datatype_mappings/connectors_to_system/oracle.py
+shared_kernel/datatype_mappings/connectors_to_system/postgres.py
+shared_kernel/datatype_mappings/connectors_to_system/redshift.py
+shared_kernel/datatype_mappings/connectors_to_system/salesforce.py
+shared_kernel/datatype_mappings/system_to_warehouse/__init__.py
+shared_kernel/datatype_mappings/system_to_warehouse/postgres.py
+shared_kernel/datatype_mappings/system_to_warehouse/redshift.py
 shared_kernel/enums/__init__.py
 shared_kernel/enums/async_task_executor.py
 shared_kernel/enums/status_tracker.py

{fdc_shared_kernel-0.0.80 → fdc_shared_kernel-0.0.83}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "fdc_shared_kernel"
-version = "0.0.80"
+version = "0.0.83"
 requires-python = ">=3.7"
 readme = "README_pypi.md"
 description = "Shared library for microservice"

{fdc_shared_kernel-0.0.80 → fdc_shared_kernel-0.0.83}/shared_kernel/auth/token_handler.py

@@ -19,15 +19,16 @@ def token_required(f):
 
         token = request.authorization.token
 
-        payload = token_handler.decode_token(token)
+        payload: dict = token_handler.decode_token(token)
 
         if 'error' in payload:
             raise Unauthorized("Failed to parse token")
 
         # Add user information to the request context
         current_user = {
-            "user_id": payload['user_id'],
-            "organization_id": payload['organization_id']
+            "user_id": payload.get('user_id'),
+            "organization_id": payload.get('organization_id'),
+            "organization_license_type": payload.get('organization_license_type'),
         }
 
         # ensure_sync is used to ensure that the decorated function is

fdc_shared_kernel-0.0.83/shared_kernel/auth/workbook_permission_handler.py

@@ -0,0 +1,92 @@
+from functools import wraps
+from flask import request, current_app
+import requests
+from shared_kernel.auth import JWTTokenHandler
+from shared_kernel.config import Config
+from shared_kernel.exceptions import Unauthorized, NotFound
+from shared_kernel.interfaces.databus import DataBus
+from shared_kernel.messaging import DataBusFactory
+from shared_kernel.registries.service_event_registry import ServiceEventRegistry
+
+
+config = Config()
+token_handler = JWTTokenHandler(config.get("JWT_SECRET_KEY"))
+service_event_registry = ServiceEventRegistry()
+databus: DataBus = DataBusFactory.create_data_bus(bus_type="HTTP", config={})
+
+
+def get_workbook_share_meta(workbook_id):
+    try:
+        event = {
+            "event_name": "GET_WORKBOOK_SHARE_META",
+            "event_payload": {"workbook_id": workbook_id},
+        }
+        response = databus.request_event(
+            getattr(service_event_registry, "GET_WORKBOOK_SHARE_META"), event
+        )
+        return response.get("data")
+
+    except requests.RequestException as e:
+        current_app.logger.error(f"Error fetching workbook: {e}")
+        return None
+
+
+def check_user_in_org(user, organization_id):
+    return user["organization_id"] == organization_id
+
+
+def has_workbook_share(user_id, shared_users):
+    return user_id in shared_users
+
+
+# Decorator to protect routes based on workbook permissions
+def workbook_permission_required(f):
+    @wraps(f)
+    def decorator(*args, **kwargs):
+        workbook_id = kwargs.get("workbook_id")
+        # Fetch the workbook share data from the master service
+        workbook_share_data = get_workbook_share_meta(workbook_id)
+        if not workbook_share_data:
+            raise NotFound("Workbook not found.")
+
+        # Public Access: No token required for public workbooks
+        if workbook_share_data.get("is_public"):
+            return current_app.ensure_sync(f)(None, *args, **kwargs)
+
+        # Non-public requests need token verification
+        token = None
+        if not request.authorization:
+            raise Unauthorized("Token is missing!")
+
+        token = request.authorization.token
+        payload = token_handler.decode_token(token)
+
+        if "error" in payload:
+            raise Unauthorized("Failed to parse token")
+
+        # Add user information to the request context
+        current_user = {
+            "user_id": payload["user_id"],
+            "organization_id": payload["organization_id"],
+        }
+
+        # Owner Access
+        if current_user["user_id"] == workbook_share_data.get("created_by"):
+            return current_app.ensure_sync(f)(current_user, *args, **kwargs)
+
+        # Organization Access
+        if workbook_share_data.get("shared_with_org") and check_user_in_org(
+            current_user, workbook_share_data["organization_id"]
+        ):
+            return current_app.ensure_sync(f)(current_user, *args, **kwargs)
+
+        # Specific User Access
+        if has_workbook_share(
+            current_user["user_id"], workbook_share_data["shared_users"]
+        ):
+            return current_app.ensure_sync(f)(current_user, *args, **kwargs)
+
+        # If none of the conditions are met, deny access
+        raise Unauthorized("You don't have permission to access this workbook.")
+
+    return decorator

fdc_shared_kernel-0.0.83/shared_kernel/datatype_mappings/connectors_to_system/__init__.py

@@ -0,0 +1 @@
+from shared_kernel.datatype_mappings.connectors_to_system.redshift import redshift_to_system

fdc_shared_kernel-0.0.83/shared_kernel/datatype_mappings/connectors_to_system/csv.py

@@ -0,0 +1 @@
+csv_to_system = {}

fdc_shared_kernel-0.0.83/shared_kernel/datatype_mappings/connectors_to_system/db2.py

@@ -0,0 +1 @@
+db2_to_system = {}

fdc_shared_kernel-0.0.83/shared_kernel/datatype_mappings/connectors_to_system/jira.py

@@ -0,0 +1 @@
+jira_to_system = {}

fdc_shared_kernel-0.0.83/shared_kernel/datatype_mappings/connectors_to_system/mssql.py

@@ -0,0 +1 @@
+mssql_to_system = {}

fdc_shared_kernel-0.0.83/shared_kernel/datatype_mappings/connectors_to_system/mysql.py

@@ -0,0 +1 @@
+mysql_to_system = {}

fdc_shared_kernel-0.0.83/shared_kernel/datatype_mappings/connectors_to_system/oracle.py

@@ -0,0 +1 @@
+oracle_to_system = {}

fdc_shared_kernel-0.0.83/shared_kernel/datatype_mappings/connectors_to_system/postgres.py

@@ -0,0 +1,16 @@
+postgres_to_system = {
+    "INTEGER": "integer",
+    "REAL": "real",
+    "NUMERIC": "numeric",
+    "DOUBLE PRECISION": "double",
+    "BOOLEAN": "boolean",
+    "BYTEA": "binary",
+    "BIT": "bit",
+    "TEXT": "string",
+    "VARCHAR": "varchar",
+    "CHAR": "character",
+    "DATE": "date",
+    "TIMESTAMP": "datetime",
+    "TIME": "time",
+    "NULL": "null"
+}

fdc_shared_kernel-0.0.83/shared_kernel/datatype_mappings/connectors_to_system/redshift.py

@@ -0,0 +1,14 @@
+redshift_to_system = {
+    "BIGINT": "integer",
+    "INTEGER": "integer",
+    "REAL": "real",
+    "DECIMAL": "numeric",
+    "DOUBLE PRECISION": "double",
+    "BOOLEAN": "boolean",
+    "VARBYTE": "binary",
+    "VARCHAR": "string",
+    "CHARACTER VARYING": "string",
+    "DATE": "date",
+    "TIMESTAMPTZ": "datetime",
+    "NULL": "null"
+}

fdc_shared_kernel-0.0.83/shared_kernel/datatype_mappings/connectors_to_system/salesforce.py

@@ -0,0 +1 @@
+salesforce_to_system = {}

fdc_shared_kernel-0.0.83/shared_kernel/datatype_mappings/system_to_warehouse/__init__.py

@@ -0,0 +1 @@
+from shared_kernel.datatype_mappings.system_to_warehouse.redshift import system_to_redshift

fdc_shared_kernel-0.0.83/shared_kernel/datatype_mappings/system_to_warehouse/postgres.py

@@ -0,0 +1,28 @@
+system_to_postgres = {
+    "integer": "INTEGER",
+    "int": "INTEGER",
+    "real": "REAL",
+    "bigint": "BIGINT",
+    "long": "BIGINT",
+    "numeric": "NUMERIC",
+    "number": "NUMERIC",
+    "decimal": "NUMERIC",
+    "double": "DOUBLE PRECISION",
+    "boolean": "BOOLEAN",
+    "binary": "BYTEA",
+    "bit": "BIT",
+    "id": "TEXT",
+    "reference": "TEXT",
+    "string": "TEXT",
+    "text": "TEXT",
+    "picklist": "TEXT",
+    "character varying": "VARCHAR",
+    "character": "CHAR",
+    "varchar": "VARCHAR",
+    "date": "DATE",
+    "timestamp": "TIMESTAMP",
+    "datetime": "TIMESTAMP",
+    "time": "TIME",
+    "none": "NULL",
+    "null": "NULL"
+}

fdc_shared_kernel-0.0.83/shared_kernel/datatype_mappings/system_to_warehouse/redshift.py

@@ -0,0 +1,28 @@
+system_to_redshift = {
+    "integer": "BIGINT",
+    "int": "BIGINT",
+    "real": "REAL",
+    "bigint": "BIGINT",
+    "long": "BIGINT",
+    "numeric": "DECIMAL",
+    "number": "DECIMAL",
+    "decimal": "DECIMAL",
+    "double": "DOUBLE PRECISION",
+    "boolean": "BOOLEAN",
+    "binary": "VARBYTE",
+    "bit": "BOOLEAN",
+    "id": "VARCHAR",
+    "reference": "VARCHAR(255)",
+    "string": "VARCHAR(65535)",
+    "text": "VARCHAR(65535)",
+    "picklist": "VARCHAR(255)",
+    "character varying": "VARCHAR(65535)",
+    "character": "VARCHAR(65535)",
+    "varchar": "VARCHAR(65535)",
+    "date": "DATE",
+    "timestamp": "TIMESTAMPTZ",
+    "datetime": "TIMESTAMPTZ",
+    "time": "VARCHAR(255)",
+    "none": "NULL",
+    "null": "NULL"
+}

{fdc_shared_kernel-0.0.80 → fdc_shared_kernel-0.0.83}/shared_kernel/registries/service_event_registry.py

@@ -39,3 +39,8 @@ class ServiceEventRegistry:
         config.get("MASTER_SERVICE_BASE_ENDPOINT")
         + "/event/subscribe-sync-event/SET_TRACKING_ID/"
     )
+
+    GET_WORKBOOK_SHARE_META = (
+        config.get("MASTER_SERVICE_BASE_ENDPOINT")
+        + "/event/subscribe-sync-event/GET_WORKBOOK_SHARE_META/"
+    )