diff --git a/dolphinscheduler-python/pydolphinscheduler/src/pydolphinscheduler/tasks/sql.py b/dolphinscheduler-python/pydolphinscheduler/src/pydolphinscheduler/tasks/sql.py index c15170c448f0f70b2557318d21368e3d860b2a0e..a1259824f14e6b6220282b7a4776ebb65ba19fcd 100644 --- a/dolphinscheduler-python/pydolphinscheduler/src/pydolphinscheduler/tasks/sql.py +++ b/dolphinscheduler-python/pydolphinscheduler/src/pydolphinscheduler/tasks/sql.py @@ -31,8 +31,8 @@ log = logging.getLogger(__file__) class SqlType: """SQL type, for now it just contain `SELECT` and `NO_SELECT`.""" - SELECT = 0 - NOT_SELECT = 1 + SELECT = "0" + NOT_SELECT = "1" class Sql(Task): @@ -64,7 +64,7 @@ class Sql(Task): name: str, datasource_name: str, sql: str, - sql_type: Optional[int] = None, + sql_type: Optional[str] = None, pre_statements: Optional[str] = None, post_statements: Optional[str] = None, display_rows: Optional[int] = 10, @@ -80,7 +80,7 @@ class Sql(Task): self.display_rows = display_rows @property - def sql_type(self) -> int: + def sql_type(self) -> str: """Judgement sql type, it will return the SQL type for type `SELECT` or `NOT_SELECT`. If `param_sql_type` dot not specific, will use regexp to check diff --git a/dolphinscheduler-python/pydolphinscheduler/tests/tasks/test_sql.py b/dolphinscheduler-python/pydolphinscheduler/tests/tasks/test_sql.py index 74ac0f67eacd14b54f7e0a33aee8b97118b8acbe..ee0acc442e11ade5b70f2749643f8a2de94358c2 100644 --- a/dolphinscheduler-python/pydolphinscheduler/tests/tasks/test_sql.py +++ b/dolphinscheduler-python/pydolphinscheduler/tests/tasks/test_sql.py @@ -89,7 +89,7 @@ def test_get_sql_type( "sql": "select 1", "type": "MYSQL", "datasource": 1, - "sqlType": SqlType.SELECT, + "sqlType": "0", "preStatements": [], "postStatements": [], "displayRows": 10, @@ -138,7 +138,7 @@ def test_sql_get_define(mock_datasource): "type": "MYSQL", "datasource": 1, "sql": command, - "sqlType": SqlType.SELECT, + "sqlType": "0", "displayRows": 10, "preStatements": [], "postStatements": [],