hd.db.connect.py

#!/usr/bin/env python3

############################################################################
#
# MODULE:       hd.db.connect
# AUTHOR(S):    Matej Krejci (matejkrejci@gmail.com
#
# COPYRIGHT:    (C) 2016 by the GRASS Development Team
#
#               This program is free software under the GNU General
#               Public License (>=v2). Read the file COPYING that
#               comes with GRASS for details.
#
#############################################################################

# %module
# % description: Connection manager for Hive and Hadoop database
# % keyword: database
# % keyword: hdfs
# %end
# %option
# % key: driver
# % type: string
# % required: no
# % description: Type of database driver
# % options: hiveserver2, hdfs, webhdfs, hive_cli
# % guisection: Connection
# %end
# %option
# % key: conn_id
# % type: string
# % required: no
# % description: Identificator of connection(free string)
# % guisection: Connection
# %end
# %option
# % key: host
# % description: host
# % type: string
# % required: no
# % guisection: Connection
# %end
# %option
# % key: port
# % type: integer
# % required: no
# % description: Port of db
# % guisection: Connection
# %end
# %option
# % key: login
# % type: string
# % required: no
# % description: Login
# % guisection: Connection
# %end
# %option
# % key: passwd
# % type: string
# % required: no
# % description: Password
# % guisection: Connection
# %end
# %option
# % key: schema
# % type: string
# % required: no
# % description: schema
# % guisection: Connection
# %end
# %option
# % key: authmechanism
# % type: string
# % required: no
# % options: PLAIN
# % description: Authentification mechanism type
# % guisection: Connection
# %end
# %option
# % key: connectionuri
# % type: string
# % required: no
# % description: connection uri string of database
# % guisection: Connection uri
# %end
# %option
# % key: rmid
# % type: integer
# % required: no
# % description: Remove connection by id
# % guisection: manager
# %end
# %flag
# % key: c
# % description: Print table of connection
# % guisection: manager
# %end
# %flag
# % key: p
# % description: Print active connection
# % guisection: manager
# %end
# %flag
# % key: r
# % description: Remove all connections
# % guisection: manager
# %end
# %flag
# % key: t
# % description: Test connection by conn_type
# % guisection: manager
# %end
# %flag
# % key: a
# % description: Set active connection by conn_id and driver
# % guisection: manager
# %end

import grass.script as gs

from hdfsgrass.hdfs_grass_lib import ConnectionManager


def main():
    # add new connection
    conn = ConnectionManager()
    if options["connectionuri"]:
        conn.set_connection_uri(options["connectionuri"])
        conn.add_connection()
        conn.test_connection()
        return

    if options["host"] and options["driver"] and options["conn_id"]:
        conn.set_connection(
            conn_type=options["driver"],
            conn_id=options["conn_id"],
            host=options["host"],
            port=options["port"],
            login=options["login"],
            password=options["passwd"],
            schema=options["schema"],
        )
        conn.add_connection()
        conn.test_connection()
        return

    if options["rmid"]:
        conn.remove_conn_Id(options["rmid"])
        return
    # print table of connection
    elif flags["c"]:
        conn.show_connections()
        return
    # drop table with connections
    elif flags["r"]:
        conn.drop_connection_table()
        conn.show_connections()
        return
    # print active connection
    elif flags["p"]:
        conn.show_active_connections()
        return
    elif flags["t"]:
        if options["driver"]:
            conn.test_connection(options["driver"])
        else:
            print("< driver > is not set")
        return
    elif flags["a"]:
        if not options["driver"] and options["conn_id"]:
            conn.set_active_connection(options["driver"], options["conn_id"])
        else:
            gs.fatal("ERROR parameter < driver > and 'conn_id' must be set")


if __name__ == "__main__":
    options, flags = gs.parser()
    main()