[bde/compta] scripts pour faciliter la vérification de répartition par catégories

[scripts-20-100.git] / sum_transacs_by_categorie.py

diff --git a/sum_transacs_by_categorie.py b/sum_transacs_by_categorie.py
index d0c1e769ec3e9ac96f7de478fb679b8cf5e15873..5359a62e1d659ce1bda77a87318e31e8f18f3571 100755 (executable)
--- a/sum_transacs_by_categorie.py
+++ b/sum_transacs_by_categorie.py
@@ -9,13 +9,8 @@ Le script mange donc une liste de fichiers contenant une description par ligne.
 """
 
 import sys
-import psycopg2
-import psycopg2.extras
 
-def getcursor():
-    con = psycopg2.connect(database='note')
-    cur = con.cursor(cursor_factory=psycopg2.extras.DictCursor)
-    return con, cur
+import base
 
 def get_sum(descriptions, debut, fin):
     """
@@ -32,7 +27,7 @@ def get_sum(descriptions, debut, fin):
           AND montant != 0
           AND description = ANY (%(descriptions)s);
     """
-    con, cur = getcursor()
+    con, cur = base.getcursor()
     cur.execute(req, {"debut" : debut + ' 00:00:00', "fin" : fin + ' 00:00:00', "descriptions" : descriptions})
     s = cur.fetchone()["s"] or 0
     return s