[bde] Création de base.py pour factorisation de getcursor()

[scripts-20-100.git] / bde / solde_date.py

diff --git a/bde/solde_date.py b/bde/solde_date.py
index 193822ac58953c1c3c17430224708ae446b49bf4..f685c736b84527e97159dbcf95e0d57a09c79571 100755 (executable)
--- a/bde/solde_date.py
+++ b/bde/solde_date.py
@@ -10,17 +10,14 @@ Usage : solde_date.py idbde date
 """
 
 import sys
-import psycopg2,psycopg2.extras
+import psycopg2
+import psycopg2.extras
 
-def getcursor():
-    """Obtenir une connexion à la base de données."""
-    con = psycopg2.connect(database='note')
-    cur = con.cursor(cursor_factory=psycopg2.extras.DictCursor)
-    return con, cur
+import base
 
 def get_solde_date(idbde, date="now"):
     """Calcule le solde du compte ``idbde`` à la ``date`` donnée."""
-    con, cur = getcursor()
+    con, cur = base.getcursor()
     cur.execute(
         """
         SELECT COALESCE(sum(montant*quantite), 0)