16

有谁知道如何使用 sql 固定装置为 auth.User 加载初始数据?对于我的模型,我刚刚在名为 sql 的文件夹中有一个 < modelname >.sql 文件,syncdb 做得很好。但我不知道如何为 auth.User 模型做这件事。我用谷歌搜索了它,但没有成功。

提前致谢,

阿尔多

4

7 回答 7

23

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

更简单且独立于数据库的方法(除非您想要运行一些额外的 SQL 魔法),只需在应用程序的 fixtures 目录中创建一个 JSON 或 YAML夹具文件,其中包含如下数据:

- 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

于 2011-02-23T19:55:28.167 回答
8

您正在寻找loaddata

manage.py loadata path/to/your/fixtureFile

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

于 2010-03-09T20:37:23.957 回答
5

有一个技巧:(在 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. Include auth_fix in INSTALLED_APPS inside settings.py

下次运行时python manage.py syncdb,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. 只需在 settings.py 中的 INSTALLED_APPS 中包含 auth_fix。

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

于 2011-09-30T18:56:45.097 回答
4

感谢您的回答。我找到了适合我的解决方案,巧合的是布赖恩的建议之一。这里是:

首先,我在syncdb之后断开了创建超级用户的信号,因为我的超级用户在我的auth_user夹具中:

模型.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之后调用的信号:

< myproject >/< 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。

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

问候,

阿尔多

于 2010-03-10T14:51:19.483 回答
3

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

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

我同意 Felix 的观点,即 Django 中没有用于使用 SQL 填充 contrib 应用程序的非平凡自然钩子。

于 2010-03-09T23:07:28.270 回答
3

如果你碰巧用南做数据库迁移,创建用户非常简单。

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

$ 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': 'nancy@example.com',
            'password': 'nancypassword',
            'staff': True,
            'superuser': True
        },
        {
            'username': 'joe',
            'email': '',
            'password': 'joepassword',
            'staff': True,
            'superuser': False
        },
        {
            'username': 'susan',
            'email': 'susan@example.com',
            '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>
于 2011-11-05T22:50:29.610 回答
1

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

于 2013-04-10T12:27:16.170 回答