Remove support for Python 2.7
issue7497 review53261002
This commit is contained in:
1 parent
20a28e5942
commit
cc1e17304c
7 files changed
+27
-35
No files matched your search
+11
-11
@@ -5,12 +5,12 @@
|
||||
import sys
|
||||
import pycountry
|
||||
|
||||
sys.stdout.write(u'<?xml version="1.0"?>\n')
|
||||
sys.stdout.write(u'<tryton>\n')
|
||||
sys.stdout.write(u' <data skiptest="1" grouped="1">\n')
|
||||
sys.stdout.write('<?xml version="1.0"?>\n')
|
||||
sys.stdout.write('<tryton>\n')
|
||||
sys.stdout.write(' <data skiptest="1" grouped="1">\n')
|
||||
|
||||
for country in pycountry.countries:
|
||||
record = u'''
|
||||
record = '''
|
||||
<record model="country.country" id="%s">
|
||||
<field name="name">%s</field>
|
||||
<field name="code">%s</field>
|
||||
@@ -19,12 +19,12 @@ for country in pycountry.countries:
|
||||
</record>\n''' % (country.alpha_2.lower(), country.name,
|
||||
country.alpha_2, country.alpha_3, country.numeric)
|
||||
sys.stdout.write(record.encode('utf-8'))
|
||||
sys.stdout.write(u' </data>\n')
|
||||
sys.stdout.write(' </data>\n')
|
||||
|
||||
subdivision_codes = {s.code.lower() for s in pycountry.subdivisions}
|
||||
existing_parents = set()
|
||||
while existing_parents < subdivision_codes:
|
||||
sys.stdout.write(u' <data skiptest="1" grouped="1">\n')
|
||||
sys.stdout.write(' <data skiptest="1" grouped="1">\n')
|
||||
new_parents = set()
|
||||
for subdivision in pycountry.subdivisions:
|
||||
if (subdivision.code.lower() in existing_parents
|
||||
@@ -33,7 +33,7 @@ while existing_parents < subdivision_codes:
|
||||
existing_parents)):
|
||||
continue
|
||||
new_parents.add(subdivision.code.lower())
|
||||
record = u'''
|
||||
record = '''
|
||||
<record model="country.subdivision" id="%s">
|
||||
<field name="name">%s</field>
|
||||
<field name="code">%s</field>
|
||||
@@ -41,15 +41,15 @@ while existing_parents < subdivision_codes:
|
||||
subdivision.name, subdivision.code,
|
||||
subdivision.type.lower())
|
||||
if subdivision.parent_code:
|
||||
record += u'''\
|
||||
record += '''\
|
||||
<field name="parent" ref="%s"/>\n''' % \
|
||||
subdivision.parent.code.lower()
|
||||
record += u'''\
|
||||
record += '''\
|
||||
<field name="country" ref="%s"/>
|
||||
</record>\n''' % subdivision.country.alpha_2.lower()
|
||||
|
||||
sys.stdout.write(record.encode('utf-8'))
|
||||
sys.stdout.write(u' </data>\n')
|
||||
sys.stdout.write(' </data>\n')
|
||||
existing_parents |= new_parents
|
||||
|
||||
sys.stdout.write(u'</tryton>\n')
|
||||
sys.stdout.write('</tryton>\n')
|
||||
@@ -76,10 +76,7 @@ def import_(data):
|
||||
widgets=[SimpleProgress(), Bar(), ETA()])
|
||||
else:
|
||||
pbar = iter
|
||||
if sys.version_info < (3,):
|
||||
f = BytesIO(data)
|
||||
else:
|
||||
f = StringIO(data.decode('utf-8'))
|
||||
f = StringIO(data.decode('utf-8'))
|
||||
for row in pbar(list(csv.DictReader(
|
||||
f, fieldnames=_fieldnames, delimiter='\t'))):
|
||||
country = get_country(row['country'])
|
||||
@@ -87,6 +84,7 @@ def import_(data):
|
||||
Zip(country=country, subdivision=subdivision, zip=row['postal'],
|
||||
city=row['place']).save()
|
||||
|
||||
|
||||
_fieldnames = ['country', 'postal', 'place', 'name1', 'code1',
|
||||
'name2', 'code2', 'name3', 'code3', 'latitude', 'longitude', 'accuracy']
|
||||
|
||||
@@ -100,6 +98,7 @@ def main(database, codes, config_file=None):
|
||||
clean(code)
|
||||
import_(fetch(code))
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
parser = ArgumentParser()
|
||||
parser.add_argument('-d', '--database', dest='database')
|
||||
|
||||
Reference in new issue
Block a user