import tarfile
import os

# Direktorij u kojem su DKP podaci
dkp_source  = 'C:/Users/darko.boto/Desktop/DKP'
# Direktori u koji se trebaju extractati podaci
output_dir  = 'C:/Users/darko.boto/Desktop/DKP_OUT'

# Parametri za spajanje na bazu
host    = 'localhost'
db_name = 'GIS'
user    = 'postgres'
passwd  = 'postgres'
port    = '5432'

shp_code_kc = '2020' # kod za prepoznavanje shp datoteka u kojima su KAT cestice

def main():
    for root, dirs, files in os.walk(dkp_source):
        for file in files:
            if file.endswith('.tar.gz'):
                tar = tarfile.open(os.path.join(root, file), "r:gz")
                for tarinfo in tar:
                    if tarinfo.isreg():
                        shp_code = tarinfo.name[-8:-4]
                        if shp_code == shp_code_kc:
                            tarinfo.path = tarinfo.path.split('/')[2]
                            print'Extracting:', tarinfo.path
                            tar.extract(tarinfo, output_dir)

def load():
    for root, dirs, files in os.walk(output_dir):
        for file in files:
            if file.endswith('.shp'):
                shp_path = os.path.join(root, file)
                print 'Loading shapefile:  ', file
                ogr2ogr_str = 'ogr2ogr -append -s_srs EPSG:3765 -t_srs EPSG:3765 -f "PostgreSQL" PG:"host=%s port=%s user=%s password=%s dbname=%s" -nln dkp_kc_load -nlt MULTIPOLYGON "%s" -progress' % (host, port, user, passwd, db_name, shp_path)
                os.system(ogr2ogr_str)

if __name__ == '__main__':
    main()
    #load()