[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 3e26574d5fe95fc6a92dd25ec39453331cb0cfc8..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):
     """
@@ -29,23 +24,31 @@ def get_sum(descriptions, debut, fin):
           AND date >= %(debut)s
           AND date < %(fin)s
           AND destinataire = 0
+          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
 
-def get_dates(y, m, d):
+def get_dates(y, m, d, yf, mf, df):
     """
     Donne les dates de découpage de ``d``/``m``/``y`` à ``d-1``/``m``/``y+1``.
     """
-    dates = ["%s-%s-%s" % (y, m, d)]
-    for i in range(m + 1, 13):
-        dates.append("%s-%s-01" % (y, i))
-    for i in range(1, m):
-        dates.append("%s-%s-01" % (y + 1, i))
-    dates.append("%s-%s-%s" % (y + 1, m, d - 1))
+    dates = ["%d-%02d-%02d" % (y, m, d)]
+    m += 1
+    if m == 13:
+        m = 1
+        y += 1
+    while (y, m) <= (yf, mf):
+        dates.append("%d-%02d-01" % (y, m))
+        m += 1
+        if m == 13:
+            m = 1
+            y += 1
+    # On a atteint le dernier mois
+    dates.append("%d-%02d-%02d" % (yf, mf, df))
     return dates
 
 def get_year_sums(dates, filename):
@@ -60,11 +63,13 @@ 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
-    dates = get_dates(y, m, d)
+    yf, mf, df = 2016, 2, 19
+    dates = get_dates(y, m, d, yf, mf, df)
     ys = {}
     for fn in filenames:
         ys[fn] = get_year_sums(dates, fn)
@@ -78,7 +83,11 @@ if __name__ == "__main__":
         )
 
     print "totaux"
-    print "%s-%s-%s\t %s-%s-%s\t %s" % (y, m, d, y+1, m, d-1,
+    print "%s-%s-%s\t %s-%s-%s\t %s" % (y, m, d, yf, mf, df,
         "\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)