django 使用随机值更新示例的有效方法

fcipmucu 于 2023-08-08 发布在 Go

关注(0)|答案(2)|浏览(109)

我正在使用Django和PostgreSQL，我想知道是否有一种方法可以用随机值更新一定数量的模型，这些模型具有uniques字段。
我的问题是我需要更新5k的用户，用随机值更改他的电子邮件和用户名。这些字段是唯一的，这意味着两个示例不能具有相同的值。我的逻辑是：对于每个用户，（1）生成随机的电子邮件字符串，然后如果没有任何用户使用该电子邮件，则使用该字符串，否则返回1

# Python2.7 Django 1.11, but everything helps
from django.contrib.auth.models import User
from django.utils.crypto import get_random_string

for user in User.objects.order_by('-pk')[:5000].iterator():

    # Generate random email
    while True:
        random_email = get_random_string(10)
        if not User.objects.filter(email=random_email).exists():
            user.email = random_email
            break

    # Generate random username
    while True:
        random_username = get_random_string(10)
        if not User.objects.filter(username=random_username).exists():
            user.username = random_username
            break

    user.save()

字符串

django

来源：https://stackoverflow.com/questions/61462779/efficient-way-to-update-instances-with-random-values

2条答案

按热度按时间

lnlaulya1#

IMO，您的实施足够有效。现在，如果您担心过多地访问DB，也许可以删除filter().exists()并预先生成唯一字符串列表。就像这样：

unique_strings = list(set([get_random_string(10) for x in range(15000)]))  # or use User.objects.all().count()

for x in range(15000 - len(unique_strings)):
   # I tested with 500,000 values, did not get any duplicates so it is highly unlikely to get in this block
   new_str = get_random_string(10)
   if not new_str in unique_strings:
        unique_strings.append(new_str)

for (unique_str, user) in zip(unique_strings, User.objects.all().iterator()):
    user.username = unique_str
    user.email = unique_str
    user.save()

字符串
然后DB命中将从3次减少到一次。

赞(0）回复(0）举报 2023-08-08

c2e8gylq2#

你可以使用bulk_update（在Django 2.2中添加）来提高效率。大致上，你把所有修改前的对象放在一个列表中，然后修改示例，然后一次调用bulk_update：

users = User.objects.all()

# Get a random string per user + 100 to cover (unlikely) collisions
# set() removes duplicates
random_emails = set(
    [get_random_string(10)
     for _ in range(users.count() + 100)])

# Here you could check if you have at least
# as many random_emails as users:
# assert len(random_emails) >= users.count()

for user, email in zip(users, random_emails):
    user.email = email

User.objects.bulk_update(users, ["email"])

字符串
一定要阅读Django文档中关于bulk_update的内容，因为它有一些限制，比如不能发出信号，不能调用save()方法等等。

赞(0）回复(0）举报 2023-08-08

我来回答

django 使用随机值更新示例的有效方法

2条答案

相关问题

热门标签

最新问答