From a030d6371fb4f4679bb00affa3c3c8ff8b08120a Mon Sep 17 00:00:00 2001
From: Marko Kreen
Date: Mon, 16 Jul 2007 14:07:35 +0000
Subject: skytools: support for psycopg2

---
 python/skytools/dbstruct.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'python/skytools/dbstruct.py')

diff --git a/python/skytools/dbstruct.py b/python/skytools/dbstruct.py
index 22333429..ece0ea54 100644
--- a/python/skytools/dbstruct.py
+++ b/python/skytools/dbstruct.py
@@ -364,8 +364,8 @@ class TableStruct(object):
         return res
 
 def test():
-    import psycopg
-    db = psycopg.connect("dbname=fooz")
+    from skytools import connect_database
+    db = connect_database("dbname=fooz")
     curs = db.cursor()
     
     s = TableStruct(curs, "public.data1")
-- 
cgit v1.2.3