如何在 Django 中为用户模型加载 sql 夹具？

Question

有谁知道如何使用 sql 夹具加载 auth.User 的初始数据？ 对于我的模型，我只需要有一个 < modelname >.sql 文件位于名为 sql 的文件夹中，syncdb 完美地完成了它的工作。但我不知道如何为 auth.User 模型做到这一点。我用谷歌搜索过，但没有成功。

提前致谢，

奥尔多

Answer 1

对于 SQL 固定装置，您必须专门为 auth 表添加插入语句。您可以使用命令 python manage.py sql auth 找到 auth 表的架构。

更简单且独立于数据库的方法（除非您想要运行一些额外的 SQL 魔法），就是制作 JSON 或 YAML fixture 文件位于应用程序的装置目录中，其中包含如下数据：

- model: auth.user
  pk: 100000
  fields:
    first_name: Admin
    last_name: User
    username: admin
    password: "<a hashed password>"

您可以在 django 中快速生成哈希密码shell

>>> from django.contrib.auth.models import User
>>> u = User()
>>> u.set_password('newpass')
>>> u.password
'sha1$e2fd5$96edae9adc8870fd87a65c051e7fdace6226b5a8'

每当您运行 syncdb 时都会加载它。

Answer 2

您正在寻找 loaddata ：

manage.py loadata path/to/your/fixtureFile

但我认为该命令只能处理 XML、YAML、Python 或 JSON 格式的文件（参见此处）。要创建此类适当的文件，请查看 dumpdata 方法。

Answer 3

感谢您的回答。我找到了适合我的解决方案，巧合的是布莱恩的建议之一。首先，

我断开了在syncdb之后创建超级用户的信号，因为我的超级用户在我的auth_user固定装置中：

models.py：

from django.db.models import signals
from django.contrib.auth.management import create_superuser
from django.contrib.auth import models as auth_app


signals.post_syncdb.disconnect(
    create_superuser,
    sender=auth_app,
    dispatch_uid = "django.contrib.auth.management.create_superuser")

然后我创建了一个在syncdb之后调用的信号：

<强><我的项目>>/< myapp >/management/__init__.py

"""
Loads fixtures for files in sql/<modelname>.sql
"""
from django.db.models import get_models, signals
from django.conf import settings 
import <myproject>.<myapp>.models as auth_app

def load_fixtures(app, **kwargs):
    import MySQLdb
    db=MySQLdb.connect(host=settings.DATABASE_HOST or "localhost", \
       user=settings.DATABASE_USER,
    passwd=settings.DATABASE_PASSWORD, port=int(settings.DATABASE_PORT or 3306))

    cursor = db.cursor()

    try:
        print "Loading fixtures to %s from file %s." % (settings.DATABASE_NAME, \
            settings.FIXTURES_FILE)
        f = open(settings.FIXTURES_FILE, 'r')
        cursor.execute("use %s;" % settings.DATABASE_NAME)
        for line in f:
            if line.startswith("INSERT"):
                try:
                    cursor.execute(line)
                except Exception, strerror:
                    print "Error on loading fixture:"
                    print "-- ", strerror
                    print "-- ", line

        print "Fixtures loaded"

    except AttributeError:
        print "FIXTURES_FILE not found in settings. Please set the FIXTURES_FILE in \
            your settings.py" 

    cursor.close()
    db.commit()
    db.close()

signals.post_syncdb.connect(load_fixtures, sender=auth_app, \
    dispatch_uid = "<myproject>.<myapp>.management.load_fixtures")

在我的 settings.py 中，我添加了 FIXTURES_FILE 以及带有 sql 转储的 .sql 文件的路径。

我还没有找到的一件事是如何仅在创建表后触发此信号，而不是每次触发syncdb 时。解决这个问题的临时方法是在我的 sql 命令中使用 INSERT IGNORE INTO 。

我知道这个解决方案远非完美，非常欢迎批评/改进/意见！

问候，

奥尔多

Answer 4

有一个技巧：（在 Django 1.3.1 上测试）

解决方案：

1. python manage.py startapp auth_fix
2. mkdir auth_fix/fixtures
3. python manage.py dumpdata auth > auth_fixtures/fixtures/initial_data.json
4. 在settings.py内的INSTALLED_APPS中包含auth_fix

下次运行python manage.pysyncdb时，Django将自动加载auth固定装置。

说明：

1. 只需创建一个空应用程序来保存灯具文件夹。将 __init__py、models.py 和views.py 保留在其中，以便 Django 将其识别为应用程序而不仅仅是文件夹。
2. 在应用程序中创建fixtures文件夹。
3. python manage.py dumpdata auth 将转储数据库中的“auth”数据以及所有组和用户信息。该命令的其余部分只是将输出重定向到名为“initial_data.json”的文件中，该文件是 Django 在运行“syncdb”时查找的文件。
4. 只需将 auth_fix 包含在 settings.py 内的 INSTALLED_APPS 中即可。

此示例展示了如何在 JSON 中执行此操作，但您基本上可以使用您选择的格式。

Answer 5

如果您碰巧正在使用South进行数据库迁移，那么创建用户非常简单。

首先，创建裸数据迁移。它需要包含在某些应用程序中。如果您有一个放置共享代码的通用应用程序，那将是一个不错的选择。如果您有一个集中用户相关代码的应用程序，那就更好了。

$ python manage.py datamigration <some app name> add_users

相关的迁移代码可能如下所示：

# encoding: utf-8
import datetime
from south.db import db
from south.v2 import DataMigration
from django.db import models
from django.contrib.auth.models import User

class Migration(DataMigration):

    users = [
        {
            'username': 'nancy',
            'email': '[email protected]',
            'password': 'nancypassword',
            'staff': True,
            'superuser': True
        },
        {
            'username': 'joe',
            'email': '',
            'password': 'joepassword',
            'staff': True,
            'superuser': False
        },
        {
            'username': 'susan',
            'email': '[email protected]',
            'password': 'susanpassword',
            'staff': False,
            'superuser': False
        }
    ]

    def forwards(self, orm):
        """
        Insert User objects
        """
        for i in Migration.users:
            u = User.objects.create_user(i['username'],  i['email'], i['password'])
            u.is_staff = i['staff']
            u.is_superuser = i['superuser']
            u.save()

    def backwards(self, orm):
        """
        Delete only these users
        """
        for i in Migration.users:
            User.objects.filter(username=i['username']).delete()

然后只需运行迁移即可插入身份验证用户。

$ python manage.py migrate <some app name>

Answer 6

一个选项是手动导入 auth.User SQL，然后将其转储到标准 Django 固定装置（如果您希望syncdb 找到它，请将其命名为initial_data）。您通常可以将此文件放入任何应用程序的固定装置目录中，因为固定数据都将使用正确的应用程序标签进行键入。或者您可以创建一个空/虚拟应用程序并将其放置在那里。

另一种选择是覆盖syncdb命令并以您认为合适的方式应用固定装置。

我同意 Felix 的观点，Django 中不存在用于使用 SQL 填充 contrib 应用程序的重要自然钩子。

Answer 7

我只是将 SQL 语句添加到另一个模型的自定义 sql 文件中。我选择了 Employee 模型，因为它取决于 auth_user。
我编写的自定义 SQL 实际上从我的遗留应用程序中读取并从中提取用户信息，并使用 REPLACE 而不是 INSERT（我使用的是 MySQL），因此我可以随时运行它。
我将 REPLACE...SELECT 语句放在一个过程中，以便可以轻松手动运行或使用 cron 计划运行。