在 django 中迁移后运行 .sql 文件

Question

我已经在 django 中设置了我的迁移（非常小的项目）；但我有一个 .sql 文件，其中包含我在数据库中需要的大量数据。

在运行迁移之后/期间执行此 .sql 文件的最佳方式是什么（或者甚至可能）？

sql 文件只包含插入的数据，如下所示：

INSERT INTO `mileages_mileages` (`id`, `miles`, `start_location`, `end_location`) VALUES
(NULL,3, 'Location 1', 'Location 2'),

我只需要在运行模型的初始迁移后执行那个 .sql 文件。

这可能吗？

Answer 1

migrations.RunSQL()

不接受文件作为输入。只有原始 SQL。要解决这个问题，您需要使用：

migrations.RunSQL(
    """
    INSERT INTO 'mileages_mileages' ('id', 'miles', 'start_location', 'end_location') 
    VALUES
       (NULL,3, 'Location 1', 'Location 2');
    """)

或

def load_data_from_sql(apps, schema_editor):
   file_path = os.path.join(os.path.dirname(__file__), 'file_name.sql')
   sql_statement = open(file_path).read()
   with connection.cursor() as c:
       c.execute(sql_statement)

class Migration(migrations.Migration):
    dependencies = [
        ('..', '...'),
    ]

    operations = [
        migrations.RunPython(load_data_from_sql),
    ]

Answer 2

阅读更多内容后 - 运行 SQL 以在迁移中加载数据是错误的做法。使用fixtures 会好很多。

我已经添加了一个夹具，并将使用该文件来引入数据。

首先，要创建夹具，我必须将数据添加到表中（我只是手动运行 SQL 文件以将其添加到相应的表中）。

然后我跑了

manage.py dumpdata mileage.mileage --indent 2 --format=json --output initial_mileages.json

这创建了我的固定装置，然后我加载了它

manage.py loaddata initial_mileages.json

而且它导入得很漂亮。

Answer 3

使用migrations.RunSQL时，如果出现如下错误：

如果不拆分 SQL，则需要 sqlparse

你只需要安装sqlparse即可解决问题：

pip install sqlparse

Answer 4

这里是完整的脚本（不完美，但原样），用于将带 DELIMITER 的 MySQL 语句转换为 SQL 执行。

关于 SQL 语句中的多重分隔符https://stackoverflow.com/a/52292690/9521312

在迁移文件中添加脚本执行

有两种使用脚本的例子：运行 sql 文件或运行原始 MySQL 语句

from anywhere import migrate_run_sql

operations = [
              migrations.RunPython(migrate_run_sql.run_sql_file('contract_triggers.sql')),
              migrations.RunPython(migrate_run_sql.run_sql(
                                                           """
                                                           DELIMITER $$
                                                           CREATE TRIGGER trigger_name BEFORE INSERT ON table
                                                           FOR EACH ROW
                                                           BEGIN
                                                             IF NEW.number <> 'anynumber' AND NEW.number <> 'anynumber'
                                                               THEN
                                                                 SET NEW.number = 'anynumber';
                                                             END IF;
                                                           END$$
                                                           """
                                                           ))
             ]

脚本文件

# -*- coding: utf-8 -*-
from django.db import connection
import re
from StringIO import StringIO
from django.conf import settings
import os

# this function get raw MySQL statement
def run_sql(sql): 
    def load_data_from_sql(app, schema_editor):
        f = StringIO(sql)
        return _runsql(f)

    return load_data_from_sql

# this function get sql file
def run_sql_file(filename):
    def load_data_from_sql(app, schema_editor):
        filepath = os.path.join(settings.PROJECT_PATH, '../deploy/mysql/', filename)
        with open(filepath, 'rb') as f:
            return _runsql(f)

    return load_data_from_sql

# in this function content splits and checks line by line
def _runsql(f):
    with connection.cursor() as c:
        file_data = f.readlines()
        statement = ''
        delimiter = ';\n'
        for line in file_data:
            if re.findall('DELIMITER', line): # found delimiter
                if re.findall('^\s*DELIMITER\s+(\S+)\s*$', line):
                    delimiter = re.findall('^\s*DELIMITER\s+(\S+)\s*$', line)[0] + '\n'
                    continue
                else:
                    raise SyntaxError('Your usage of DELIMITER is not correct, go and fix it!')
            statement += line # add lines while not met lines with current delimiter
            if line.endswith(delimiter):
                if delimiter != ';\n':
                    statement = statement.replace(';', '; --').replace(delimiter, ';') # found delimiter, add dash symbols (or any symbols you want) for converting MySQL statements with multiply delimiters in SQL statement
                c.execute(statement) # execute current statement
                statement = '' # begin collect next statement

希望对您有所帮助！

Answer 5

您可以执行原始 SQL

operations = [
    migrations.RunSQL('sql statement here')
]