Method get_primary_keys

web/pgadmin/tools/sqleditor/command.py:507–545

This function is used to fetch the primary key columns.

(self, default_conn=None)

        return sql

    def get_primary_keys(self, default_conn=None):
        """
        This function is used to fetch the primary key columns.
        """
        driver = get_driver(PG_DEFAULT_DRIVER)
        if default_conn is None:
            manager = driver.connection_manager(self.sid)
            conn = manager.connection(did=self.did, conn_id=self.conn_id)
        else:
            conn = default_conn

        pk_names = ''
        primary_keys = OrderedDict()

        if conn.connected():

            # Fetch the primary key column names
            query = render_template(
                "/".join([self.sql_path, 'primary_keys.sql']),
                table_name=self.object_name,
                table_nspname=self.nsp_name,
                conn=conn,
            )

            status, result = conn.execute_dict(query)
            if not status:
                raise ExecuteError(result)

            for row in result['rows']:
                pk_names += driver.qtIdent(conn, row['attname']) + ','
                primary_keys[row['attname']] = row['typname']

            if pk_names != '':
                # Remove last character from the string
                pk_names = pk_names[:-1]
        else:
            raise InternalServerError(SERVER_CONNECTION_CLOSED)

        return pk_names, primary_keys

    def can_edit(self):
        return True