openmetadata-ingestion 1.4.6.0__py3-none-any.whl → 1.5.0.0rc1__py3-none-any.whl

metadata/generated/schema/type/entityUsage.py

@@ -1,23 +1,28 @@
 # generated by datamodel-codegen:
 #   filename:  type/entityUsage.json
-#   timestamp: 2024-08-02T13:31:44+00:00
+#   timestamp: 2024-08-05T15:12:43+00:00
 
 from __future__ import annotations
 
 from typing import List
 
-from pydantic import BaseModel, Extra, Field
+from pydantic import ConfigDict, Field
+from typing_extensions import Annotated
+
+from metadata.ingestion.models.custom_pydantic import BaseModel
 
 from . import entityReference, usageDetails
 
 
 class EntityUsage(BaseModel):
-    class Config:
-        extra = Extra.forbid
-
-    entity: entityReference.EntityReference = Field(
-        ..., description='Entity for which usage is returned.'
-    )
-    usage: List[usageDetails.UsageDetails] = Field(
-        ..., description='List usage details per day.'
+    model_config = ConfigDict(
+        extra='forbid',
     )
+    entity: Annotated[
+        entityReference.EntityReference,
+        Field(description='Entity for which usage is returned.'),
+    ]
+    usage: Annotated[
+        List[usageDetails.UsageDetails],
+        Field(description='List usage details per day.'),
+    ]

metadata/generated/schema/type/filterPattern.py

@@ -1,12 +1,15 @@
 # generated by datamodel-codegen:
 #   filename:  type/filterPattern.json
-#   timestamp: 2024-08-02T13:31:44+00:00
+#   timestamp: 2024-08-05T15:12:43+00:00
 
 from __future__ import annotations
 
 from typing import List, Optional
 
-from pydantic import BaseModel, Extra, Field
+from pydantic import ConfigDict, Field
+from typing_extensions import Annotated
+
+from metadata.ingestion.models.custom_pydantic import BaseModel
 
 
 class FilterPatternModel(BaseModel):
@@ -14,14 +17,20 @@ class FilterPatternModel(BaseModel):
 
 
 class FilterPattern(BaseModel):
-    class Config:
-        extra = Extra.forbid
-
-    includes: Optional[List[str]] = Field(
-        [],
-        description='List of strings/regex patterns to match and include only database entities that match.',
-    )
-    excludes: Optional[List[str]] = Field(
-        [],
-        description='List of strings/regex patterns to match and exclude only database entities that match.',
+    model_config = ConfigDict(
+        extra='forbid',
     )
+    includes: Annotated[
+        Optional[List[str]],
+        Field(
+            [],
+            description='List of strings/regex patterns to match and include only database entities that match.',
+        ),
+    ]
+    excludes: Annotated[
+        Optional[List[str]],
+        Field(
+            [],
+            description='List of strings/regex patterns to match and exclude only database entities that match.',
+        ),
+    ]

metadata/generated/schema/type/function.py

@@ -1,13 +1,16 @@
 # generated by datamodel-codegen:
 #   filename:  type/function.json
-#   timestamp: 2024-08-02T13:31:44+00:00
+#   timestamp: 2024-08-05T15:12:43+00:00
 
 from __future__ import annotations
 
 from enum import Enum
 from typing import Any, List, Optional
 
-from pydantic import BaseModel, Extra, Field
+from pydantic import ConfigDict, Field
+from typing_extensions import Annotated
+
+from metadata.ingestion.models.custom_pydantic import BaseModel
 
 
 class ParameterType(Enum):
@@ -19,27 +22,33 @@ class ParameterType(Enum):
 
 
 class ParamAdditionalContext(BaseModel):
-    class Config:
-        extra = Extra.forbid
-
-    data: Optional[Any] = Field(None, description='List of Entities')
+    model_config = ConfigDict(
+        extra='forbid',
+    )
+    data: Annotated[Optional[Any], Field(None, description='List of Entities')]
 
 
 class Function(BaseModel):
-    class Config:
-        extra = Extra.forbid
-
-    name: Optional[str] = Field(None, description='Name of the function.')
-    input: Optional[str] = Field(
-        None, description='Description of input taken by the function.'
-    )
-    description: Optional[str] = Field(
-        None, description='Description for the function.'
-    )
-    examples: Optional[List] = Field(
-        None, description='Examples of the function to help users author conditions.'
-    )
-    parameterInputType: Optional[ParameterType] = Field(
-        None, description='List of receivers to send mail to'
+    model_config = ConfigDict(
+        extra='forbid',
     )
+    name: Annotated[Optional[str], Field(None, description='Name of the function.')]
+    input: Annotated[
+        Optional[str],
+        Field(None, description='Description of input taken by the function.'),
+    ]
+    description: Annotated[
+        Optional[str], Field(None, description='Description for the function.')
+    ]
+    examples: Annotated[
+        Optional[List],
+        Field(
+            None,
+            description='Examples of the function to help users author conditions.',
+        ),
+    ]
+    parameterInputType: Annotated[
+        Optional[ParameterType],
+        Field(None, description='List of receivers to send mail to'),
+    ]
     paramAdditionalContext: Optional[ParamAdditionalContext] = None

metadata/generated/schema/type/include.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/include.json
-#   timestamp: 2024-08-02T13:31:44+00:00
+#   timestamp: 2024-08-05T15:12:43+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/jdbcConnection.py

@@ -1,36 +1,41 @@
 # generated by datamodel-codegen:
 #   filename:  type/jdbcConnection.json
-#   timestamp: 2024-08-02T13:31:44+00:00
+#   timestamp: 2024-08-05T15:12:43+00:00
 
 from __future__ import annotations
 
-from pydantic import BaseModel, Extra, Field
+from pydantic import ConfigDict, Field, RootModel
+from typing_extensions import Annotated
 
+from metadata.ingestion.models.custom_pydantic import BaseModel
 
-class DriverClass(BaseModel):
-    __root__: str = Field(..., description='Type used for JDBC driver class.')
 
+class DriverClass(RootModel[str]):
+    root: Annotated[str, Field(description='Type used for JDBC driver class.')]
 
-class ConnectionUrl(BaseModel):
-    __root__: str = Field(
-        ...,
-        description='Type used for JDBC connection URL of format `url_scheme://<username>:<password>@<host>:<port>/<db_name>`.',
-    )
 
+class ConnectionUrl(RootModel[str]):
+    root: Annotated[
+        str,
+        Field(
+            description='Type used for JDBC connection URL of format `url_scheme://<username>:<password>@<host>:<port>/<db_name>`.'
+        ),
+    ]
 
-class JdbcInfo(BaseModel):
-    class Config:
-        extra = Extra.forbid
 
+class JdbcInfo(BaseModel):
+    model_config = ConfigDict(
+        extra='forbid',
+    )
     driverClass: DriverClass
     connectionUrl: ConnectionUrl
 
 
 class JdbcConnection(BaseModel):
-    class Config:
-        extra = Extra.forbid
-
-    driverClass: DriverClass = Field(..., description='JDBC driver class.')
-    connectionUrl: ConnectionUrl = Field(..., description='JDBC connection URL.')
-    userName: str = Field(..., description='Login user name.')
-    password: str = Field(..., description='Login password.')
+    model_config = ConfigDict(
+        extra='forbid',
+    )
+    driverClass: Annotated[DriverClass, Field(description='JDBC driver class.')]
+    connectionUrl: Annotated[ConnectionUrl, Field(description='JDBC connection URL.')]
+    userName: Annotated[str, Field(description='Login user name.')]
+    password: Annotated[str, Field(description='Login password.')]

metadata/generated/schema/type/lifeCycle.py

@@ -1,43 +1,64 @@
 # generated by datamodel-codegen:
 #   filename:  type/lifeCycle.json
-#   timestamp: 2024-08-02T13:31:44+00:00
+#   timestamp: 2024-08-05T15:12:43+00:00
 
 from __future__ import annotations
 
 from typing import Optional
 
-from pydantic import BaseModel, Extra, Field
+from pydantic import ConfigDict, Field
+from typing_extensions import Annotated
+
+from metadata.ingestion.models.custom_pydantic import BaseModel
 
 from . import basic, entityReference
 
 
 class AccessDetails(BaseModel):
-    class Config:
-        extra = Extra.forbid
-
-    timestamp: basic.Timestamp = Field(
-        ..., description='Timestamp of data asset accessed for creation, update, read.'
-    )
-    accessedBy: Optional[entityReference.EntityReference] = Field(
-        None,
-        description='User, Pipeline, Query that created,updated or accessed the data asset',
-    )
-    accessedByAProcess: Optional[str] = Field(
-        None,
-        description='Any process that accessed the data asset that is not captured in OpenMetadata.',
+    model_config = ConfigDict(
+        extra='forbid',
     )
+    timestamp: Annotated[
+        basic.Timestamp,
+        Field(
+            description='Timestamp of data asset accessed for creation, update, read.'
+        ),
+    ]
+    accessedBy: Annotated[
+        Optional[entityReference.EntityReference],
+        Field(
+            None,
+            description='User, Pipeline, Query that created,updated or accessed the data asset',
+        ),
+    ]
+    accessedByAProcess: Annotated[
+        Optional[str],
+        Field(
+            None,
+            description='Any process that accessed the data asset that is not captured in OpenMetadata.',
+        ),
+    ]
 
 
 class LifeCycle(BaseModel):
-    class Config:
-        extra = Extra.forbid
-
-    created: Optional[AccessDetails] = Field(
-        None, description='Access Details about created aspect of the data asset'
-    )
-    updated: Optional[AccessDetails] = Field(
-        None, description='Access Details about updated aspect of the data asset'
-    )
-    accessed: Optional[AccessDetails] = Field(
-        None, description='Access Details about accessed aspect of the data asset'
+    model_config = ConfigDict(
+        extra='forbid',
     )
+    created: Annotated[
+        Optional[AccessDetails],
+        Field(
+            None, description='Access Details about created aspect of the data asset'
+        ),
+    ]
+    updated: Annotated[
+        Optional[AccessDetails],
+        Field(
+            None, description='Access Details about updated aspect of the data asset'
+        ),
+    ]
+    accessed: Annotated[
+        Optional[AccessDetails],
+        Field(
+            None, description='Access Details about accessed aspect of the data asset'
+        ),
+    ]

metadata/generated/schema/type/paging.py

@@ -1,32 +1,43 @@
 # generated by datamodel-codegen:
 #   filename:  type/paging.json
-#   timestamp: 2024-08-02T13:31:44+00:00
+#   timestamp: 2024-08-05T15:12:43+00:00
 
 from __future__ import annotations
 
 from typing import Optional
 
-from pydantic import BaseModel, Extra, Field
+from pydantic import ConfigDict, Field
+from typing_extensions import Annotated
 
+from metadata.ingestion.models.custom_pydantic import BaseModel
 
-class Paging(BaseModel):
-    class Config:
-        extra = Extra.forbid
 
-    before: Optional[str] = Field(
-        None,
-        description='Before cursor used for getting the previous page (see API pagination for details).',
-    )
-    after: Optional[str] = Field(
-        None,
-        description='After cursor used for getting the next page (see API pagination for details).',
-    )
-    offset: Optional[int] = Field(
-        None, description='Offset used in case of offset based pagination.'
-    )
-    limit: Optional[int] = Field(
-        None, description='Limit used in case of offset based pagination.'
-    )
-    total: int = Field(
-        ..., description='Total number of entries available to page through.'
+class Paging(BaseModel):
+    model_config = ConfigDict(
+        extra='forbid',
     )
+    before: Annotated[
+        Optional[str],
+        Field(
+            None,
+            description='Before cursor used for getting the previous page (see API pagination for details).',
+        ),
+    ]
+    after: Annotated[
+        Optional[str],
+        Field(
+            None,
+            description='After cursor used for getting the next page (see API pagination for details).',
+        ),
+    ]
+    offset: Annotated[
+        Optional[int],
+        Field(None, description='Offset used in case of offset based pagination.'),
+    ]
+    limit: Annotated[
+        Optional[int],
+        Field(None, description='Limit used in case of offset based pagination.'),
+    ]
+    total: Annotated[
+        int, Field(description='Total number of entries available to page through.')
+    ]

metadata/generated/schema/type/profile.py

@@ -1,20 +1,22 @@
 # generated by datamodel-codegen:
 #   filename:  type/profile.json
-#   timestamp: 2024-08-02T13:31:44+00:00
+#   timestamp: 2024-08-05T15:12:43+00:00
 
 from __future__ import annotations
 
 from typing import Optional
 
-from pydantic import AnyUrl, BaseModel, Extra
+from pydantic import AnyUrl, ConfigDict
+
+from metadata.ingestion.models.custom_pydantic import BaseModel
 
 from ..entity.events import webhook
 
 
 class ImageList(BaseModel):
-    class Config:
-        extra = Extra.forbid
-
+    model_config = ConfigDict(
+        extra='forbid',
+    )
     image: Optional[AnyUrl] = None
     image24: Optional[AnyUrl] = None
     image32: Optional[AnyUrl] = None
@@ -25,9 +27,9 @@ class ImageList(BaseModel):
 
 
 class MessagingProvider(BaseModel):
-    class Config:
-        extra = Extra.forbid
-
+    model_config = ConfigDict(
+        extra='forbid',
+    )
     slack: Optional[webhook.Webhook] = None
     msTeams: Optional[webhook.Webhook] = None
     gChat: Optional[webhook.Webhook] = None
@@ -35,8 +37,8 @@ class MessagingProvider(BaseModel):
 
 
 class Profile(BaseModel):
-    class Config:
-        extra = Extra.forbid
-
+    model_config = ConfigDict(
+        extra='forbid',
+    )
     images: Optional[ImageList] = None
     subscription: Optional[MessagingProvider] = None

metadata/generated/schema/type/queryParserData.py

@@ -1,46 +1,61 @@
 # generated by datamodel-codegen:
 #   filename:  type/queryParserData.json
-#   timestamp: 2024-08-02T13:31:44+00:00
+#   timestamp: 2024-08-05T15:12:43+00:00
 
 from __future__ import annotations
 
 from typing import Any, Dict, List, Optional
 
-from pydantic import BaseModel, Extra, Field
+from pydantic import ConfigDict, Field
+from typing_extensions import Annotated
+
+from metadata.ingestion.models.custom_pydantic import BaseModel
 
 
 class ParsedData(BaseModel):
-    tables: List[str] = Field(..., description='List of tables used in query')
-    databaseName: Optional[str] = Field(
-        None, description='Database associated with the table in the query'
-    )
-    joins: Optional[Dict[str, Any]] = Field(
-        None,
-        description='Maps each parsed table name of a query to the join information',
-    )
-    sql: str = Field(..., description='SQL query')
-    query_type: Optional[str] = Field(None, description='SQL query type')
-    exclude_usage: Optional[bool] = Field(
-        None,
-        description='Flag to check if query is to be excluded while processing usage',
-    )
-    serviceName: str = Field(
-        ..., description='Name that identifies this database service.'
-    )
-    userName: Optional[str] = Field(
-        None, description='Name of the user that executed the SQL query'
-    )
-    date: Optional[str] = Field(None, description='Date of execution of SQL query')
-    databaseSchema: Optional[str] = Field(
-        None, description='Database schema of the associated with query'
-    )
-    duration: Optional[float] = Field(
-        None, description='How long did the query took to run in milliseconds.'
-    )
+    tables: Annotated[List[str], Field(description='List of tables used in query')]
+    databaseName: Annotated[
+        Optional[str],
+        Field(None, description='Database associated with the table in the query'),
+    ]
+    joins: Annotated[
+        Optional[Dict[str, Any]],
+        Field(
+            None,
+            description='Maps each parsed table name of a query to the join information',
+        ),
+    ]
+    sql: Annotated[str, Field(description='SQL query')]
+    query_type: Annotated[Optional[str], Field(None, description='SQL query type')]
+    exclude_usage: Annotated[
+        Optional[bool],
+        Field(
+            None,
+            description='Flag to check if query is to be excluded while processing usage',
+        ),
+    ]
+    serviceName: Annotated[
+        str, Field(description='Name that identifies this database service.')
+    ]
+    userName: Annotated[
+        Optional[str],
+        Field(None, description='Name of the user that executed the SQL query'),
+    ]
+    date: Annotated[
+        Optional[str], Field(None, description='Date of execution of SQL query')
+    ]
+    databaseSchema: Annotated[
+        Optional[str],
+        Field(None, description='Database schema of the associated with query'),
+    ]
+    duration: Annotated[
+        Optional[float],
+        Field(None, description='How long did the query took to run in milliseconds.'),
+    ]
 
 
 class QueryParserData(BaseModel):
-    class Config:
-        extra = Extra.forbid
-
+    model_config = ConfigDict(
+        extra='forbid',
+    )
     parsedData: Optional[List[ParsedData]] = None

metadata/generated/schema/type/reaction.py

@@ -1,13 +1,16 @@
 # generated by datamodel-codegen:
 #   filename:  type/reaction.json
-#   timestamp: 2024-08-02T13:31:44+00:00
+#   timestamp: 2024-08-05T15:12:43+00:00
 
 from __future__ import annotations
 
 from enum import Enum
 from typing import List
 
-from pydantic import BaseModel, Extra, Field
+from pydantic import ConfigDict, Field, RootModel
+from typing_extensions import Annotated
+
+from metadata.ingestion.models.custom_pydantic import BaseModel
 
 from . import entityReference
 
@@ -24,12 +27,14 @@ class ReactionType(Enum):
 
 
 class Reaction(BaseModel):
-    class Config:
-        extra = Extra.forbid
-
+    model_config = ConfigDict(
+        extra='forbid',
+    )
     reactionType: ReactionType
-    user: entityReference.EntityReference = Field(..., description='User who reacted.')
+    user: Annotated[
+        entityReference.EntityReference, Field(description='User who reacted.')
+    ]
 
 
-class ReactionList(BaseModel):
-    __root__: List[Reaction]
+class ReactionList(RootModel[List[Reaction]]):
+    root: List[Reaction]

metadata/generated/schema/type/schedule.py

@@ -1,24 +1,31 @@
 # generated by datamodel-codegen:
 #   filename:  type/schedule.json
-#   timestamp: 2024-08-02T13:31:44+00:00
+#   timestamp: 2024-08-05T15:12:43+00:00
 
 from __future__ import annotations
 
 from typing import Optional
 
-from pydantic import BaseModel, Extra, Field
+from pydantic import ConfigDict, Field
+from typing_extensions import Annotated
+
+from metadata.ingestion.models.custom_pydantic import BaseModel
 
 from . import basic
 
 
 class Schedule(BaseModel):
-    class Config:
-        extra = Extra.forbid
-
-    startDate: Optional[basic.DateTime] = Field(
-        None, description='Start date and time of the schedule.'
-    )
-    repeatFrequency: Optional[basic.Duration] = Field(
-        None,
-        description="Repeat frequency in ISO 8601 duration format. Example - 'P23DT23H'.",
+    model_config = ConfigDict(
+        extra='forbid',
     )
+    startDate: Annotated[
+        Optional[basic.DateTime],
+        Field(None, description='Start date and time of the schedule.'),
+    ]
+    repeatFrequency: Annotated[
+        Optional[basic.Duration],
+        Field(
+            None,
+            description="Repeat frequency in ISO 8601 duration format. Example - 'P23DT23H'.",
+        ),
+    ]