有谁知道如何使用 sql 固定装置为 auth.User 加载初始数据?对于我的模型,我刚刚在名为 sql 的文件夹中有一个 < modelname >.sql 文件,syncdb 做得很好。但我不知道如何为 auth.User 模型做这件事。我用谷歌搜索了它,但没有成功。
提前致谢,
阿尔多
对于 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
。
有一个技巧:(在 Django 1.3.1 上测试)
解决方案:
python manage.py startapp auth_fix
mkdir auth_fix/fixtures
python manage.py dumpdata auth > auth_fixtures/fixtures/initial_data.json
Include auth_fix in INSTALLED_APPS inside settings.py
下次运行时python manage.py syncdb
,Django 会自动加载 auth 夹具。
解释:
python manage.py dumpdata auth
将使用所有组和用户信息转储数据库中的“auth”数据。该命令的其余部分只是将输出重定向到一个名为“initial_data.json”的文件中,这是 Django 在运行“syncdb”时查找的文件。此示例展示了如何在 JSON 中执行此操作,但您基本上可以使用您选择的格式。
感谢您的回答。我找到了适合我的解决方案,巧合的是布赖恩的建议之一。这里是:
首先,我在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。
我知道这个解决方案远非完美,非常欢迎批评/改进/意见!
问候,
阿尔多
一个选项是手动导入您的 auth.User SQL,然后将其转储到标准 Django 夹具(如果您希望 syncdb 找到它,请将其命名为 initial_data)。您通常可以将此文件放入任何应用程序的固定器目录中,因为固定器数据将全部使用正确的 app_label 键入。或者您可以创建一个空/虚拟应用程序并将其放置在那里。
另一种选择是覆盖 syncdb 命令并以您认为合适的方式应用夹具。
我同意 Felix 的观点,即 Django 中没有用于使用 SQL 填充 contrib 应用程序的非平凡自然钩子。
如果你碰巧用南做数据库迁移,创建用户非常简单。
首先,创建裸数据迁移。它需要包含在某些应用程序中。如果您有一个放置共享代码的通用应用程序,那将是一个不错的选择。如果您有一个应用程序可以集中与用户相关的代码,那就更好了。
$ 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>
我只是将 SQL 语句添加到另一个模型的自定义 sql 文件中。我选择了我的 Employee 模型,因为它依赖于 auth_user。我编写的自定义 SQL 实际上是从我的遗留应用程序中读取并从中提取用户信息,并使用 REPLACE 而不是 INSERT(我正在使用 MySQL),因此我可以随时运行它。我将 REPLACE...SELECT 语句放在一个过程中,以便手动运行或使用 cron 调度。