2024-06-20 16:54:12 +02:00
|
|
|
"""Models for the TableDiff test case"""
|
2024-07-02 12:36:03 +02:00
|
|
|
|
2024-06-20 16:54:12 +02:00
|
|
|
from typing import List, Optional
|
|
|
|
|
|
|
|
from pydantic import BaseModel
|
|
|
|
|
2024-11-11 10:07:23 +01:00
|
|
|
from metadata.generated.schema.entity.data.table import Column, TableProfilerConfig
|
|
|
|
from metadata.generated.schema.entity.services.databaseService import (
|
|
|
|
DatabaseServiceType,
|
|
|
|
)
|
2024-07-02 12:36:03 +02:00
|
|
|
|
|
|
|
|
|
|
|
class TableParameter(BaseModel):
|
|
|
|
serviceUrl: str
|
|
|
|
path: str
|
|
|
|
columns: List[Column]
|
2024-11-11 10:07:23 +01:00
|
|
|
database_service_type: DatabaseServiceType
|
2024-07-02 12:36:03 +02:00
|
|
|
|
2024-06-20 16:54:12 +02:00
|
|
|
|
|
|
|
class TableDiffRuntimeParameters(BaseModel):
|
2024-07-02 12:36:03 +02:00
|
|
|
table1: TableParameter
|
|
|
|
table2: TableParameter
|
2024-06-20 16:54:12 +02:00
|
|
|
keyColumns: List[str]
|
|
|
|
extraColumns: List[str]
|
|
|
|
whereClause: Optional[str]
|
2024-11-11 10:07:23 +01:00
|
|
|
table_profile_config: Optional[TableProfilerConfig]
|