#!/usr/bin/python

"""
PYTHONPATH="../.." DJANGO_SETTINGS_MODULE="settings" python load_esri.py ../../epsg/gdal/esri_extra.wkt
"""

from osgeo import osr
import sys
from srsbrowser.models import * 

def run(file, type):
    f = open(file)
    for line in f:
        if line[0] == "#":
            continue
        (id,wkt) = line.split(",", 1)
        srs = osr.SpatialReference(wkt)
        title = srs.GetAttrValue("PROJCS") or srs.GetAttrValue("GEOGCS")
        title = title.replace("_", " ")
        s = SRS.objects.filter(type=type, srs_id=id)
        if s.count():
            s = s[0]
            s.title = title
            s.wkt = wkt
        else:
            s = SRS(srs_id=id, title=title, wkt=wkt, type=type)
        s.save()
        print "Saved %s:%s as %s" % (type.upper(), id, s.id)
if __name__ == "__main__":
        print "Uage: %s file.wkt (esri|iau2000)" % sys.argv[0]
        run(sys.argv[1], sys.argv[2])