acryl-datahub 1.0.0rc8__py3-none-any.whl → 1.0.0rc9__py3-none-any.whl

datahub/metadata/_schema_classes.py

@@ -6272,6 +6272,7 @@ class VersionPropertiesClass(_Aspect):
         sortId: str,
         aliases: Optional[List["VersionTagClass"]]=None,
         comment: Union[None, str]=None,
+        versioningScheme: Optional[Union[str, "VersioningSchemeClass"]]=None,
         sourceCreatedTimestamp: Union[None, "AuditStampClass"]=None,
         metadataCreatedTimestamp: Union[None, "AuditStampClass"]=None,
         isLatest: Union[None, bool]=None,
@@ -6287,6 +6288,11 @@ class VersionPropertiesClass(_Aspect):
             self.aliases = aliases
         self.comment = comment
         self.sortId = sortId
+        if versioningScheme is None:
+            # default: 'LEXICOGRAPHIC_STRING'
+            self.versioningScheme = self.RECORD_SCHEMA.fields_dict["versioningScheme"].default
+        else:
+            self.versioningScheme = versioningScheme
         self.sourceCreatedTimestamp = sourceCreatedTimestamp
         self.metadataCreatedTimestamp = metadataCreatedTimestamp
         self.isLatest = isLatest
@@ -6297,6 +6303,7 @@ class VersionPropertiesClass(_Aspect):
         self.aliases = list()
         self.comment = self.RECORD_SCHEMA.fields_dict["comment"].default
         self.sortId = str()
+        self.versioningScheme = self.RECORD_SCHEMA.fields_dict["versioningScheme"].default
         self.sourceCreatedTimestamp = self.RECORD_SCHEMA.fields_dict["sourceCreatedTimestamp"].default
         self.metadataCreatedTimestamp = self.RECORD_SCHEMA.fields_dict["metadataCreatedTimestamp"].default
         self.isLatest = self.RECORD_SCHEMA.fields_dict["isLatest"].default
@@ -6353,6 +6360,17 @@ class VersionPropertiesClass(_Aspect):
         self._inner_dict['sortId'] = value
     
     
+    @property
+    def versioningScheme(self) -> Union[str, "VersioningSchemeClass"]:
+        """What versioning scheme `sortId` belongs to.
+    Defaults to a plain string that is lexicographically sorted."""
+        return self._inner_dict.get('versioningScheme')  # type: ignore
+    
+    @versioningScheme.setter
+    def versioningScheme(self, value: Union[str, "VersioningSchemeClass"]) -> None:
+        self._inner_dict['versioningScheme'] = value
+    
+    
     @property
     def sourceCreatedTimestamp(self) -> Union[None, "AuditStampClass"]:
         """Timestamp reflecting when this asset version was created in the source system."""
@@ -9810,7 +9828,7 @@ class DataProductKeyClass(_Aspect):
 
 
     ASPECT_NAME = 'dataProductKey'
