48 lines
1.3 KiB
Python
48 lines
1.3 KiB
Python
|
from django.core.management.base import BaseCommand, CommandError
|
||
|
|
|
||
|
|
from events.models.locale import Country
|
||
|
|
|
||
|
|
# Fields from geoname table, from http://download.geonames.org/export/dump/readme.txt
|
||
|
|
ISO=0
|
||
|
|
ISO3=1
|
||
|
|
ISO_NUMERIC=2
|
||
|
|
FIPS=3
|
||
|
|
COUNTRY=4
|
||
|
|
CAPITAL=5
|
||
|
|
AREA=6
|
||
|
|
POPULATION=7
|
||
|
|
CONTINENT=8
|
||
|
|
TLD=9
|
||
|
|
CURRENCYCODE=10
|
||
|
|
CURRENCYNAME=11
|
||
|
|
PHONE=12
|
||
|
|
POSTAL_CODE_FORMAT=13
|
||
|
|
POSTAL_CODE_REGEX=14
|
||
|
|
LANGUAGES=15
|
||
|
|
GEONAMEID=16
|
||
|
|
NEIGHBOURS=17
|
||
|
|
EQUIVALENTFIPSCODE=18
|
||
|
|
|
||
|
|
class Command(BaseCommand):
|
||
|
|
help = 'Loads country data from GeoNames database file'
|
||
|
|
|
||
|
|
def add_arguments(self, parser):
|
||
|
|
parser.add_argument('file', type=str)
|
||
|
|
|
||
|
|
def handle(self, *args, **options):
|
||
|
|
if 'file' in options:
|
||
|
|
countries_file = open(options['file'], 'r')
|
||
|
|
for country_line in countries_file.readlines():
|
||
|
|
if country_line.startswith("#"):
|
||
|
|
continue
|
||
|
|
country = country_line.split("\t")
|
||
|
|
if len(country) == 19:
|
||
|
|
#print("%s - %s" % (country[ISO], country[COUNTRY]))
|
||
|
|
Country.objects.get_or_create(name=country[COUNTRY], code=country[ISO])
|
||
|
|
else:
|
||
|
|
print("Short line (%s): %s" % (len(country), country_line))
|
||
|
|
countries_file.close()
|
||
|
|
|
||
|
|
else:
|
||
|
|
print("No File in options!")
|