Skip to content

Commit

Permalink
Merge branch 'ft-782' of https://github.com/atlanhq/atlan-python into…
Browse files Browse the repository at this point in the history
… ft-782
  • Loading branch information
ErnestoLoma committed Nov 25, 2024
2 parents 8f44bb9 + 905d823 commit c3db9c7
Show file tree
Hide file tree
Showing 55 changed files with 1,293 additions and 3,946 deletions.
Original file line number Diff line number Diff line change
@@ -1,10 +1,10 @@
.. _applicationcontainer:
.. _businessprocessmodel:

ApplicationContainer
BusinessProcessModel
====================

.. module:: pyatlan.model.assets
:no-index:

.. autoclass:: ApplicationContainer
.. autoclass:: BusinessProcessModel
:members:
10 changes: 10 additions & 0 deletions docs/asset/businessprocessmodelentity.rst
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
.. _businessprocessmodelentity:

BusinessProcessModelEntity
==========================

.. module:: pyatlan.model.assets
:no-index:

.. autoclass:: BusinessProcessModelEntity
:members:
8 changes: 3 additions & 5 deletions docs/assets.rst
Original file line number Diff line number Diff line change
Expand Up @@ -31,8 +31,8 @@ You can interact with all of the following different kinds of assets:
asset/airflowtask
asset/anomalo
asset/anomalocheck
asset/app
asset/application
asset/applicationcontainer
asset/asset
asset/atlasglossary
asset/atlasglossarycategory
Expand All @@ -52,6 +52,8 @@ You can interact with all of the following different kinds of assets:
asset/businesspolicyexception
asset/businesspolicyincident
asset/businesspolicylog
asset/businessprocessmodel
asset/businessprocessmodelentity
asset/calculationview
asset/catalog
asset/cloud
Expand Down Expand Up @@ -83,10 +85,6 @@ You can interact with all of the following different kinds of assets:
asset/cubedimension
asset/cubefield
asset/cubehierarchy
asset/custom
asset/customdataset
asset/customfield
asset/customtable
asset/datacontract
asset/datadomain
asset/datamesh
Expand Down
3 changes: 3 additions & 0 deletions pyatlan/client/constants.py
Original file line number Diff line number Diff line change
Expand Up @@ -488,6 +488,9 @@
HTTPStatus.OK,
endpoint=EndPoint.HERACLES,
)
GET_ALL_CREDENTIALS = API(
CREDENTIALS_API, HTTPMethod.GET, HTTPStatus.OK, endpoint=EndPoint.HERACLES
)
UPDATE_CREDENTIAL_BY_GUID = API(
CREDENTIALS_API + "/{credential_guid}",
HTTPMethod.POST,
Expand Down
42 changes: 42 additions & 0 deletions pyatlan/client/credential.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,11 @@
from json import dumps
from typing import Any, Dict, Optional

from pydantic.v1 import validate_arguments

from pyatlan.client.common import ApiCaller
from pyatlan.client.constants import (
GET_ALL_CREDENTIALS,
GET_CREDENTIAL_BY_GUID,
TEST_CREDENTIAL,
UPDATE_CREDENTIAL_BY_GUID,
Expand All @@ -10,6 +14,7 @@
from pyatlan.model.credential import (
Credential,
CredentialResponse,
CredentialResponseList,
CredentialTestResponse,
)

Expand Down Expand Up @@ -47,6 +52,43 @@ def get(self, guid: str) -> CredentialResponse:
return raw_json
return CredentialResponse(**raw_json)

@validate_arguments
def get_all(
self,
filter: Optional[Dict[str, Any]] = None,
limit: Optional[int] = None,
offset: Optional[int] = None,
) -> CredentialResponseList:
"""
Retrieves all credentials.
:param filter: (optional) dictionary specifying the filter criteria.
:param limit: (optional) maximum number of credentials to retrieve.
:param offset: (optional) number of credentials to skip before starting retrieval.
:returns: CredentialResponseList instance.
:raises: AtlanError on any error during API invocation.
"""
params: Dict[str, Any] = {}
if filter is not None:
params["filter"] = dumps(filter)
if limit is not None:
params["limit"] = limit
if offset is not None:
params["offset"] = offset

raw_json = self._client._call_api(
GET_ALL_CREDENTIALS.format_path_with_params(), query_params=params
)

if not isinstance(raw_json, dict) or "records" not in raw_json:
raise ErrorCode.JSON_ERROR.exception_with_parameters(
"No records found in response",
400,
"API response did not contain the expected 'records' key",
)

return CredentialResponseList(**raw_json)

@validate_arguments
def test(self, credential: Credential) -> CredentialTestResponse:
"""
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -6,11 +6,11 @@
*,
name: str,
connection_qualified_name: str,
) -> ApplicationContainer:
) -> Application:
validate_required_fields(
["name", "connection_qualified_name"], [name, connection_qualified_name]
)
attributes = ApplicationContainer.Attributes.creator(
attributes = Application.Attributes.creator(
name=name,
connection_qualified_name=connection_qualified_name,
)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -6,11 +6,11 @@
*,
name: str,
connection_qualified_name: str,
) -> ApplicationContainer.Attributes:
) -> Application.Attributes:
validate_required_fields(
["name", "connection_qualified_name"], [name, connection_qualified_name]
)
return ApplicationContainer.Attributes(
return Application.Attributes(
name=name,
qualified_name=f"{connection_qualified_name}/{name}",
connection_qualified_name=connection_qualified_name,
Expand Down
14 changes: 6 additions & 8 deletions pyatlan/model/assets/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,11 +21,11 @@
"Folder",
"Airflow",
"DataContract",
"App",
"ADF",
"DataQuality",
"BI",
"Resource",
"Custom",
"DataMesh",
"SQL",
"NoSQL",
Expand All @@ -35,11 +35,11 @@
"Spark",
"Tag",
"SchemaRegistry",
"Application",
"Fivetran",
"Stakeholder",
"AirflowDag",
"AirflowTask",
"Application",
"AdfDataflow",
"AdfDataset",
"AdfPipeline",
Expand All @@ -64,9 +64,10 @@
"Function",
"TablePartition",
"Column",
"DatabricksUnityCatalogTag",
"SnowflakeStream",
"CalculationView",
"Database",
"CalculationView",
"Procedure",
"SnowflakeTag",
"CosmosMongoDB",
Expand All @@ -87,7 +88,6 @@
"ModelDataModel",
"SparkJob",
"SchemaRegistrySubject",
"ApplicationContainer",
"FivetranConnector",
"AnomaloCheck",
"MCIncident",
Expand Down Expand Up @@ -136,6 +136,7 @@
"event_store": ["EventStore"],
"insight": ["Insight"],
"a_p_i": ["API"],
"business_process_model": ["BusinessProcessModel"],
"google": ["Google"],
"azure": ["Azure"],
"a_w_s": ["AWS"],
Expand Down Expand Up @@ -167,10 +168,6 @@
"cube_hierarchy": ["CubeHierarchy"],
"cube_field": ["CubeField"],
"cube_dimension": ["CubeDimension"],
"custom_field": ["CustomField"],
"custom_dataset": ["CustomDataset"],
"custom_table": ["CustomTable"],
"databricks_unity_catalog_tag": ["DatabricksUnityCatalogTag"],
"kafka": ["Kafka"],
"azure_service_bus": ["AzureServiceBus"],
"dynamo_d_b": ["DynamoDB"],
Expand All @@ -181,6 +178,7 @@
"a_p_i_object": ["APIObject"],
"a_p_i_path": ["APIPath"],
"a_p_i_field": ["APIField"],
"business_process_model_entity": ["BusinessProcessModelEntity"],
"data_studio_asset": ["DataStudioAsset"],
"s3_bucket": ["S3Bucket"],
"s3_object": ["S3Object"],
Expand Down
24 changes: 10 additions & 14 deletions pyatlan/model/assets/__init__.pyi
Original file line number Diff line number Diff line change
Expand Up @@ -18,11 +18,11 @@ __all__ = [
"Folder",
"Airflow",
"DataContract",
"App",
"ADF",
"DataQuality",
"BI",
"Resource",
"Custom",
"DataMesh",
"SQL",
"NoSQL",
Expand All @@ -32,11 +32,11 @@ __all__ = [
"Spark",
"Tag",
"SchemaRegistry",
"Application",
"Fivetran",
"Stakeholder",
"AirflowDag",
"AirflowTask",
"Application",
"AdfDataflow",
"AdfDataset",
"AdfPipeline",
Expand All @@ -61,9 +61,10 @@ __all__ = [
"Function",
"TablePartition",
"Column",
"DatabricksUnityCatalogTag",
"SnowflakeStream",
"CalculationView",
"Database",
"CalculationView",
"Procedure",
"SnowflakeTag",
"CosmosMongoDB",
Expand All @@ -84,7 +85,6 @@ __all__ = [
"ModelDataModel",
"SparkJob",
"SchemaRegistrySubject",
"ApplicationContainer",
"FivetranConnector",
"AnomaloCheck",
"MCIncident",
Expand Down Expand Up @@ -133,6 +133,7 @@ __all__ = [
"NoSQL",
"Insight",
"API",
"BusinessProcessModel",
"Google",
"Azure",
"AWS",
Expand Down Expand Up @@ -164,10 +165,6 @@ __all__ = [
"CubeHierarchy",
"CubeField",
"CubeDimension",
"CustomField",
"CustomDataset",
"CustomTable",
"DatabricksUnityCatalogTag",
"Kafka",
"AzureServiceBus",
"DynamoDB",
Expand All @@ -178,6 +175,7 @@ __all__ = [
"APIObject",
"APIPath",
"APIField",
"BusinessProcessModelEntity",
"DataStudioAsset",
"S3Bucket",
"S3Object",
Expand Down Expand Up @@ -321,6 +319,8 @@ from .business_policy import BusinessPolicy
from .business_policy_exception import BusinessPolicyException
from .business_policy_incident import BusinessPolicyIncident
from .business_policy_log import BusinessPolicyLog
from .business_process_model import BusinessProcessModel
from .business_process_model_entity import BusinessProcessModelEntity
from .cloud import Cloud
from .cognite import Cognite
from .cognite3_d_model import Cognite3DModel
Expand Down Expand Up @@ -352,8 +352,8 @@ from .core.airflow_dag import AirflowDag
from .core.airflow_task import AirflowTask
from .core.anomalo import Anomalo
from .core.anomalo_check import AnomaloCheck
from .core.app import App
from .core.application import Application
from .core.application_container import ApplicationContainer
from .core.asset import Asset
from .core.atlas_glossary import AtlasGlossary
from .core.atlas_glossary_category import AtlasGlossaryCategory
Expand All @@ -369,13 +369,13 @@ from .core.cosmos_mongo_d_b import CosmosMongoDB
from .core.cosmos_mongo_d_b_account import CosmosMongoDBAccount
from .core.cosmos_mongo_d_b_collection import CosmosMongoDBCollection
from .core.cosmos_mongo_d_b_database import CosmosMongoDBDatabase
from .core.custom import Custom
from .core.data_contract import DataContract
from .core.data_domain import DataDomain
from .core.data_mesh import DataMesh
from .core.data_product import DataProduct
from .core.data_quality import DataQuality
from .core.database import Database
from .core.databricks_unity_catalog_tag import DatabricksUnityCatalogTag
from .core.dbt import Dbt
from .core.dbt_metric import DbtMetric
from .core.dbt_model import DbtModel
Expand Down Expand Up @@ -451,13 +451,9 @@ from .cube import Cube
from .cube_dimension import CubeDimension
from .cube_field import CubeField
from .cube_hierarchy import CubeHierarchy
from .custom_dataset import CustomDataset
from .custom_field import CustomField
from .custom_table import CustomTable
from .data_set import DataSet
from .data_studio import DataStudio
from .data_studio_asset import DataStudioAsset
from .databricks_unity_catalog_tag import DatabricksUnityCatalogTag
from .dbt_column_process import DbtColumnProcess
from .dbt_process import DbtProcess
from .dbt_tag import DbtTag
Expand Down
Loading

0 comments on commit c3db9c7

Please sign in to comment.