Fix an import error in Python 3

2018-01-18 13:58:12 +01:00 · 2018-01-18 13:58:12 +01:00 · da0252974f
parent 02420d7a1b
commit da0252974f
1 changed files with 5 additions and 1 deletions
--- a/flatisfy/data_files/init.py
+++ b/flatisfy/data_files/init.py
@ -8,8 +8,12 @@ import io
 import json
 import logging
 import os
+import sys

-from backports import csv
+if sys.version_info >= (3,0):
+    import csv
+else:
+    from backports import csv

 from flatisfy.models.postal_code import PostalCode
 from flatisfy.models.public_transport import PublicTransport