first comit

venv/lib/python3.10/site-packages/django/db/backends/mysql/creation.py

@@ -0,0 +1,87 @@
+import os
+import subprocess
+import sys
+
+from django.db.backends.base.creation import BaseDatabaseCreation
+
+from .client import DatabaseClient
+
+
+class DatabaseCreation(BaseDatabaseCreation):
+    def sql_table_creation_suffix(self):
+        suffix = []
+        test_settings = self.connection.settings_dict["TEST"]
+        if test_settings["CHARSET"]:
+            suffix.append("CHARACTER SET %s" % test_settings["CHARSET"])
+        if test_settings["COLLATION"]:
+            suffix.append("COLLATE %s" % test_settings["COLLATION"])
+        return " ".join(suffix)
+
+    def _execute_create_test_db(self, cursor, parameters, keepdb=False):
+        try:
+            super()._execute_create_test_db(cursor, parameters, keepdb)
+        except Exception as e:
+            if len(e.args) < 1 or e.args[0] != 1007:
+                # All errors except "database exists" (1007) cancel tests.
+                self.log("Got an error creating the test database: %s" % e)
+                sys.exit(2)
+            else:
+                raise
+
+    def _clone_test_db(self, suffix, verbosity, keepdb=False):
+        source_database_name = self.connection.settings_dict["NAME"]
+        target_database_name = self.get_test_db_clone_settings(suffix)["NAME"]
+        test_db_params = {
+            "dbname": self.connection.ops.quote_name(target_database_name),
+            "suffix": self.sql_table_creation_suffix(),
+        }
+        with self._nodb_cursor() as cursor:
+            try:
+                self._execute_create_test_db(cursor, test_db_params, keepdb)
+            except Exception:
+                if keepdb:
+                    # If the database should be kept, skip everything else.
+                    return
+                try:
+                    if verbosity >= 1:
+                        self.log(
+                            "Destroying old test database for alias %s..."
+                            % (
+                                self._get_database_display_str(
+                                    verbosity, target_database_name
+                                ),
+                            )
+                        )
+                    cursor.execute("DROP DATABASE %(dbname)s" % test_db_params)
+                    self._execute_create_test_db(cursor, test_db_params, keepdb)
+                except Exception as e:
+                    self.log("Got an error recreating the test database: %s" % e)
+                    sys.exit(2)
+        self._clone_db(source_database_name, target_database_name)
+
+    def _clone_db(self, source_database_name, target_database_name):
+        cmd_args, cmd_env = DatabaseClient.settings_to_cmd_args_env(
+            self.connection.settings_dict, []
+        )
+        dump_cmd = [
+            "mysqldump",
+            *cmd_args[1:-1],
+            "--routines",
+            "--events",
+            source_database_name,
+        ]
+        dump_env = load_env = {**os.environ, **cmd_env} if cmd_env else None
+        load_cmd = cmd_args
+        load_cmd[-1] = target_database_name
+
+        with subprocess.Popen(
+            dump_cmd, stdout=subprocess.PIPE, env=dump_env
+        ) as dump_proc:
+            with subprocess.Popen(
+                load_cmd,
+                stdin=dump_proc.stdout,
+                stdout=subprocess.DEVNULL,
+                env=load_env,
+            ):
+                # Allow dump_proc to receive a SIGPIPE if the load process exits.
+                dump_proc.stdout.close()