added --mysql-socket option for unix socket path

Author: mackuba

README.md

@@ -51,6 +51,7 @@ Options:
   --mysql-password TEXT           MySQL password
   -h, --mysql-host TEXT           MySQL host. Defaults to localhost.
   -P, --mysql-port INTEGER        MySQL port. Defaults to 3306.
+  --mysql-socket TEXT             Path to MySQL unix socket file.
   -S, --skip-ssl                  Disable MySQL connection encryption.
   -i, --mysql-insert-method [DEFAULT|IGNORE|UPDATE]
                                   MySQL insert method. DEFAULT will throw

docs/README.rst

@@ -29,6 +29,7 @@ Connection Options
 - ``-h, --mysql-host TEXT``: MySQL host. Defaults to localhost.
 - ``-P, --mysql-port INTEGER``: MySQL port. Defaults to 3306.
 - ``-S, --skip-ssl``: Disable MySQL connection encryption.
+- ``--mysql-socket TEXT``: Path to MySQL unix socket file.
 
 Transfer Options
 """"""""""""""""

src/sqlite3_to_mysql/cli.py

@@ -62,6 +62,7 @@
 @click.option("--mysql-password", default=None, help="MySQL password")
 @click.option("-h", "--mysql-host", default="localhost", help="MySQL host. Defaults to localhost.")
 @click.option("-P", "--mysql-port", type=int, default=3306, help="MySQL port. Defaults to 3306.")
+@click.option("--mysql-socket", default=None, help="Path to MySQL unix socket file.")
 @click.option("-S", "--skip-ssl", is_flag=True, help="Disable MySQL connection encryption.")
 @click.option(
     "-i",
@@ -137,6 +138,7 @@ def cli(
     mysql_database: str,
     mysql_host: str,
     mysql_port: int,
+    mysql_socket: str,
     skip_ssl: bool,
     mysql_insert_method: str,
     mysql_truncate_tables: bool,
@@ -183,6 +185,7 @@ def cli(
             mysql_database=mysql_database,
             mysql_host=mysql_host,
             mysql_port=mysql_port,
+            mysql_socket=mysql_socket,
             mysql_ssl_disabled=skip_ssl,
             mysql_insert_method=mysql_insert_method,
             mysql_truncate_tables=mysql_truncate_tables,

src/sqlite3_to_mysql/transporter.py

@@ -74,6 +74,8 @@ def __init__(self, **kwargs: tx.Unpack[SQLite3toMySQLParams]):
 
         self._mysql_port = kwargs.get("mysql_port", 3306) or 3306
 
+        self._mysql_socket = str(kwargs.get("mysql_socket")) if kwargs.get("mysql_socket") else None
+
         self._sqlite_tables = kwargs.get("sqlite_tables") or tuple()
 
         self._without_foreign_keys = bool(self._sqlite_tables) or bool(kwargs.get("without_foreign_keys", False))
@@ -143,6 +145,7 @@ def __init__(self, **kwargs: tx.Unpack[SQLite3toMySQLParams]):
                 password=self._mysql_password,
                 host=self._mysql_host,
                 port=self._mysql_port,
+                unix_socket=self._mysql_socket,
                 ssl_disabled=self._mysql_ssl_disabled,
                 use_pure=True,
                 charset=self._mysql_charset,

src/sqlite3_to_mysql/types.py

@@ -20,6 +20,7 @@ class SQLite3toMySQLParams(tx.TypedDict):
     mysql_password: t.Optional[t.Union[str, bool]]
     mysql_host: t.Optional[str]
     mysql_port: t.Optional[int]
+    mysql_socket: t.Optional[str]
     mysql_ssl_disabled: t.Optional[bool]
     chunk: t.Optional[int]
     quiet: t.Optional[bool]
@@ -49,6 +50,7 @@ class SQLite3toMySQLAttributes:
     _mysql_password: t.Optional[str]
     _mysql_host: str
     _mysql_port: int
+    _mysql_socket: str
     _mysql_ssl_disabled: bool
     _chunk_size: t.Optional[int]
     _quiet: bool