1
0
forked from ScoDoc/ScoDoc
ScoDoc/app/scodoc/sco_dump_db.py

228 lines
7.4 KiB
Python
Raw Normal View History

2020-09-26 16:19:37 +02:00
# -*- mode: python -*-
# -*- coding: utf-8 -*-
##############################################################################
#
# Gestion scolarite IUT
#
2023-01-02 13:16:27 +01:00
# Copyright (c) 1999 - 2023 Emmanuel Viennet. All rights reserved.
2020-09-26 16:19:37 +02:00
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
#
# Emmanuel Viennet emmanuel.viennet@viennet.net
#
##############################################################################
"""Dump base de données pour debug et support technique
Le principe est le suivant:
1- S'il existe une base en cours d'anonymisation, s'arrête et affiche un msg d'erreur l'utilisateur,
qui peut décider de la supprimer.
2- ScoDoc lance un script qui duplique la base (la copie de SCORT devient ANORT)
- (si elle existe deja, s'arrête)
createdb -E UTF-8 ANORT
pg_dump SCORT | psql ANORT
3- ScoDoc lance le script d'anonymisation config/anonymize_db.py qui:
- vide ou anonymise certaines colonnes
- dump cette base modifiée
- supprime cette base.
4- La copie dump anonymisé est uploadée.
"""
2022-09-10 15:23:54 +02:00
import base64
2020-09-26 16:19:37 +02:00
import fcntl
2022-09-10 15:23:54 +02:00
import os
2020-09-26 16:19:37 +02:00
import requests
2022-09-10 15:23:54 +02:00
import subprocess
import traceback
2021-02-02 14:49:49 +01:00
from flask import g, request
from flask_login import current_user
2020-09-26 16:19:37 +02:00
import app.scodoc.notesdb as ndb
import app.scodoc.sco_utils as scu
2021-08-29 19:57:32 +02:00
from app import log
from app.scodoc import sco_users
2021-08-21 17:07:44 +02:00
import sco_version
from app.scodoc.sco_exceptions import ScoValueError
2020-09-26 16:19:37 +02:00
SCO_DUMP_LOCK = "/tmp/scodump.lock"
2022-09-10 15:23:54 +02:00
def sco_dump_and_send_db(
message: str = "", request_url: str = "", traceback_str_base64: str = ""
):
2021-08-13 00:34:58 +02:00
"""Dump base de données et l'envoie anonymisée pour debug"""
2022-09-10 15:23:54 +02:00
traceback_str = base64.urlsafe_b64decode(traceback_str_base64).decode(
scu.SCO_ENCODING
)
# get current (dept) DB name:
cursor = ndb.SimpleQuery("SELECT current_database()", {})
2020-09-26 16:19:37 +02:00
db_name = cursor.fetchone()[0]
ano_db_name = "ANO" + db_name
# Lock
try:
x = open(SCO_DUMP_LOCK, "w+")
fcntl.flock(x, fcntl.LOCK_EX | fcntl.LOCK_NB)
2021-08-13 00:34:58 +02:00
except (IOError, OSError):
2020-09-26 16:19:37 +02:00
raise ScoValueError(
"Un envoi de la base "
+ db_name
+ " est déjà en cours, re-essayer plus tard"
)
try:
# Drop if exists
_drop_ano_db(ano_db_name)
# Duplicate database
_duplicate_db(db_name, ano_db_name)
# Anonymisation
_anonymize_db(ano_db_name)
# Send
2022-09-10 15:23:54 +02:00
r = _send_db(ano_db_name, message, request_url, traceback_str=traceback_str)
code = r.status_code
2020-09-26 16:19:37 +02:00
finally:
# Drop anonymized database
2021-01-01 23:46:51 +01:00
# XXX _drop_ano_db(ano_db_name)
2020-09-26 16:19:37 +02:00
# Remove lock
fcntl.flock(x, fcntl.LOCK_UN)
log("sco_dump_and_send_db: done.")
return code
2020-09-26 16:19:37 +02:00
def _duplicate_db(db_name, ano_db_name):
"""Create new database, and copy old one into"""
cmd = ["createdb", "-E", "UTF-8", ano_db_name]
log("sco_dump_and_send_db/_duplicate_db: {}".format(cmd))
try:
2021-02-04 20:02:44 +01:00
_ = subprocess.check_output(cmd)
2020-09-26 16:19:37 +02:00
except subprocess.CalledProcessError as e:
log("sco_dump_and_send_db: exception createdb {}".format(e))
raise ScoValueError(
"erreur lors de la creation de la base {}".format(ano_db_name)
)
cmd = "pg_dump {} | psql {}".format(db_name, ano_db_name)
log("sco_dump_and_send_db/_duplicate_db: {}".format(cmd))
try:
2021-02-04 20:02:44 +01:00
_ = subprocess.check_output(cmd, shell=1)
2020-09-26 16:19:37 +02:00
except subprocess.CalledProcessError as e:
log("sco_dump_and_send_db: exception {}".format(e))
raise ScoValueError(
"erreur lors de la duplication de la base {} vers {}".format(
db_name, ano_db_name
)
)
def _anonymize_db(ano_db_name):
"""Anonymize a departement database"""
cmd = os.path.join(scu.SCO_TOOLS_DIR, "anonymize_db.py")
2020-09-26 16:19:37 +02:00
log("_anonymize_db: {}".format(cmd))
try:
2021-02-04 20:02:44 +01:00
_ = subprocess.check_output([cmd, ano_db_name])
2020-09-26 16:19:37 +02:00
except subprocess.CalledProcessError as e:
log("sco_dump_and_send_db: exception in anonymisation: {}".format(e))
raise ScoValueError(
"erreur lors de l'anonymisation de la base {}".format(ano_db_name)
)
def _get_scodoc_serial():
2020-09-26 16:19:37 +02:00
try:
with open(os.path.join(scu.SCODOC_VERSION_DIR, "scodoc.sn")) as f:
return int(f.read())
2020-09-26 16:19:37 +02:00
except:
return 0
2022-09-10 15:23:54 +02:00
def _send_db(
ano_db_name: str, message: str = "", request_url: str = "", traceback_str: str = ""
):
"""Dump this (anonymized) database and send it to tech support"""
2021-09-19 16:13:57 +02:00
log(f"dumping anonymized database {ano_db_name}")
2020-09-26 16:19:37 +02:00
try:
2021-09-19 16:19:02 +02:00
dump = subprocess.check_output(
2021-09-19 16:13:57 +02:00
f"pg_dump --format=custom {ano_db_name}", shell=1
2020-09-26 16:19:37 +02:00
)
2021-09-19 16:13:57 +02:00
except subprocess.CalledProcessError as e:
log(f"sco_dump_and_send_db: exception in anonymisation: {e}")
raise ScoValueError(
f"erreur lors de l'anonymisation de la base {ano_db_name}"
) from e
2022-09-10 15:23:54 +02:00
log(f"traceback_str={traceback_str}")
2020-09-26 16:19:37 +02:00
log("uploading anonymized dump...")
2021-09-19 16:19:02 +02:00
files = {"file": (ano_db_name + ".dump", dump)}
try:
r = requests.post(
scu.SCO_DUMP_UP_URL,
files=files,
data={
"dept_name": getattr(g, "scodoc_dept", "-"),
"message": message or "",
"request_url": request_url or request.url,
"serial": _get_scodoc_serial(),
"sco_user": str(current_user),
2022-10-03 11:59:38 +02:00
"sent_by": f'"{current_user.get_nomcomplet()}" <{current_user.email}>',
"sco_version": sco_version.SCOVERSION,
"sco_fullversion": scu.get_scodoc_version(),
2022-09-10 15:23:54 +02:00
"traceback_str": traceback_str,
},
)
except requests.exceptions.ConnectionError as exc:
2022-09-10 15:23:54 +02:00
log("ConnectionError: Impossible de joindre le serveur d'assistance")
raise ScoValueError(
"""
Impossible de joindre le serveur d'assistance (scodoc.org).
Veuillez contacter le service informatique de votre établissement pour
corriger la configuration de ScoDoc. Dans la plupart des cas, il
s'agit d'un proxy mal configuré.
"""
) from exc
2020-09-26 16:19:37 +02:00
return r
def _drop_ano_db(ano_db_name):
"""drop temp database if it exists"""
existing_databases = [
s.split("|")[0].strip()
2021-08-17 08:49:19 +02:00
for s in subprocess.check_output(["psql", "-l"])
.decode(scu.SCO_ENCODING)
.split("\n")[3:]
2020-09-26 16:19:37 +02:00
]
if ano_db_name not in existing_databases:
log("_drop_ano_db: no temp db, nothing to drop")
return
cmd = ["dropdb", ano_db_name]
log("sco_dump_and_send_db: {}".format(cmd))
try:
2021-02-04 20:02:44 +01:00
_ = subprocess.check_output(cmd)
2020-09-26 16:19:37 +02:00
except subprocess.CalledProcessError as e:
log("sco_dump_and_send_db: exception dropdb {}".format(e))
raise ScoValueError(
"erreur lors de la suppression de la base {}".format(ano_db_name)
)