Primer commit del proyecto RSS

venv/lib/python3.12/site-packages/mysql/connector/django/client.py

@@ -0,0 +1,106 @@
+# Copyright (c) 2020, 2024, Oracle and/or its affiliates.
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License, version 2.0, as
+# published by the Free Software Foundation.
+#
+# This program is designed to work with certain software (including
+# but not limited to OpenSSL) that is licensed under separate terms,
+# as designated in a particular file or component or in included license
+# documentation. The authors of MySQL hereby grant you an
+# additional permission to link the program and your derivative works
+# with the separately licensed software that they have either included with
+# the program or referenced in the documentation.
+#
+# Without limiting anything contained in the foregoing, this file,
+# which is part of MySQL Connector/Python, is also subject to the
+# Universal FOSS Exception, version 1.0, a copy of which can be found at
+# http://oss.oracle.com/licenses/universal-foss-exception.
+#
+# This program is distributed in the hope that it will be useful, but
+# WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
+# See the GNU General Public License, version 2.0, for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software Foundation, Inc.,
+# 51 Franklin St, Fifth Floor, Boston, MA 02110-1301  USA
+
+"""Database Client."""
+
+import os
+import subprocess
+
+from typing import Any, Dict, Iterable, List, Optional, Tuple
+
+from django.db.backends.base.client import BaseDatabaseClient
+
+
+class DatabaseClient(BaseDatabaseClient):
+    """Encapsulate backend-specific methods for opening a client shell."""
+
+    executable_name = "mysql"
+
+    @classmethod
+    def settings_to_cmd_args_env(
+        cls, settings_dict: Dict[str, Any], parameters: Optional[Iterable[str]] = None
+    ) -> Tuple[List[str], Optional[Dict[str, Any]]]:
+        args = [cls.executable_name]
+
+        db = settings_dict["OPTIONS"].get("database", settings_dict["NAME"])
+        user = settings_dict["OPTIONS"].get("user", settings_dict["USER"])
+        passwd = settings_dict["OPTIONS"].get("password", settings_dict["PASSWORD"])
+        host = settings_dict["OPTIONS"].get("host", settings_dict["HOST"])
+        port = settings_dict["OPTIONS"].get("port", settings_dict["PORT"])
+        ssl_ca = settings_dict["OPTIONS"].get("ssl_ca")
+        ssl_cert = settings_dict["OPTIONS"].get("ssl_cert")
+        ssl_key = settings_dict["OPTIONS"].get("ssl_key")
+        defaults_file = settings_dict["OPTIONS"].get("read_default_file")
+        charset = settings_dict["OPTIONS"].get("charset")
+
+        # --defaults-file should always be the first option
+        if defaults_file:
+            args.append(f"--defaults-file={defaults_file}")
+
+        # Load any custom init_commands. We always force SQL_MODE to TRADITIONAL
+        init_command = settings_dict["OPTIONS"].get("init_command", "")
+        args.append(f"--init-command=SET @@session.SQL_MODE=TRADITIONAL;{init_command}")
+
+        if user:
+            args.append(f"--user={user}")
+        if passwd:
+            args.append(f"--password={passwd}")
+
+        if host:
+            if "/" in host:
+                args.append(f"--socket={host}")
+            else:
+                args.append(f"--host={host}")
+
+        if port:
+            args.append(f"--port={port}")
+
+        if db:
+            args.append(f"--database={db}")
+
+        if ssl_ca:
+            args.append(f"--ssl-ca={ssl_ca}")
+        if ssl_cert:
+            args.append(f"--ssl-cert={ssl_cert}")
+        if ssl_key:
+            args.append(f"--ssl-key={ssl_key}")
+
+        if charset:
+            args.append(f"--default-character-set={charset}")
+
+        if parameters:
+            args.extend(parameters)
+
+        return args, None
+
+    def runshell(self, parameters: Optional[Iterable[str]] = None) -> None:
+        args, env = self.settings_to_cmd_args_env(
+            self.connection.settings_dict, parameters
+        )
+        env = {**os.environ, **env} if env else None
+        subprocess.run(args, env=env, check=True)