summaryrefslogtreecommitdiff
path: root/tablib/compat.py
blob: 916d6ab42dedccdbfab3b9cf541aa33e0f6b34de (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
# -*- coding: utf-8 -*-

"""
tablib.compat
~~~~~~~~~~~~~

Tablib compatiblity module.

"""

import sys

is_py3 = (sys.version_info[0] > 2)


if is_py3:
    from io import BytesIO
    from itertools import zip_longest as izip_longest
    from statistics import median
    from tablib.packages import markup3 as markup
    import tablib.packages.dbfpy3 as dbfpy

    import csv
    from io import StringIO
    # py3 mappings

    ifilter = filter
    unicode = str
    bytes = bytes
    basestring = str
    xrange = range

else:
    from cStringIO import StringIO as BytesIO
    from cStringIO import StringIO
    from tablib.packages import markup
    from tablib.packages.statistics import median
    from itertools import ifilter, izip_longest

    import unicodecsv as csv
    import tablib.packages.dbfpy as dbfpy

    unicode = unicode
    xrange = xrange