added dict option to query function

"""Allow opening with a psycopg2 connection."""

try:

    import psycopg2

    import psycopg2.extras

except ImportError:

    import sys

            self.connection.set_client_encoding("UTF8")

            self.logger.debug("Successfully opened connection to database")

            self.cursor = self.connection.cursor()

            self.dict_cursor = self.connection.cursor(cursor_factory=psycopg2.extras.DictCursor)

            self.logger.debug("Successfully created a cursor")

            if isinstance(search_path, list):

                self.search_path = (",").join(search_path)

            return False

        return True

    def query(self, query):

    def query(self, query, as_dict=False):

        """Query the database.

        :param query: A Valid SQL statement to send to the database.

            self.open()  # assume that it is already open - check if it is

        self.logger.debug("Submitting user specified query to database.")

        try:

            self.cursor.execute(query)

            if self.cursor.description is not None:

                return self.cursor.fetchall()

            this_cursor = self.cursor if not as_dict else self.dict_cursor

            this_cursor.execute(query)

            if this_cursor.description is not None:

                return this_cursor.fetchall()

            return None

        except psycopg2.InterfaceError as error:

            self.logger.error(f"An unexpected InterfaceError occurred: {error}")