diff --git a/gestion/gen_confs/generate.py b/gestion/gen_confs/generate.py index 387a8b67..6b2b84c0 100644 --- a/gestion/gen_confs/generate.py +++ b/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 diff --git a/surveillance/analyse.py b/surveillance/analyse.py index e6f2c641..141249dd 100644 --- a/surveillance/analyse.py +++ b/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() diff --git a/surveillance/deconnexion.py b/surveillance/deconnexion.py index 714fd49c..f872b4b0 100644 --- a/surveillance/deconnexion.py +++ b/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.autocommit = True +pgsql = psycopg2.connect(database='filtrage', user='crans') +# 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 diff --git a/surveillance/filtrage_firewall.py b/surveillance/filtrage_firewall.py index 58a3ff3d..950253f2 100644 --- a/surveillance/filtrage_firewall.py +++ b/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.autocommit = True +pgsql = psycopg2.connect(host='pgsql.adm.crans.org', database='filtrage', user='crans') +# 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() diff --git a/surveillance/filtrage_netacct.py b/surveillance/filtrage_netacct.py index 68c44e9e..b3fcc777 100644 --- a/surveillance/filtrage_netacct.py +++ b/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 : ########################################## diff --git a/surveillance/statistiques.py b/surveillance/statistiques.py index b23ab1e0..055c1c66 100644 --- a/surveillance/statistiques.py +++ b/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', - database='filtrage', user='crans') +pgsql = psycopg2.connect(database="filtrage", user="crans") curseur = pgsql.cursor() ###########################