Geografic-Information-System/lib/python/pygrass/tests/set_mapset.py

# -*- coding: utf-8 -*-
"""
Created on Thu Aug 23 11:07:38 2012

@author: pietro

"""
from __future__ import (nested_scopes, generators, division, absolute_import,
                        with_statement, print_function, unicode_literals)

import os
import subprocess
import optparse

from grass.script import core as grasscore

def read_gisrc(gisrcpath):
    gisrc = open(gisrcpath, 'r')
    diz = {}
    for row in gisrc:
        key, val = row.split(':')
        diz[key.strip()] = val.strip()
    return diz

def main():
    # default option
    gisrc = read_gisrc(os.environ['GISRC'])
    user = os.environ['USER']
    # start optparse
    usage = "usage: %prog [options]"
    parser = optparse.OptionParser(usage=usage)
    parser.add_option("-U", "--user", dest="user", default=user,
                      help="PostgreSQL user [default=%default]")
    parser.add_option("-P", "--password", dest="passwd", default=None,
                      help="PostgreSQL password for user [default=%default]")
    parser.add_option("-D", "--database", dest="db", default='pygrassdb_doctest',
                      help="PostgreSQL database name [default=%default]")

    (opts, args) = parser.parse_args()
    #
    # Create DB
    #
    print("\n\nCreate a new DB: %s...\n" % opts.db)
    createdb = ['createdb', '--encoding=UTF-8', '--owner=%s' % opts.user,
                '--host=localhost', '--username=%s' % opts.user, opts.db]
    if opts.passwd:
        print(opts.passwd)
        createdb.append("--password=%s" % opts.passwd)
    else:
        createdb.append("--no-password")
    subprocess.Popen(createdb)

    #
    # set postgreSQL
    #
    print("\n\nSet Postgres connection...\n")
    grasscore.run_command('db.connect', driver='pg',
                          database='host=localhost,dbname=%s' % opts.db)

    grasscore.run_command('db.login', user=opts.user)
    print("\n\nCopy the map from PERMANENT to user1...\n")
    grasscore.run_command('g.copy',
                          vector="boundary_municp@PERMANENT,boundary_municp_pg",
                          overwrite=True)
    print("\n\nBuild topology...\n")
    grasscore.run_command('v.build', map='boundary_municp_pg', overwrite=True)


    #
    # set sqlite
    #
    db = [gisrc['GISDBASE'], gisrc['LOCATION_NAME'], gisrc['MAPSET'], 'sqlite.db']
    print("\n\nSet Sqlite connection...\n")
    grasscore.run_command('db.connect', driver='sqlite',
                          database=os.path.join(db))
    print("\n\nCopy the map from PERMANENT to user1...\n")
    grasscore.run_command('g.copy',
                          vector="boundary_municp@PERMANENT,boundary_municp_sqlite",
                          overwrite=True)
    print("\n\nBuild topology...\n")
    grasscore.run_command('v.build', map='boundary_municp_sqlite', overwrite=True)

if __name__ == "__main__":
    main()