-    ASPECT_INFO = {'keyForEntity': 'dataProduct', 'entityCategory': 'core', 'entityAspects': ['ownership', 'glossaryTerms', 'globalTags', 'domains', 'dataProductProperties', 'institutionalMemory', 'status', 'structuredProperties', 'forms', 'testResults']}
+    ASPECT_INFO = {'keyForEntity': 'dataProduct', 'entityCategory': 'core', 'entityAspects': ['ownership', 'glossaryTerms', 'globalTags', 'domains', 'dataProductProperties', 'institutionalMemory', 'status', 'structuredProperties', 'forms', 'testResults', 'subTypes']}
     RECORD_SCHEMA = get_schema_type("com.linkedin.pegasus2avro.dataproduct.DataProductKey")
 
     def __init__(self,
@@ -14670,7 +14688,7 @@ class CorpGroupKeyClass(_Aspect):
 
 
     ASPECT_NAME = 'corpGroupKey'
-    ASPECT_INFO = {'keyForEntity': 'corpGroup', 'entityCategory': '_unset_', 'entityAspects': ['corpGroupInfo', 'corpGroupEditableInfo', 'globalTags', 'ownership', 'status', 'origin', 'roleMembership', 'structuredProperties', 'forms', 'testResults'], 'entityDoc': 'CorpGroup represents an identity of a group of users in the enterprise.'}
+    ASPECT_INFO = {'keyForEntity': 'corpGroup', 'entityCategory': '_unset_', 'entityAspects': ['corpGroupInfo', 'corpGroupEditableInfo', 'globalTags', 'ownership', 'status', 'origin', 'roleMembership', 'structuredProperties', 'forms', 'testResults', 'subTypes'], 'entityDoc': 'CorpGroup represents an identity of a group of users in the enterprise.'}
     RECORD_SCHEMA = get_schema_type("com.linkedin.pegasus2avro.metadata.key.CorpGroupKey")
 
     def __init__(self,
@@ -14699,7 +14717,7 @@ class CorpUserKeyClass(_Aspect):
 
 
     ASPECT_NAME = 'corpUserKey'
-    ASPECT_INFO = {'keyForEntity': 'corpuser', 'entityCategory': '_unset_', 'entityAspects': ['corpUserInfo', 'corpUserEditableInfo', 'corpUserStatus', 'groupMembership', 'globalTags', 'status', 'corpUserCredentials', 'nativeGroupMembership', 'corpUserSettings', 'origin', 'roleMembership', 'structuredProperties', 'forms', 'testResults'], 'entityDoc': 'CorpUser represents an identity of a person (or an account) in the enterprise.'}
+    ASPECT_INFO = {'keyForEntity': 'corpuser', 'entityCategory': '_unset_', 'entityAspects': ['corpUserInfo', 'corpUserEditableInfo', 'corpUserStatus', 'groupMembership', 'globalTags', 'status', 'corpUserCredentials', 'nativeGroupMembership', 'corpUserSettings', 'origin', 'roleMembership', 'structuredProperties', 'forms', 'testResults', 'subTypes'], 'entityDoc': 'CorpUser represents an identity of a person (or an account) in the enterprise.'}
     RECORD_SCHEMA = get_schema_type("com.linkedin.pegasus2avro.metadata.key.CorpUserKey")
 
     def __init__(self,
@@ -15357,7 +15375,7 @@ class DataProcessKeyClass(_Aspect):
 
 
     ASPECT_NAME = 'dataProcessKey'
-    ASPECT_INFO = {'keyForEntity': 'dataProcess', 'entityCategory': '_unset_', 'entityAspects': ['dataProcessInfo', 'ownership', 'status', 'testResults']}
+    ASPECT_INFO = {'keyForEntity': 'dataProcess', 'entityCategory': '_unset_', 'entityAspects': ['dataProcessInfo', 'ownership', 'status', 'testResults', 'subTypes']}
     RECORD_SCHEMA = get_schema_type("com.linkedin.pegasus2avro.metadata.key.DataProcessKey")
 
     def __init__(self,
@@ -15613,7 +15631,7 @@ class GlossaryNodeKeyClass(_Aspect):
 
 
     ASPECT_NAME = 'glossaryNodeKey'
-    ASPECT_INFO = {'keyForEntity': 'glossaryNode', 'entityCategory': 'core', 'entityAspects': ['glossaryNodeInfo', 'institutionalMemory', 'ownership', 'status', 'structuredProperties', 'forms', 'testResults']}
+    ASPECT_INFO = {'keyForEntity': 'glossaryNode', 'entityCategory': 'core', 'entityAspects': ['glossaryNodeInfo', 'institutionalMemory', 'ownership', 'status', 'structuredProperties', 'forms', 'testResults', 'subTypes']}
     RECORD_SCHEMA = get_schema_type("com.linkedin.pegasus2avro.metadata.key.GlossaryNodeKey")
 
     def __init__(self,
@@ -15642,7 +15660,7 @@ class GlossaryTermKeyClass(_Aspect):
 
 
     ASPECT_NAME = 'glossaryTermKey'
-    ASPECT_INFO = {'keyForEntity': 'glossaryTerm', 'entityCategory': 'core', 'entityAspects': ['glossaryTermInfo', 'glossaryRelatedTerms', 'institutionalMemory', 'schemaMetadata', 'ownership', 'deprecation', 'domains', 'status', 'browsePaths', 'structuredProperties', 'forms', 'testResults']}
+    ASPECT_INFO = {'keyForEntity': 'glossaryTerm', 'entityCategory': 'core', 'entityAspects': ['glossaryTermInfo', 'glossaryRelatedTerms', 'institutionalMemory', 'schemaMetadata', 'ownership', 'deprecation', 'domains', 'status', 'browsePaths', 'structuredProperties', 'forms', 'testResults', 'subTypes']}
     RECORD_SCHEMA = get_schema_type("com.linkedin.pegasus2avro.metadata.key.GlossaryTermKey")
 
     def __init__(self,
@@ -15729,7 +15747,7 @@ class MLFeatureKeyClass(_Aspect):
 
 
     ASPECT_NAME = 'mlFeatureKey'
-    ASPECT_INFO = {'keyForEntity': 'mlFeature', 'entityCategory': 'core', 'entityAspects': ['glossaryTerms', 'editableMlFeatureProperties', 'domains', 'mlFeatureProperties', 'ownership', 'institutionalMemory', 'status', 'deprecation', 'browsePaths', 'globalTags', 'dataPlatformInstance', 'browsePathsV2', 'structuredProperties', 'forms', 'testResults']}
+    ASPECT_INFO = {'keyForEntity': 'mlFeature', 'entityCategory': 'core', 'entityAspects': ['glossaryTerms', 'editableMlFeatureProperties', 'domains', 'mlFeatureProperties', 'ownership', 'institutionalMemory', 'status', 'deprecation', 'browsePaths', 'globalTags', 'dataPlatformInstance', 'browsePathsV2', 'structuredProperties', 'forms', 'testResults', 'subTypes']}
     RECORD_SCHEMA = get_schema_type("com.linkedin.pegasus2avro.metadata.key.MLFeatureKey")
 
     def __init__(self,
@@ -15771,7 +15789,7 @@ class MLFeatureTableKeyClass(_Aspect):
 
 
     ASPECT_NAME = 'mlFeatureTableKey'
-    ASPECT_INFO = {'keyForEntity': 'mlFeatureTable', 'entityCategory': 'core', 'entityAspects': ['glossaryTerms', 'editableMlFeatureTableProperties', 'domains', 'mlFeatureTableProperties', 'ownership', 'institutionalMemory', 'status', 'deprecation', 'browsePaths', 'globalTags', 'dataPlatformInstance', 'browsePathsV2', 'structuredProperties', 'forms', 'testResults']}
+    ASPECT_INFO = {'keyForEntity': 'mlFeatureTable', 'entityCategory': 'core', 'entityAspects': ['glossaryTerms', 'editableMlFeatureTableProperties', 'domains', 'mlFeatureTableProperties', 'ownership', 'institutionalMemory', 'status', 'deprecation', 'browsePaths', 'globalTags', 'dataPlatformInstance', 'browsePathsV2', 'structuredProperties', 'forms', 'testResults', 'subTypes']}
     RECORD_SCHEMA = get_schema_type("com.linkedin.pegasus2avro.metadata.key.MLFeatureTableKey")
 
     def __init__(self,
@@ -15868,7 +15886,7 @@ class MLModelGroupKeyClass(_Aspect):
 
 
     ASPECT_NAME = 'mlModelGroupKey'
-    ASPECT_INFO = {'keyForEntity': 'mlModelGroup', 'entityCategory': 'core', 'entityAspects': ['glossaryTerms', 'editableMlModelGroupProperties', 'domains', 'mlModelGroupProperties', 'ownership', 'status', 'deprecation', 'browsePaths', 'globalTags', 'dataPlatformInstance', 'browsePathsV2', 'structuredProperties', 'forms', 'testResults']}
+    ASPECT_INFO = {'keyForEntity': 'mlModelGroup', 'entityCategory': 'core', 'entityAspects': ['glossaryTerms', 'editableMlModelGroupProperties', 'domains', 'mlModelGroupProperties', 'ownership', 'status', 'deprecation', 'browsePaths', 'globalTags', 'dataPlatformInstance', 'browsePathsV2', 'structuredProperties', 'forms', 'testResults', 'subTypes']}
     RECORD_SCHEMA = get_schema_type("com.linkedin.pegasus2avro.metadata.key.MLModelGroupKey")
 
     def __init__(self,
@@ -15923,7 +15941,7 @@ class MLModelKeyClass(_Aspect):
 
 
     ASPECT_NAME = 'mlModelKey'
-    ASPECT_INFO = {'keyForEntity': 'mlModel', 'entityCategory': 'core', 'entityAspects': ['glossaryTerms', 'editableMlModelProperties', 'domains', 'ownership', 'mlModelProperties', 'intendedUse', 'mlModelFactorPrompts', 'mlModelMetrics', 'mlModelEvaluationData', 'mlModelTrainingData', 'mlModelQuantitativeAnalyses', 'mlModelEthicalConsiderations', 'mlModelCaveatsAndRecommendations', 'institutionalMemory', 'sourceCode', 'status', 'cost', 'deprecation', 'browsePaths', 'globalTags', 'dataPlatformInstance', 'browsePathsV2', 'structuredProperties', 'forms', 'testResults', 'versionProperties']}
+    ASPECT_INFO = {'keyForEntity': 'mlModel', 'entityCategory': 'core', 'entityAspects': ['glossaryTerms', 'editableMlModelProperties', 'domains', 'ownership', 'mlModelProperties', 'intendedUse', 'mlModelFactorPrompts', 'mlModelMetrics', 'mlModelEvaluationData', 'mlModelTrainingData', 'mlModelQuantitativeAnalyses', 'mlModelEthicalConsiderations', 'mlModelCaveatsAndRecommendations', 'institutionalMemory', 'sourceCode', 'status', 'cost', 'deprecation', 'browsePaths', 'globalTags', 'dataPlatformInstance', 'browsePathsV2', 'structuredProperties', 'forms', 'testResults', 'versionProperties', 'subTypes']}
     RECORD_SCHEMA = get_schema_type("com.linkedin.pegasus2avro.metadata.key.MLModelKey")
 
     def __init__(self,
@@ -15978,7 +15996,7 @@ class MLPrimaryKeyKeyClass(_Aspect):
 
 
     ASPECT_NAME = 'mlPrimaryKeyKey'
-    ASPECT_INFO = {'keyForEntity': 'mlPrimaryKey', 'entityCategory': 'core', 'entityAspects': ['glossaryTerms', 'editableMlPrimaryKeyProperties', 'domains', 'mlPrimaryKeyProperties', 'ownership', 'institutionalMemory', 'status', 'deprecation', 'globalTags', 'dataPlatformInstance', 'structuredProperties', 'forms', 'testResults']}
+    ASPECT_INFO = {'keyForEntity': 'mlPrimaryKey', 'entityCategory': 'core', 'entityAspects': ['glossaryTerms', 'editableMlPrimaryKeyProperties', 'domains', 'mlPrimaryKeyProperties', 'ownership', 'institutionalMemory', 'status', 'deprecation', 'globalTags', 'dataPlatformInstance', 'structuredProperties', 'forms', 'testResults', 'subTypes']}
     RECORD_SCHEMA = get_schema_type("com.linkedin.pegasus2avro.metadata.key.MLPrimaryKeyKey")
 
     def __init__(self,
@@ -16092,7 +16110,7 @@ class PostKeyClass(_Aspect):
 
 
     ASPECT_NAME = 'postKey'
-    ASPECT_INFO = {'keyForEntity': 'post', 'entityCategory': 'core', 'entityAspects': ['postInfo']}
+    ASPECT_INFO = {'keyForEntity': 'post', 'entityCategory': 'core', 'entityAspects': ['postInfo', 'subTypes']}
     RECORD_SCHEMA = get_schema_type("com.linkedin.pegasus2avro.metadata.key.PostKey")
 
     def __init__(self,
@@ -16179,7 +16197,7 @@ class SchemaFieldKeyClass(_Aspect):
 
 
     ASPECT_NAME = 'schemaFieldKey'
-    ASPECT_INFO = {'keyForEntity': 'schemaField', 'entityCategory': 'core', 'entityAspects': ['schemafieldInfo', 'structuredProperties', 'forms', 'businessAttributes', 'status', 'schemaFieldAliases', 'documentation', 'testResults', 'deprecation']}
+    ASPECT_INFO = {'keyForEntity': 'schemaField', 'entityCategory': 'core', 'entityAspects': ['schemafieldInfo', 'structuredProperties', 'forms', 'businessAttributes', 'status', 'schemaFieldAliases', 'documentation', 'testResults', 'deprecation', 'subTypes']}
     RECORD_SCHEMA = get_schema_type("com.linkedin.pegasus2avro.metadata.key.SchemaFieldKey")
 
     def __init__(self,
@@ -25170,7 +25188,7 @@ class VersionSetPropertiesClass(_Aspect):
     def _restore_defaults(self) -> None:
         self.customProperties = dict()
         self.latest = str()
-        self.versioningScheme = VersioningSchemeClass.ALPHANUMERIC_GENERATED_BY_DATAHUB
+        self.versioningScheme = VersioningSchemeClass.LEXICOGRAPHIC_STRING
     
     
     @property
@@ -25206,7 +25224,12 @@ class VersionSetPropertiesClass(_Aspect):
 class VersioningSchemeClass(object):
     # No docs available.
     
+    LEXICOGRAPHIC_STRING = "LEXICOGRAPHIC_STRING"
+    """String sorted lexicographically."""
+    
     ALPHANUMERIC_GENERATED_BY_DATAHUB = "ALPHANUMERIC_GENERATED_BY_DATAHUB"
+    """String managed by DataHub. Currently, an 8 character alphabetical string."""
+    
     
     
 class DataHubViewDefinitionClass(DictWrapper):
@@ -26540,6 +26563,7 @@ ASPECT_NAME_MAP: Dict[str, Type[_Aspect]] = {
     for aspect in ASPECT_CLASSES
 }
 
+from typing import Literal
 from typing_extensions import TypedDict
 
 class AspectBag(TypedDict, total=False):
@@ -26816,4 +26840,123 @@ KEY_ASPECTS: Dict[str, Type[_Aspect]] = {
     'platformResource': PlatformResourceKeyClass
 }
 
+ENTITY_TYPE_NAMES: List[str] = [
+    'businessAttribute',
+    'dataType',
+    'dataProduct',
+    'telemetry',
+    'dataHubAccessToken',
+    'dataHubConnection',
+    'dataHubRole',
+    'mlModel',
+    'notebook',
+    'role',
+    'globalSettings',
+    'dataset',
+    'chart',
+    'glossaryNode',
+    'assertion',
+    'dataHubExecutionRequest',
+    'mlModelGroup',
+    'mlModelDeployment',
+    'dataFlow',
+    'dataJob',
+    'corpGroup',
+    'dataPlatform',
+    'corpuser',
+    'dashboard',
+    'mlPrimaryKey',
+    'query',
+    'ownershipType',
+    'domain',
+    'mlFeature',
+    'mlFeatureTable',
+    'dataHubView',
+    'dataHubAction',
+    'dataHubUpgrade',
+    'dataHubStepState',
+    'erModelRelationship',
+    'dataHubPolicy',
+    'dataHubRetention',
+    'dataContract',
+    'dataProcess',
+    'dataProcessInstance',
+    'glossaryTerm',
+    'form',
+    'schemaField',
+    'tag',
+    'post',
+    'dataHubIngestionSource',
+    'inviteToken',
+    'dataHubPersona',
+    'container',
+    'incident',
+    'dataPlatformInstance',
+    'test',
+    'versionSet',
+    'dataHubSecret',
+    'entityType',
+    'structuredProperty',
+    'platformResource'
+]
+EntityTypeName = Literal[
+    'businessAttribute',
+    'dataType',
+    'dataProduct',
+    'telemetry',
+    'dataHubAccessToken',
+    'dataHubConnection',
+    'dataHubRole',
+    'mlModel',
+    'notebook',
+    'role',
+    'globalSettings',
+    'dataset',
+    'chart',
+    'glossaryNode',
+    'assertion',
+    'dataHubExecutionRequest',
+    'mlModelGroup',
+    'mlModelDeployment',
+    'dataFlow',
+    'dataJob',
+    'corpGroup',
+    'dataPlatform',
+    'corpuser',
+    'dashboard',
+    'mlPrimaryKey',
+    'query',
+    'ownershipType',
+    'domain',
+    'mlFeature',
+    'mlFeatureTable',
+    'dataHubView',
+    'dataHubAction',
+    'dataHubUpgrade',
+    'dataHubStepState',
+    'erModelRelationship',
+    'dataHubPolicy',
+    'dataHubRetention',
+    'dataContract',
+    'dataProcess',
+    'dataProcessInstance',
+    'glossaryTerm',
+    'form',
+    'schemaField',
+    'tag',
+    'post',
+    'dataHubIngestionSource',
+    'inviteToken',
+    'dataHubPersona',
+    'container',
+    'incident',
+    'dataPlatformInstance',
+    'test',
+    'versionSet',
+    'dataHubSecret',
+    'entityType',
+    'structuredProperty',
+    'platformResource'
+]
+
 # fmt: on