Merge pull request #4 from kzajaczkowski/feature/collation-dev

Add collation and charset support for mariadb and mysql5
dbeatty10 · Apr 26, 2024 · b855c1e · b855c1e
2 parents 9da8355 + a40f2df
commit b855c1e
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/dbt/adapters/mariadb/connections.py b/dbt/adapters/mariadb/connections.py
@@ -26,6 +26,7 @@ class MariaDBCredentials(Credentials):
     password: Optional[str] = None
     charset: Optional[str] = None
     ssl_disabled: Optional[bool] = None
+    collation: Optional[str] = None
 
     _ALIASES = {
         "UID": "username",
@@ -98,6 +99,12 @@ def open(cls, connection):
         if credentials.port:
             kwargs["port"] = credentials.port
 
+        if credentials.charset:
+            kwargs["charset"] = credentials.charset
+
+        if credentials.collation:
+            kwargs["collation"] = credentials.collation
+
         try:
             connection.handle = mysql.connector.connect(**kwargs)
             connection.state = "open"

diff --git a/dbt/adapters/mysql5/connections.py b/dbt/adapters/mysql5/connections.py
@@ -26,6 +26,7 @@ class MySQLCredentials(Credentials):
     password: Optional[str] = None
     charset: Optional[str] = None
     ssl_disabled: Optional[bool] = None
+    collation: Optional[str] = None
 
     _ALIASES = {
         "UID": "username",
@@ -98,6 +99,12 @@ def open(cls, connection):
         if credentials.port:
             kwargs["port"] = credentials.port
 
+        if credentials.charset:
+            kwargs["charset"] = credentials.charset
+
+        if credentials.collation:
+            kwargs["collation"] = credentials.collation
+
         try:
             connection.handle = mysql.connector.connect(**kwargs)
             connection.state = "open"