import psycopg


class Connection:

    def __init__(self, name, database, location=''):
        self.id = name
        self.name = name
        self.database = database
        self.location = location
        self.db = Database(database)
        self.connection = None
        return


    def open(self):
        if self.connection == None:
            self.connection = psycopg.connect(self.db.database)
        self.connection.rollback()
        return


    def close(self):
        if self.connection != None:
            self.connection.rollback()
            self.connection.close()
        self.connection = None
        return


    def state(self):
        return int(self.connection != None)


    def commit(self):
        self.connection.commit()
        return


    def rollback(self):
        self.connection.rollback()
        return


class Database:

    def __init__(self, database):
        self.database = database
        self.dbname   = ''
        self.user     = ''
        self.password = ''
        self.host     = 'localhost'
        self.port     = '5432'
        for pair in database.split(' '):
            try:
                (key, value) = pair.split('=')
                if key == 'dbname':
                    self.dbname = value
                elif key == 'user':
                    self.user = value
                elif key == 'password':
                    self.password = value
                elif key == 'host':
                    self.host = value
                elif key == 'port':
                    self.port = value
            except Exception:
                pass
        return