django_address模块是否提供了一种为初始国家数据设定种子的方法?



我正在使用Django 2.0,Python 3.7和MySql 5。 我最近安装了django_address模块。 我注意到当我根据我的 models.py 文件运行初始迁移时......

from django.db import models
from address.models import AddressField
from phonenumber_field.modelfields import PhoneNumberField

class CoopType(models.Model):
name = models.CharField(max_length=200, null=False)
class Meta:
unique_together = ("name",)

class Coop(models.Model):
type = models.ForeignKey(CoopType, on_delete=None)
address = AddressField(on_delete=models.CASCADE)
enabled = models.BooleanField(default=True, null=False)
phone = PhoneNumberField(null=True)
email = models.EmailField(null=True)
web_site = models.TextField()

它创建了一些地址表,包括...

mysql> show create table address_country;
+-----------------+---------------------------------------------------+
| Table           | Create Table                                      |
+-----------------+---------------------------------------------------+
| address_country | CREATE TABLE `address_country` (                  |
|                 |   `id` int(11) NOT NULL AUTO_INCREMENT,           |
|                 |   `name` varchar(40) COLLATE utf8_bin NOT NULL,   |
|                 |   `code` varchar(2) COLLATE utf8_bin NOT NULL,    |
|                 |   PRIMARY KEY (`id`),                             |
|                 |   UNIQUE KEY `name` (`name`)                      |
|                 | ) ENGINE=InnoDB                                   |
|                 | DEFAULT CHARSET=utf8 COLLATE=utf8_bin             |
+-----------------+---------------------------------------------------+

但是,此表中没有数据。 有没有办法获取模块生成的表的种子数据,还是我需要自己挖掘它?

您可以使用pycountry包自己轻松生成国家/地区。

由于创建的Country模型上的code字段的最大长度为两个字符,因此您需要使用alpha_2代码。

我通常使用自定义管理命令来处理此类事情。也许添加一个检查以查看是否已创建任何对象,然后根据需要进行处理。

外壳python manage.py create_countries的用法

from address.models import Country
from pycountry import countries
from django.core.management.base import BaseCommand
class Command(BaseCommand):
help = 'Initialize Country model'
def handle(self, *args, **kwargs):
create_countries = [
Country(name=country.name[:40], code=country.alpha_2)
for country in countries
]
Country.objects.bulk_create(create_countries)
self.stdout.write(f'Created {len(countries)} countries.n')

如果生产服务器没有运行 Python/Django,那么您可以使用pycountry创建一个包含相关数据的 CSV 文件。假设你使用的是PostgreSQL,那么你可以使用COPY FROM命令来填充数据库。

import csv
from pycountry import countries
with open('countries.csv', mode='w') as countries_file:
# specify delimiter because some countries have a comma
writer = csv.writer(countries_file, delimiter='t')
writer.writerow(['id', 'name', 'code'])
writer.writerows([
[index + 1, country.name, country.alpha_2]
for index, country in enumerate(countries)
])

我建议你编写一个简单的管理命令,将数据从pycountry导入到你的地址模型中(从这里借用的方法(。 pycountry是ISO标准国家列表的包装器 - 也就是说,它与您将要获得的国家列表一样规范。

将所有国家/地区填充到Country模型中的管理命令如下所示:

import pycountry
from django.core.management.base import BaseCommand, CommandError
from address.models import Country
class Command(BaseCommand):
help = "Populates address.Country with data from pycountry."
def handle(self, *args, **options):
countries = [
Country(
code=country.alpha_2,
name=country.name[:40],  # NOTE - concat to 40 chars because of limit on the model field
)
for country in pycountry.countries
]
Country.objects.bulk_create(countries)
self.stdout.write("Successfully added %s countries." % len(countries))

这将使用ISO国家/地区列表填充您的模型。

这里需要注意的是,address.Country.name字段限制为 40 个字符(这对我来说似乎是一个有问题的设计决策,以及不使国家/地区代码唯一的决定 - ISO 2 字母代码绝对是唯一的(,因此上面的脚本截断了名称以适应。如果这对您来说是一个问题,我建议您设置自己的地址模型,借用django-address并提高字符限制。

如上所述,您需要自己添加实际数据。您需要准备它并进行一次性上传。如果您只查找国家/地区数据,这是一个很好的来源。还有一个名为django-countries的django应用程序,它可以让您拥有更多的数据和控件,包括标志,ISO代码等。另一个具有 3 个字母代码的数据库是 IBAN 列表。希望有帮助。

你可以使用 django-countries,并在你的模型中包含一个 CountryField。这包括对模型和表单选择字段的支持。

由于这是内置的,因此您可以将其包含在模型中,而不必担心播种表。

简单更好。通过 django 文档,您可以创建一个数据迁移: https://docs.djangoproject.com/en/3.0/topics/migrations/#data-migrations

假设models.py的应用名称是coops,请执行以下操作:

第一:

  • pip install django-countries(我更喜欢pipenv安装(https://github.com/SmileyChris/django-countries

第二:

  • django_countries添加到INSTALLED_APPS

第三:

  • 创建一个空的迁移文件:python manage.py makemigrations --empty coops

四:

  • 编辑迁移文件,例如:vi coops/migrations/0002_auto_20200205_0421.py
  • 文件内容:
# Generated by Django 2.2 on 2020-02-05 04:21
from django.db import migrations

def init_countries(apps, schema_editor):
from django_countries import countries
from address.models import Country
countries = [
Country(code=code, name=name) for code, name in countries
]
Country.objects.bulk_create(countries)

class Migration(migrations.Migration):
dependencies = [
('coops', '0001_initial'),
]
operations = [
migrations.RunPython(init_countries),
]

第五:

  • 运行python manage.py migrate

迁移后address_country表应包含如下数据:

In [1]: from address.models import *
In [2]: Country.objects.all()
Out[2]: <QuerySet [<Country: Afghanistan>, <Country: Albania>, <Country: Algeria>, <Country: American Samoa>, <Country: Andorra>, <Country: Angola>, <Country: Anguilla>, <Country: Antarctica>, <Country: Antigua and Barbuda>, <Country: Argentina>, <Country: Armenia>, <Country: Aruba>, <Country: Australia>, <Country: Austria>, <Country: Azerbaijan>, <Country: Bahamas>, <Country: Bahrain>, <Country: Bangladesh>, <Country: Barbados>, <Country: Belarus>, '...(remaining elements truncated)...']>
In [3]: Country.objects.all().count()
Out[3]: 249

最新更新