From: Randall Smith <randall@tnr.cc>
Date: Wed, 11 Feb 2009 06:29:39 +0000 (+0000)
Subject: set info_cache class
X-Git-Tag: rel_0_6_6~285
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=afb8e037e3c52106eaaa96aaf3385c2ded5dc812;p=thirdparty%2Fsqlalchemy%2Fsqlalchemy.git

set info_cache class
---

diff --git a/lib/sqlalchemy/dialects/postgres/base.py b/lib/sqlalchemy/dialects/postgres/base.py
index b4cf435058..a27af65ddd 100644
--- a/lib/sqlalchemy/dialects/postgres/base.py
+++ b/lib/sqlalchemy/dialects/postgres/base.py
@@ -432,6 +432,7 @@ class PGDialect(default.DefaultDialect):
     type_compiler = PGTypeCompiler
     preparer = PGIdentifierPreparer
     defaultrunner = PGDefaultRunner
+    info_cache = PGInfoCache
 
 
     def do_begin_twophase(self, connection, xid):
@@ -613,9 +614,9 @@ class PGDialect(default.DefaultDialect):
         else:
             current_schema = self.get_default_schema_name(connection)
         if info_cache:
-            view = info_cache.getView(viewname, current_schema)
-            if view.get('definition'):
-                return view['definition']
+            view_cache = info_cache.getView(viewname, current_schema)
+            if view_cache and 'definition' in view_cache:
+                return view_cache['definition']
         s = """
         SELECT definition FROM pg_views
         WHERE schemaname = :schemaname