From bff0c9f555ac019c35290321314abb4e4dc80c70 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Pierre-Elliott=20B=C3=A9cue?= Date: Sat, 26 Jan 2013 23:22:39 +0100 Subject: [PATCH] =?UTF-8?q?Dans=20la=20s=C3=A9rie=20pgsql=20=3D>=20thot,?= =?UTF-8?q?=20on=20transf=C3=A8re=20les=20bases=20de=20donn=C3=A9es,=20pou?= =?UTF-8?q?r=20cela,=20on=20s'assure=20que=20tout=20est=20pr=C3=AAt=20nive?= =?UTF-8?q?au=20scripts.?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 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 | 9 +++++++++ surveillance/analyse.py | 4 ++-- surveillance/deconnexion.py | 7 ++++--- surveillance/filtrage_firewall.py | 8 +++++--- surveillance/filtrage_netacct.py | 5 +++-- surveillance/statistiques.py | 5 ++--- 6 files changed, 25 insertions(+), 13 deletions(-) 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() ###########################