diff --git a/src/openhound/core/convert.py b/src/openhound/core/convert.py index df718a7..5af52e1 100644 --- a/src/openhound/core/convert.py +++ b/src/openhound/core/convert.py @@ -10,7 +10,6 @@ from dlt.pipeline.pipeline import Pipeline from openhound.core.asset import BaseAsset -from openhound.core.clients.bloodhound import BloodHound from openhound.core.logging import logger_override from openhound.core.lookup import LookupManager from openhound.core.pipeline import BasePipeline @@ -22,13 +21,6 @@ logger = logging.getLogger(__name__) -@dataclass -class Credentials: - url: str - token_key: str - token_id: str - - class Method(str, Enum): write = "write" ingest = "ingest" @@ -52,16 +44,6 @@ def __init__( self.lookup = lookup self.progress = progress self.method = method - self.client: BloodHound | None = None - self.upload_id: int | None = None - - @property - def _credentials(self) -> Credentials: - return Credentials( - url=dlt.secrets["destination.bloodhound.url"], - token_id=dlt.secrets["destination.bloodhound.token_id"], - token_key=dlt.secrets["destination.bloodhound.token_key"], - ) @property def pipeline(self) -> Pipeline: diff --git a/src/openhound/destinations/bloodhound/destination.py b/src/openhound/destinations/bloodhound/destination.py index 43dc4c6..4273d06 100644 --- a/src/openhound/destinations/bloodhound/destination.py +++ b/src/openhound/destinations/bloodhound/destination.py @@ -29,7 +29,7 @@ class Permissions(str, Enum): def saved_searches( items: TDataItems, table: TTableSchema, - url: str = dlt.secrets.value, + url: str = dlt.config.value, token: str = dlt.secrets.value, strategy: Strategy = Strategy.skip, permissions: Permissions = Permissions.public, @@ -68,7 +68,7 @@ def saved_searches( def privilege_zones( items: TDataItems, table: TTableSchema, - url: str = dlt.secrets.value, + url: str = dlt.config.value, token: str = dlt.secrets.value, strategy: Strategy = Strategy.skip, ):