iapb2postgis

import psycopg2
import sys
import datetime

try:
    conn = psycopg2.connect(database='gis', user='gisuser')
    curs = conn.cursor()

#    try:
#        # Try create db for IAPB data
#        curs.execute("""CREATE TABLE iapb (id integer, date date, geom geometry)""")
#        conn.commit()
#    except:
#        print 'The table already exist!'

    ff_in = sys.argv[1]
    ff = open(ff_in, 'r')
    lines = ff.readlines()

    for line in lines:
        year, mon, day, hh, id, lat, lon = (line.replace('\n','')).split()
        #print 'Year: %s Month: %s Day: %s Hour: %s ID: %s lat: %s lon: %s' % (year, mon, day, hh, id, lat, lon)
        if int(mon)<10:
            mon = '0' + mon
        if int(day)<10:
            day = '0' + day
        if int(hh)<10:
            hh = '0' + hh
        try:
            dt_str = '%s-%s-%s %s:00' % (year, mon, day, hh)
            idate = datetime.datetime.strptime(dt_str, '%Y-%m-%d %H:%M')
            #print idate, id
            try:
                sql = "INSERT INTO iapb (id,date,geom) VALUES (%s,%s,ST_GeomFromText(%s));"
                params = [int(id),dt_str,'POINT( %s %s )' % (lon,lat)]
                try:
                    if int(hh)==0:
                        curs.execute(sql, params)
                        conn.commit()
                except:
                    print "Execute insert error!"

            except:
                print 'Insert error!'


        except:
            print 'Date formating error!'
    conn.close()
    ff.close()

except:
    print 'Connect error!'