我正在使用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