我正在创建一个自定义命令这个命令我想将用户配置文件添加到所有公司用户所以这样做
companys = Users.objects.filter(is_company=True)
我已经创建了用户而没有为每个用户分配配置文件现在我想使用 for 循环遍历每个用户并为该用户创建配置文件但现在的挑战是获得一个UNIQUE CONSTRAINTS FAILED
我被第一个用户卡住了
from django.contrib.auth import get_user_model
from django.core.management.base import BaseCommand
from accounts import models as accounts
from company import models as company
class Command(BaseCommand):
def add_arguments(self, parser):
return super(Command, self).add_arguments(parser)
def handle(self, *args, **kwargs):
users = User.objects.all().filter(is_company=True)
seen = set()
for com_users in users:
print(com_users)
for company_detail in open('filenames/company_name.txt', 'r'):
country = generate_countrys()
name_email_tagline_address = company_detail.split('-')
name = name_email_tagline_address[0]
tagline = name_email_tagline_address[1]
address = name_email_tagline_address[2]
user = com_users
department = generate_department()
account_type = generate_account_type()
date_added = generate_date_added()
company_to_save = company.CompanyProfile.objects.create(
user=user, name=name, tagline=tagline, account_type=account_type, country=country,
joined_date=date_added, address=address, confirmed=True, department=department
)
company_to_save.save()
seen.add(users)
self.stdout.write(self.style.SUCCESS('Jobs Generated Successfully...'))
有没有办法实现这个?