From 7e1c763d5af5a406418c8ec85de68f525363d3ec Mon Sep 17 00:00:00 2001
From: zYne <zYne@625475ce-881a-0410-a577-b389adb331d8>
Date: Wed, 24 Jan 2007 22:12:34 +0000
Subject: [PATCH] fixed pgsql import driver

---
 lib/Doctrine/Import/Pgsql.php | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/lib/Doctrine/Import/Pgsql.php b/lib/Doctrine/Import/Pgsql.php
index df40cfa5e..30dd14a99 100644
--- a/lib/Doctrine/Import/Pgsql.php
+++ b/lib/Doctrine/Import/Pgsql.php
@@ -88,7 +88,7 @@ class Doctrine_Import_Pgsql extends Doctrine_Import
                                                    WHERE oid IN (
                                                         SELECT indexrelid
                                                         FROM pg_index, pg_class
-                                                        WHERE pg_class.relname=%s
+                                                        WHERE pg_class.relname = %s
                                                             AND pg_class.oid=pg_index.indrelid
                                                             AND indisunique != 't'
                                                             AND indisprimary != 't'
@@ -148,8 +148,10 @@ class Doctrine_Import_Pgsql extends Doctrine_Import
      */
     public function listTableColumns($table)
     {
+        $table = $this->conn->quote($table);
+        $query = sprintf($this->sql['listTableColumns'], $table);
+        $result = $this->conn->fetchAssoc($query);
 
-        $result = $this->dbh->query($this->sql['listTableColumns'])->fetchAll(PDO::FETCH_ASSOC);
         $columns     = array();
         foreach ($result as $key => $val) {
             if ($val['type'] === 'varchar') {
@@ -190,7 +192,7 @@ class Doctrine_Import_Pgsql extends Doctrine_Import
      */
     public function listTables($database = null)
     {
-        return $this->dbh->query($this->sql['listTables'])->fetchAll(PDO::FETCH_ASSOC);
+        return $this->conn->fetchAssoc($this->sql['listTables']);
     }
     /**
      * lists table triggers