[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 39dfb67569e4356564ccf42e5d3b0b44e0980d16..5359a62e1d659ce1bda77a87318e31e8f18f3571 100755 (executable)
--- a/sum_transacs_by_categorie.py
+++ b/sum_transacs_by_categorie.py
@@ -2,20 +2,15 @@
 # -*- coding:utf8 -*-
 
 """
-Récupère les gains BDE par mois et par catégorie.
+Affiche les gains BDE par mois et par catégorie.
 
 Une catégorie est définie par une liste de descriptions.
 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
@@ -68,11 +63,12 @@ def get_year_sums(dates, filename):
         sums.append(get_sum(descriptions, ddebut, dfin))
     return sums
 
-if __name__ == "__main__":
-    filenames = sys.argv[1:]
+
+def main(filenames):
+    """Fait le boulot."""
     print filenames
     y, m, d = 2015, 2, 16
-    yf, mf, df = 2016, 2, 18
+    yf, mf, df = 2016, 2, 19
     dates = get_dates(y, m, d, yf, mf, df)
     ys = {}
     for fn in filenames:
@@ -91,3 +87,7 @@ if __name__ == "__main__":
         "\t".join([str(sum(ys[fn])) for fn in filenames]),)
     print "total"
     print str(sum([sum(ys[fn]) for fn in filenames]))
+
+if __name__ == "__main__":
+    filenames = sys.argv[1:]
+    main(filenames)