Dans la série pgsql => thot, on transfère les bases de données, pour cela, on s'assure que tout est prêt niveau scripts.

Ignore-this: 1310db9d47caaf448d3727daafb43b77
Et on fait des patchs dégueux parce que psycopg est dans une trop vieille version...

darcs-hash:20130126222239-afe24-302736b502be967e4295205a6b865a9f55fb8e63.gz

gestion/gen_confs/generate.py

@@ -201,6 +201,15 @@ class pgsql(base_reconfigure):
         from gen_confs.surveillance import machines
         self._do(machines(), self._machines())
 
+class thot(base_reconfigure):
+    def surveillance_exemptions(self):
+        from gen_confs.surveillance import exemptions
+        self._do(exemptions())
+
+    def surveillance_machines(self):
+        from gen_confs.surveillance import machines
+        self._do(machines(), self._machines())
+
 class zamok(base_reconfigure):
     def del_user(self, args):
         # Suppression des fichies d'impression

surveillance/analyse.py

@@ -3,7 +3,7 @@
 
 import socket
 import sys, re
-from pyPgSQL import PgSQL
+import psycopg2
 sys.path.append('/usr/scripts/gestion/')
 from affich_tools import tableau
 
@@ -112,7 +112,7 @@ def stats(ip_crans=[], ip_ext=[],
             'sort_mini': upload_mini*1024*1024,
             'show_limit': show_limit }
 
-    pgsql = PgSQL.connect(host='/var/run/postgresql', database='filtrage', user='crans')
+    pgsql = psycopg2.connect(database="filtrage", user="crans")
     curseur = pgsql.cursor()
     curseur.execute(requete)
     results = curseur.fetchall()

surveillance/deconnexion.py

@@ -18,7 +18,7 @@ Licence : GPL v2
 
 import commands
 import sys
-from pyPgSQL import PgSQL
+import psycopg2
 sys.path.append('/usr/scripts/gestion')
 from config import upload, virus, p2p, NETs
 import smtplib
@@ -58,8 +58,9 @@ ip_src_in_crans = '(%s)' % ' or '.join([ "ip_src<<=inet('%s')" % net for net in
 ###############
 
 # Connection à la base sql via pgsql
-pgsql = PgSQL.connect(host='/var/run/postgresql', database='filtrage', user='crans')
+pgsql = psycopg2.connect(database='filtrage', user='crans')
-pgsql.autocommit = True
+# Il faudra remplacer la ligne ci-dessous par pgsql.set_session(autocommit = True) sous wheezy
+pgsql.set_isolation_level(0)
 curseur = pgsql.cursor()
 
 # Le smtp est assez capricieux

surveillance/filtrage_firewall.py

@@ -12,7 +12,7 @@ import os
 import sys
 sys.path.append('/usr/scripts/gestion')
 import iptools
-from pyPgSQL import PgSQL
+import psycopg2
 import re
 sys.path.append('/usr/script/surveillance')
 import strptime
@@ -23,8 +23,10 @@ reseau = ["138.231.136.0/21", "138.231.148.0/22"]
 
 # Ouverture de la base de données :
 ###################################
-pgsql = PgSQL.connect(host='pgsql.adm.crans.org', database='filtrage', user='crans')
+pgsql = psycopg2.connect(host='pgsql.adm.crans.org', database='filtrage', user='crans')
-pgsql.autocommit = True
+# Ancienne méthode pour faire de l'autocommit.
+# Sous wheezy, il faudra remplacer par pgsql.set_session(autocommit=True) !!!
+pgsql.set_isolation_level(0)
 curseur = pgsql.cursor()
 
 

surveillance/filtrage_netacct.py

@@ -11,7 +11,7 @@ import os
 import sys
 sys.path.append('/usr/scripts/gestion')
 import iptools
-from pyPgSQL import PgSQL
+import psycopg2
 from time import strftime, localtime
 sys.path.append('/usr/scripts/surveillance')
 import strptime, config
@@ -28,7 +28,8 @@ reseau = config.NETs['all']
 # Ouverture de la base de données :
 ###################################
 pgsql = PgSQL.connect(host='pgsql.adm.crans.org', database='filtrage', user='crans')
-pgsql.autocommit = True
+# Il faudra remplacer la ligne ci-dessous par pgsql.set_session(autocommit=True) sous wheezy
+pgsql.set_isolation_level(0)
 
 # Récupération de la table de protocoles :
 ##########################################

surveillance/statistiques.py

@@ -15,7 +15,7 @@ Licence : GPLv2
 
 import commands
 import sys, os
-from pyPgSQL import PgSQL
+import psycopg2
 sys.path.append('/usr/scripts/gestion')
 sys.path.append('/usr/scripts/surveillance')
 import config
@@ -49,8 +49,7 @@ for m in AssociationCrans().machines():
 # Ouverture des connexions : #
 ##############################
 
-pgsql = PgSQL.connect(host='/var/run/postgresql',
+pgsql = psycopg2.connect(database="filtrage", user="crans")
-    database='filtrage', user='crans')
 curseur = pgsql.cursor()
 
 ###########################