Remove support for Python 2.7

issue7497
review53261002
This commit is contained in:
Cédric Krier committed 2018-06-26 19:11:47 +02:00
1 parent 20a28e5942
commit cc1e17304c
7 files changed
+27 -35

No files matched your search

+11 -11
View File
@@ -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')
+3 -4
View File
@@ -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')