Make a GDAL csvt file for a csv file

makecsvt.py

"""
makecsvt.py - make GDAL .csvt file

Terry Brown, [email protected], Thu Aug 04 10:04:14 2016
"""

import csv
import sys

def get_type_text(path):
    reader = csv.reader(open(path))
    headers = next(reader)
    types = [int] * len(headers)
    # sys.stderr.write("%s\n" % ','.join(headers))
    for row in reader:
        for n, i in enumerate(row):
            if types[n] == int:
                try:
                    int(i)
                except ValueError:
                    types[n] = float
            if types[n] == float:
                try:
                    float(i)
                except ValueError:
                    types[n] = str
    return ','.join(
        '"%s"'%{int:'Integer', float:'Real', str:'String'}[i]
        for i in types
    )

def main():
    print(get_type_text(sys.argv[1]))

if __name__ == '__main__':
    main()