Skip to content

Commit a40f2df

Browse files
committed
Add collation and charset support for mariadb and mysql5
1 parent 9da8355 commit a40f2df

File tree

2 files changed

+14
-0
lines changed

2 files changed

+14
-0
lines changed

dbt/adapters/mariadb/connections.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@ class MariaDBCredentials(Credentials):
2626
password: Optional[str] = None
2727
charset: Optional[str] = None
2828
ssl_disabled: Optional[bool] = None
29+
collation: Optional[str] = None
2930

3031
_ALIASES = {
3132
"UID": "username",
@@ -98,6 +99,12 @@ def open(cls, connection):
9899
if credentials.port:
99100
kwargs["port"] = credentials.port
100101

102+
if credentials.charset:
103+
kwargs["charset"] = credentials.charset
104+
105+
if credentials.collation:
106+
kwargs["collation"] = credentials.collation
107+
101108
try:
102109
connection.handle = mysql.connector.connect(**kwargs)
103110
connection.state = "open"

dbt/adapters/mysql5/connections.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@ class MySQLCredentials(Credentials):
2626
password: Optional[str] = None
2727
charset: Optional[str] = None
2828
ssl_disabled: Optional[bool] = None
29+
collation: Optional[str] = None
2930

3031
_ALIASES = {
3132
"UID": "username",
@@ -98,6 +99,12 @@ def open(cls, connection):
9899
if credentials.port:
99100
kwargs["port"] = credentials.port
100101

102+
if credentials.charset:
103+
kwargs["charset"] = credentials.charset
104+
105+
if credentials.collation:
106+
kwargs["collation"] = credentials.collation
107+
101108
try:
102109
connection.handle = mysql.connector.connect(**kwargs)
103110
connection.state = "open"

0 commit comments

Comments
 (0)