如何使用manage.py加载.sql文件到PostgreSQL数据库？

嘿，你踩了个常见的小坑——python manage.py loaddata是专门用来加载Django fixture文件的（比如.json、.xml或.yaml格式，这些是Django导出的模型数据），直接给它喂.sql文件肯定会报错。下面给你两种靠谱的方案，把你的.sql文件导入PostgreSQL数据库：

方法一：直接用PostgreSQL原生psql命令（最推荐） #

这是最直接高效的方式，不需要依赖Django的命令：

• 先打开你的Django项目的settings.py，找到DATABASES配置块，记下NAME（数据库名）、USER（数据库用户名）、HOST（主机）、PORT（端口）这些参数。
• 打开终端，运行以下格式的命令：

  psql -h <数据库主机> -p <端口> -U <用户名> -d <数据库名> -f <你的sql文件绝对路径>
  

  举个实际例子：

  psql -h localhost -p 5432 -U my_django_user -d my_project_db -f /home/user/backup/prod_data.sql
  

• 如果你的数据库是本地的，且用的是PostgreSQL默认配置，命令可以简化成：

  psql -U my_django_user -d my_project_db -f /home/user/backup/prod_data.sql
  

• 执行命令后，输入数据库用户的密码，等待脚本执行完成即可。

方法二：通过Django环境执行（适合不想切换到纯PostgreSQL命令的场景） #

如果你想在Django的上下文里完成操作，可以用manage.py shell或者写个自定义命令：

方式A：用manage.py shell快速执行 #

1. 运行命令进入Django shell：

   python manage.py shell
   

2. 在shell里输入以下代码（记得替换/path/to/your/file.sql为你的实际文件路径）：

   import subprocess
   from django.conf import settings
   
   # 从Django配置里读取数据库参数
   db = settings.DATABASES['default']
   # 构造并执行psql命令
   subprocess.run([
       'psql',
       '-h', db['HOST'],
       '-p', db['PORT'],
       '-U', db['USER'],
       '-d', db['NAME'],
       '-f', '/path/to/your/file.sql'
   ], check=True)
   

3. 按回车执行，输入数据库密码即可。

方式B：写个自定义Django命令（适合重复使用） #

如果以后还要频繁导入，你可以在Django app下的management/commands目录（没有就创建）里写个自定义命令，比如叫importsql.py：

from django.core.management.base import BaseCommand
from django.conf import settings
import subprocess

class Command(BaseCommand):
    help = 'Import a SQL file into the PostgreSQL database'

    def add_arguments(self, parser):
        parser.add_argument('sql_file', type=str, help='Path to the SQL file')

    def handle(self, *args, **options):
        sql_file = options['sql_file']
        db = settings.DATABASES['default']
        try:
            subprocess.run([
                'psql',
                '-h', db['HOST'],
                '-p', db['PORT'],
                '-U', db['USER'],
                '-d', db['NAME'],
                '-f', sql_file
            ], check=True)
            self.stdout.write(self.style.SUCCESS('SQL file imported successfully!'))
        except subprocess.CalledProcessError as e:
            self.stderr.write(self.style.ERROR(f'Failed to import SQL file: {e}'))

然后就可以用以下命令导入：

python manage.py importsql /path/to/your/file.sql

几个重要注意事项 #

• 确保你的PostgreSQL服务处于运行状态，不然会连不上数据库。
• 执行命令的用户需要有足够的数据库权限（比如CREATE TABLE、INSERT、UPDATE等），如果权限不足，先给用户赋权。
• 如果你的.sql文件里包含创建数据库的语句，要先切换到postgres系统数据库执行，或者确保目标数据库已经存在。
• 如果生产导出的.sql和本地数据库结构有冲突（比如表已存在），可以先清理本地数据库，或者在导入前检查sql文件里的语句，必要时添加DROP TABLE IF EXISTS之类的语句避免报错。

内容的提问来源于stack exchange，提问作者TJB