Changes to support MSSQL

dev/dev.env

@@ -21,3 +21,6 @@ VERTICA_DB_NAME=vertica
 # leave VMART_DIR and VMART_ETL_SCRIPT empty.
 VMART_DIR=
 VMART_ETL_SCRIPT=
+
+ACCEPT_EULA=Y
+MSSQL_SA_PASSWORD=<password!CAP>

docker-compose.yml

@@ -117,13 +117,28 @@ services:
       networks:
       - local
 
-
+    mssql:
+      container_name: dd-mssql
+      image: mcr.microsoft.com/mssql/server:2022-latest
+      restart: always
+      volumes:
+        - mssql-data:/var/opt/mssql/data:delegated
+      ports:
+        - '8001:1433'
+      expose:
+        - '8001'
+      env_file:
+        - dev/dev.env
+      tty: true
+      networks:
+        - local
 
 volumes:
   postgresql-data:
   mysql-data:
   clickhouse-data:
   vertica-data:
+  mssql-data:
 
 networks:
   local:

docs/supported-databases.md

@@ -14,6 +14,7 @@
 | Clickhouse    |  💛    | `clickhouse://<username>:<password>@<hostname>:9000/<database>`                                                                     |
 | Vertica       |  💛    | `vertica://<username>:<password>@<hostname>:5433/<database>`                                                                        |
 | DuckDB        |  💛    |                                                                                                                                     |
+| MsSQL         |  ⏳    | `mssql+pymssql://<user>:<password>@<host>:<port>/<database>`                                                                        |
 | ElasticSearch |  📝    |                                                                                                                                     |
 | Planetscale   |  📝    |                                                                                                                                     |
 | Pinot         |  📝    |                                                                                                                                     |

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "sqeleton"
-version = "0.1.7"
+version = "0.1.8"
 description = "Python library for querying SQL databases"
 authors = ["Erez Shinan <[email protected]>"]
 license = "MIT"
@@ -23,7 +23,7 @@ packages = [{ include = "sqeleton" }]
 
 [tool.poetry.dependencies]
 python = "^3.8"
-runtype = ">=0.5.0"
+runtype = "0.5.0"
 dsnparse = "*"
 click = ">=8.1"
 rich = "*"
@@ -41,6 +41,7 @@ textual = {version=">=0.9.1", optional=true}
 textual-select = {version="*", optional=true}
 pygments = {version=">=2.13.0", optional=true}
 prompt-toolkit = {version=">=3.0.36", optional=true}
+pymssql = {version=">=2.3.2", optional=true}
 
 [tool.poetry.dev-dependencies]
 parameterized = "*"
@@ -56,6 +57,7 @@ trino = ">=0.314.0"
 presto-python-client = "*"
 clickhouse-driver = "*"
 vertica-python = "*"
+pymssql = "*"
 
 [tool.poetry.extras]
 mysql = ["mysql-connector-python"]
@@ -69,6 +71,7 @@ clickhouse = ["clickhouse-driver"]
 vertica = ["vertica-python"]
 duckdb = ["duckdb"]
 tui = ["textual", "textual-select", "pygments", "prompt-toolkit"]
+mssql = ["pymssql"]
 
 [build-system]
 requires = ["poetry-core>=1.0.0"]

sqeleton/databases/__init__.py

@@ -14,5 +14,6 @@
 from .clickhouse import Clickhouse
 from .vertica import Vertica
 from .duckdb import DuckDB
+from .mssql import MsSQL
 
 connect = Connect()

sqeleton/databases/_connect.py

@@ -22,6 +22,7 @@
 from .clickhouse import Clickhouse
 from .vertica import Vertica
 from .duckdb import DuckDB
+from .mssql import MsSQL
 
 
 @dataclass
@@ -87,6 +88,7 @@ def match_path(self, dsn):
     "trino": Trino,
     "clickhouse": Clickhouse,
     "vertica": Vertica,
+    "pymssql": MsSQL,
 }