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

671 lines
23 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
#
##############################################################################
"""Photos: trombinoscopes
"""
2021-07-26 17:11:00 +02:00
import io
2020-09-26 16:19:37 +02:00
from zipfile import ZipFile, BadZipfile
2021-10-20 16:47:41 +02:00
from flask.templating import render_template
2021-02-05 22:16:30 +01:00
import reportlab
from reportlab.lib.units import cm, mm
from reportlab.platypus import Paragraph
from reportlab.platypus import Table, TableStyle
from reportlab.platypus.doctemplate import BaseDocTemplate
2021-02-05 22:16:30 +01:00
from reportlab.lib import styles
from reportlab.lib import colors
from PIL import Image as PILImage
2020-09-26 16:19:37 +02:00
2021-08-01 10:16:16 +02:00
import flask
from flask import url_for, g, send_file, request
from app import db, log
from app.models import Identite
import app.scodoc.sco_utils as scu
from app.scodoc.TrivialFormulator import TrivialFormulator
from app.scodoc.sco_exceptions import ScoValueError
from app.scodoc.sco_pdf import SU
from app.scodoc import html_sco_header
from app.scodoc import htmlutils
2021-06-21 10:17:16 +02:00
from app.scodoc import sco_import_etuds
2022-04-08 13:01:47 +02:00
from app.scodoc import sco_etud
from app.scodoc import sco_excel
from app.scodoc import sco_groups_view
from app.scodoc import sco_pdf
from app.scodoc import sco_photos
from app.scodoc import sco_portal_apogee
from app.scodoc import sco_preferences
2022-04-08 13:01:47 +02:00
from app.scodoc import sco_trombino_doc
2020-09-26 16:19:37 +02:00
def trombino(
2022-04-08 13:01:47 +02:00
group_ids=(), # liste des groupes à afficher
2020-09-26 16:19:37 +02:00
formsemestre_id=None, # utilisé si pas de groupes selectionné
etat=None,
fmt="html",
2020-09-26 16:19:37 +02:00
dialog_confirmed=False,
):
"""Trombinoscope"""
if not etat:
etat = None # may be passed as ''
# Informations sur les groupes à afficher:
groups_infos = sco_groups_view.DisplayedGroupsInfos(
group_ids, formsemestre_id=formsemestre_id, etat=etat
2020-09-26 16:19:37 +02:00
)
#
if fmt != "html" and not dialog_confirmed:
ok, dialog = check_local_photos_availability(groups_infos, fmt=fmt)
2020-09-26 16:19:37 +02:00
if not ok:
return dialog
if fmt == "zip":
2021-08-31 20:18:50 +02:00
return _trombino_zip(groups_infos)
elif fmt == "pdf":
return _trombino_pdf(groups_infos)
elif fmt == "pdflist":
return _listeappel_photos_pdf(groups_infos)
elif fmt == "doc":
2022-04-08 13:01:47 +02:00
return sco_trombino_doc.trombino_doc(groups_infos)
2020-09-26 16:19:37 +02:00
else:
raise Exception("invalid format")
def _trombino_html_header():
return html_sco_header.sco_header(javascripts=["js/trombino.js"])
2020-09-26 16:19:37 +02:00
def trombino_html(groups_infos):
2020-09-26 16:19:37 +02:00
"HTML snippet for trombino (with title and menu)"
menu_trombi = [
2021-06-14 18:08:52 +02:00
{
"title": "Charger des photos...",
"endpoint": "scolar.photos_import_files_form",
"args": {"group_ids": groups_infos.group_ids},
},
2020-09-26 16:19:37 +02:00
{
"title": "Obtenir archive Zip des photos",
2021-06-14 18:08:52 +02:00
"endpoint": "scolar.trombino",
"args": {"group_ids": groups_infos.group_ids, "fmt": "zip"},
2020-09-26 16:19:37 +02:00
},
{
"title": "Recopier les photos depuis le portail",
2021-06-14 18:08:52 +02:00
"endpoint": "scolar.trombino_copy_photos",
"args": {"group_ids": groups_infos.group_ids},
2020-09-26 16:19:37 +02:00
},
]
if groups_infos.members:
if groups_infos.tous_les_etuds_du_sem:
group_txt = "Tous les étudiants"
2020-09-26 16:19:37 +02:00
else:
group_txt = f"Groupe {groups_infos.groups_titles}"
2020-09-26 16:19:37 +02:00
else:
group_txt = "Aucun étudiant inscrit dans ce groupe !"
2020-09-26 16:19:37 +02:00
H = [
f"""<table style="padding-top: 10px; padding-bottom: 10px;">
<tr>
<td><span
style="font-style: bold; font-size: 150%%; padding-right: 20px;"
>{group_txt}</span></td>"""
2020-09-26 16:19:37 +02:00
]
if groups_infos.members:
H.append(
"<td>"
+ htmlutils.make_menu("Gérer les photos", menu_trombi, alone=True)
+ "</td>"
2020-09-26 16:19:37 +02:00
)
H.append("</tr></table>")
H.append("<div>")
i = 0
for t in groups_infos.members:
H.append(
'<span class="trombi_box"><span class="trombi-photo" id="trombi-%s">'
% t["etudid"]
)
if sco_photos.etud_photo_is_local(t["photo_filename"], size="small"):
2021-09-24 12:10:53 +02:00
foto = sco_photos.etud_photo_html(t, title="")
2020-09-26 16:19:37 +02:00
else: # la photo n'est pas immédiatement dispo
foto = f"""<span class="unloaded_img" id="{t["etudid"]
}"><img border="0" height="90" alt="en cours" src="{scu.STATIC_DIR}/icons/loading.jpg"/></span>"""
H.append(
'<a href="%s">%s</a>'
% (
url_for(
"scolar.ficheEtud", scodoc_dept=g.scodoc_dept, etudid=t["etudid"]
),
foto,
)
)
2020-09-26 16:19:37 +02:00
H.append("</span>")
H.append(
'<span class="trombi_legend"><span class="trombi_prenom">'
+ sco_etud.format_prenom(t["prenom"])
2020-09-26 16:19:37 +02:00
+ '</span><span class="trombi_nom">'
+ sco_etud.format_nom(t["nom"])
2021-11-06 17:24:11 +01:00
+ (" <i>(dem.)</i>" if t["etat"] == "D" else "")
2020-09-26 16:19:37 +02:00
)
H.append("</span></span></span>")
i += 1
H.append("</div>")
H.append(
2022-04-08 13:01:47 +02:00
f"""<div style="margin-bottom:15px;">
<a class="stdlink" href="{url_for('scolar.trombino', scodoc_dept=g.scodoc_dept,
fmt='pdf', group_ids=groups_infos.group_ids)}">Version PDF</a>
2022-04-08 13:01:47 +02:00
&nbsp;&nbsp;
<a class="stdlink" href="{url_for('scolar.trombino', scodoc_dept=g.scodoc_dept,
fmt='doc', group_ids=groups_infos.group_ids)}">Version doc</a>
2022-04-08 13:01:47 +02:00
</div>"""
2020-09-26 16:19:37 +02:00
)
return "\n".join(H)
def check_local_photos_availability(groups_infos, fmt=""):
2021-12-20 22:53:09 +01:00
"""Vérifie que toutes les photos (des groupes indiqués) sont copiées
localement dans ScoDoc (seules les photos dont nous disposons localement
peuvent être exportées en pdf ou en zip).
Si toutes ne sont pas dispo, retourne un dialogue d'avertissement
pour l'utilisateur.
2020-09-26 16:19:37 +02:00
"""
nb_missing = 0
for t in groups_infos.members:
_ = sco_photos.etud_photo_url(t) # -> copy distant files if needed
if not sco_photos.etud_photo_is_local(t["photo_filename"]):
2020-09-26 16:19:37 +02:00
nb_missing += 1
if nb_missing > 0:
parameters = {"group_ids": groups_infos.group_ids, "fmt": fmt}
2020-09-26 16:19:37 +02:00
return (
False,
scu.confirm_dialog(
f"""<p>Attention: {nb_missing} photos ne sont pas disponibles
et ne peuvent pas être exportées.</p>
<p>Vous pouvez <a class="stdlink"
href="{groups_infos.base_url}&dialog_confirmed=1&fmt={fmt}"
>exporter seulement les photos existantes</a>""",
2020-09-26 16:19:37 +02:00
dest_url="trombino",
OK="Exporter seulement les photos existantes",
2021-05-11 11:48:32 +02:00
cancel_url="groups_view?curtab=tab-photos&"
2020-09-26 16:19:37 +02:00
+ groups_infos.groups_query_args,
parameters=parameters,
),
)
return True, ""
2020-09-26 16:19:37 +02:00
2021-08-31 20:18:50 +02:00
def _trombino_zip(groups_infos):
2020-09-26 16:19:37 +02:00
"Send photos as zip archive"
2021-07-26 17:11:00 +02:00
data = io.BytesIO()
with ZipFile(data, "w") as zip_file:
# assume we have the photos (or the user acknowledged the fact)
# Archive originals (not reduced) images, in JPEG
for t in groups_infos.members:
im_path = sco_photos.photo_pathname(t["photo_filename"], size="orig")
if not im_path:
continue
img = open(im_path, "rb").read()
code_nip = t["code_nip"]
if code_nip:
filename = code_nip + ".jpg"
else:
filename = f'{t["nom"]}_{t["prenom"]}_{t["etudid"]}.jpg'
zip_file.writestr(filename, img)
2020-09-26 16:19:37 +02:00
size = data.tell()
2022-04-08 13:01:47 +02:00
log(f"trombino_zip: {size} bytes")
2021-08-31 20:18:50 +02:00
data.seek(0)
return send_file(
data,
mimetype="application/zip",
download_name="trombi.zip",
as_attachment=True,
2020-09-26 16:19:37 +02:00
)
# Copy photos from portal to ScoDoc
def trombino_copy_photos(group_ids=[], dialog_confirmed=False):
2020-09-26 16:19:37 +02:00
"Copy photos from portal to ScoDoc (overwriting local copy)"
groups_infos = sco_groups_view.DisplayedGroupsInfos(group_ids)
2021-05-11 11:48:32 +02:00
back_url = "groups_view?%s&curtab=tab-photos" % groups_infos.groups_query_args
2020-09-26 16:19:37 +02:00
2021-08-20 10:51:42 +02:00
portal_url = sco_portal_apogee.get_portal_url()
header = html_sco_header.sco_header(page_title="Chargement des photos")
footer = html_sco_header.sco_footer()
2020-09-26 16:19:37 +02:00
if not portal_url:
return f"""{ header }
<p>portail non configuré</p>
<p><a href="{back_url}" class="stdlink">Retour au trombinoscope</a></p>
{ footer }
"""
2020-09-26 16:19:37 +02:00
if not dialog_confirmed:
return scu.confirm_dialog(
f"""<h2>Copier les photos du portail vers ScoDoc ?</h2>
<p>Les photos du groupe {groups_infos.groups_titles} présentes
dans ScoDoc seront remplacées par celles du portail (si elles existent).
</p>
<p>(les photos sont normalement automatiquement copiées
lors de leur première utilisation, l'usage de cette fonction
n'est nécessaire que si les photos du portail ont été modifiées)
</p>
""",
2020-09-26 16:19:37 +02:00
dest_url="",
cancel_url=back_url,
parameters={"group_ids": group_ids},
)
msg = []
nok = 0
for etud in groups_infos.members:
path, diag = sco_photos.copy_portal_photo_to_fs(etud["etudid"])
2020-09-26 16:19:37 +02:00
msg.append(diag)
if path:
nok += 1
msg.append(f"<b>{nok} photos correctement chargées</b>")
2020-09-26 16:19:37 +02:00
return f"""{ header }
<h2>Chargement des photos depuis le portail</h2>
<ul><li>
{ '</li><li>'.join(msg) }
</li></ul>
<p><a href="{back_url}">retour au trombinoscope</a>
{ footer }
"""
2020-09-26 16:19:37 +02:00
def _get_etud_platypus_image(t, image_width=2 * cm):
2021-12-20 22:53:09 +01:00
"""Returns a platypus object for the photo of student t"""
2020-09-26 16:19:37 +02:00
try:
2021-12-20 22:53:09 +01:00
path = sco_photos.photo_pathname(t["photo_filename"], size="small")
2020-09-26 16:19:37 +02:00
if not path:
# log('> unknown')
path = sco_photos.UNKNOWN_IMAGE_PATH
im = PILImage.open(path)
w0, h0 = im.size[0], im.size[1]
if w0 > h0:
W = image_width
H = h0 * W / w0
else:
H = image_width
W = w0 * H / h0
return reportlab.platypus.Image(path, width=W, height=H)
except:
log(
"*** exception while processing photo of %s (%s) (path=%s)"
% (t["nom"], t["etudid"], path)
)
raise
def _trombino_pdf(groups_infos):
2020-09-26 16:19:37 +02:00
"Send photos as pdf page"
# Generate PDF page
filename = f"trombino_{groups_infos.groups_filename}.pdf"
2020-09-26 16:19:37 +02:00
sem = groups_infos.formsemestre # suppose 1 seul semestre
PHOTO_WIDTH = 3 * cm
COL_WIDTH = 3.6 * cm
2020-09-26 16:19:37 +02:00
N_PER_ROW = 5 # XXX should be in ScoDoc preferences
style_sheet = styles.getSampleStyleSheet()
2021-07-26 17:11:00 +02:00
report = io.BytesIO() # in-memory document, no disk file
2020-09-26 16:19:37 +02:00
objects = [
Paragraph(
SU("Trombinoscope " + sem["titreannee"] + " " + groups_infos.groups_titles),
style_sheet["Heading3"],
2020-09-26 16:19:37 +02:00
)
]
L = []
n = 0
currow = []
log(f"_trombino_pdf {len(groups_infos.members)} elements")
2020-09-26 16:19:37 +02:00
for t in groups_infos.members:
img = _get_etud_platypus_image(t, image_width=PHOTO_WIDTH)
2020-09-26 16:19:37 +02:00
elem = Table(
[
[img],
[
Paragraph(
SU(sco_etud.format_nomprenom(t)),
style_sheet["Normal"],
2020-09-26 16:19:37 +02:00
)
],
],
colWidths=[PHOTO_WIDTH],
2020-09-26 16:19:37 +02:00
)
currow.append(elem)
if n == (N_PER_ROW - 1):
L.append(currow)
currow = []
n = (n + 1) % N_PER_ROW
if currow:
currow += [" "] * (N_PER_ROW - len(currow))
L.append(currow)
if not L:
table = Paragraph(SU("Aucune photo à exporter !"), style_sheet["Normal"])
2020-09-26 16:19:37 +02:00
else:
table = Table(
L,
colWidths=[COL_WIDTH] * N_PER_ROW,
2020-09-26 16:19:37 +02:00
style=TableStyle(
[
# ('RIGHTPADDING', (0,0), (-1,-1), -5*mm),
("VALIGN", (0, 0), (-1, -1), "TOP"),
("GRID", (0, 0), (-1, -1), 0.25, colors.grey),
]
),
)
objects.append(table)
# Build document
document = BaseDocTemplate(report)
document.addPageTemplates(
sco_pdf.ScoDocPageTemplate(
2020-10-21 22:56:25 +02:00
document,
preferences=sco_preferences.SemPreferences(sem["formsemestre_id"]),
2020-09-26 16:19:37 +02:00
)
)
document.build(objects)
2021-08-31 20:18:50 +02:00
report.seek(0)
return send_file(
report,
mimetype=scu.PDF_MIMETYPE,
download_name=scu.sanitize_filename(filename),
as_attachment=True,
)
2020-09-26 16:19:37 +02:00
# --------------------- Sur une idée de l'IUT d'Orléans:
def _listeappel_photos_pdf(groups_infos):
2020-09-26 16:19:37 +02:00
"Doc pdf pour liste d'appel avec photos"
filename = f"trombino_{groups_infos.groups_filename}.pdf"
2020-09-26 16:19:37 +02:00
sem = groups_infos.formsemestre # suppose 1 seul semestre
PHOTO_WIDTH = 2 * cm
2021-02-01 23:54:46 +01:00
# COLWIDTH = 3.6 * cm
# ROWS_PER_PAGE = 26 # XXX should be in ScoDoc preferences
2020-09-26 16:19:37 +02:00
style_sheet = styles.getSampleStyleSheet()
2021-07-26 17:11:00 +02:00
report = io.BytesIO() # in-memory document, no disk file
2020-09-26 16:19:37 +02:00
objects = [
Paragraph(
SU(
f"""{sem["titreannee"]} {groups_infos.groups_titles} ({len(groups_infos.members)})"""
2020-09-26 16:19:37 +02:00
),
style_sheet["Heading3"],
2020-09-26 16:19:37 +02:00
)
]
L = []
n = 0
currow = []
log(f"_listeappel_photos_pdf {len(groups_infos.members)} elements")
2020-09-26 16:19:37 +02:00
n = len(groups_infos.members)
# npages = n / 2*ROWS_PER_PAGE + 1 # nb de pages papier
# for page in range(npages):
for i in range(n): # page*2*ROWS_PER_PAGE, (page+1)*2*ROWS_PER_PAGE):
t = groups_infos.members[i]
img = _get_etud_platypus_image(t, image_width=PHOTO_WIDTH)
2020-09-26 16:19:37 +02:00
txt = Paragraph(
SU(sco_etud.format_nomprenom(t)),
style_sheet["Normal"],
2020-09-26 16:19:37 +02:00
)
if currow:
currow += [""]
currow += [img, txt, ""]
if i % 2:
L.append(currow)
currow = []
if currow:
currow += [" "] * 3
L.append(currow)
if not L:
table = Paragraph(SU("Aucune photo à exporter !"), style_sheet["Normal"])
2020-09-26 16:19:37 +02:00
else:
table = Table(
L,
colWidths=[2 * cm, 4 * cm, 27 * mm, 5 * mm, 2 * cm, 4 * cm, 27 * mm],
style=TableStyle(
[
# ('RIGHTPADDING', (0,0), (-1,-1), -5*mm),
("VALIGN", (0, 0), (-1, -1), "TOP"),
("GRID", (0, 0), (2, -1), 0.25, colors.grey),
("GRID", (4, 0), (-1, -1), 0.25, colors.grey),
]
),
)
objects.append(table)
# Build document
document = BaseDocTemplate(report)
document.addPageTemplates(
sco_pdf.ScoDocPageTemplate(
2020-10-21 22:56:25 +02:00
document,
preferences=sco_preferences.SemPreferences(sem["formsemestre_id"]),
2020-09-26 16:19:37 +02:00
)
)
document.build(objects)
data = report.getvalue()
return scu.sendPDFFile(data, filename)
2020-09-26 16:19:37 +02:00
# --------------------- Upload des photos de tout un groupe
2022-04-08 13:01:47 +02:00
def photos_generate_excel_sample(group_ids=()):
2020-10-21 22:56:25 +02:00
"""Feuille excel pour import fichiers photos"""
2021-06-21 10:17:16 +02:00
fmt = sco_import_etuds.sco_import_format()
data = sco_import_etuds.sco_import_generate_excel_sample(
2020-09-26 16:19:37 +02:00
fmt,
group_ids=group_ids,
only_tables=["identite"],
exclude_cols=[
"date_naissance",
"lieu_naissance",
"nationalite",
"statut",
"photo_filename",
],
extra_cols=["fichier_photo"],
)
2021-09-18 10:11:46 +02:00
return scu.send_file(
data, "ImportPhotos", scu.XLSX_SUFFIX, scu.XLSX_MIMETYPE, attached=True
)
# return sco_excel.send_excel_file(data, "ImportPhotos" + scu.XLSX_SUFFIX)
2020-09-26 16:19:37 +02:00
2022-04-08 13:01:47 +02:00
def photos_import_files_form(group_ids=()):
2020-10-21 22:56:25 +02:00
"""Formulaire pour importation photos"""
2021-11-27 18:44:32 +01:00
if not group_ids:
raise ScoValueError("paramètre manquant !")
groups_infos = sco_groups_view.DisplayedGroupsInfos(group_ids)
2022-04-08 13:01:47 +02:00
back_url = f"groups_view?{groups_infos.groups_query_args}&curtab=tab-photos"
2020-09-26 16:19:37 +02:00
H = [
html_sco_header.sco_header(page_title="Import des photos des étudiants"),
2022-04-08 13:01:47 +02:00
f"""<h2 class="formsemestre">Téléchargement des photos des étudiants</h2>
<p><b>Vous pouvez aussi charger les photos individuellement via la fiche
de chaque étudiant (menu "Étudiant" / "Changer la photo").</b>
2022-04-08 13:01:47 +02:00
</p>
<p class="help">Cette page permet de charger en une seule fois les photos
de plusieurs étudiants.<br>
2022-04-08 13:01:47 +02:00
Il faut d'abord remplir une feuille excel donnant les noms
2020-09-26 16:19:37 +02:00
des fichiers images (une image par étudiant).
</p>
2022-04-08 13:01:47 +02:00
<p class="help">Ensuite, réunir vos images dans un fichier zip, puis télécharger
2020-09-26 16:19:37 +02:00
simultanément le fichier excel et le fichier zip.
</p>
<ol>
2022-04-08 13:01:47 +02:00
<li><a class="stdlink" href="photos_generate_excel_sample?{groups_infos.groups_query_args}">
2020-09-26 16:19:37 +02:00
Obtenir la feuille excel à remplir</a>
</li>
<li style="padding-top: 2em;">
2022-04-08 13:01:47 +02:00
""",
2020-09-26 16:19:37 +02:00
]
F = html_sco_header.sco_footer()
2021-09-28 09:14:04 +02:00
vals = scu.get_request_args()
vals["group_ids"] = groups_infos.group_ids
2020-09-26 16:19:37 +02:00
tf = TrivialFormulator(
request.base_url,
vals,
2020-09-26 16:19:37 +02:00
(
("xlsfile", {"title": "Fichier Excel:", "input_type": "file", "size": 40}),
("zipfile", {"title": "Fichier zip:", "input_type": "file", "size": 40}),
("group_ids", {"input_type": "hidden", "type": "list"}),
),
)
if tf[0] == 0:
return "\n".join(H) + tf[1] + "</li></ol>" + F
elif tf[0] == -1:
2021-07-31 18:01:10 +02:00
return flask.redirect(back_url)
2020-09-26 16:19:37 +02:00
else:
2021-10-20 16:47:41 +02:00
def callback(etud: Identite, data, filename):
return sco_photos.store_photo(etud, data, filename)
2021-10-20 16:47:41 +02:00
(
ignored_zipfiles,
unmatched_files,
stored_etud_filename,
) = zip_excel_import_files(
2020-09-26 16:19:37 +02:00
xlsfile=tf[2]["xlsfile"],
zipfile=tf[2]["zipfile"],
2021-10-20 16:47:41 +02:00
callback=callback,
filename_title="fichier_photo",
back_url=back_url,
2021-10-20 16:47:41 +02:00
)
return render_template(
"scolar/photos_import_files.j2",
2021-10-20 16:47:41 +02:00
page_title="Téléchargement des photos des étudiants",
ignored_zipfiles=ignored_zipfiles,
unmatched_files=unmatched_files,
stored_etud_filename=stored_etud_filename,
next_page=url_for(
"scolar.groups_view",
scodoc_dept=g.scodoc_dept,
formsemestre_id=groups_infos.formsemestre_id,
curtab="tab-photos",
),
2020-09-26 16:19:37 +02:00
)
def _norm_zip_filename(fn, lowercase=True):
"normalisation used to match filenames"
fn = fn.replace("\\", "/") # not sure if this is necessary ?
fn = fn.strip()
if lowercase:
fn = fn.lower()
fn = fn.split("/")[-1] # use only last component, not directories
return fn
2020-09-26 16:19:37 +02:00
def zip_excel_import_files(
xlsfile=None,
zipfile=None,
callback=None,
filename_title="", # doit obligatoirement etre specifié
back_url=None,
2020-09-26 16:19:37 +02:00
):
"""Importation de fichiers à partir d'un excel et d'un zip
La fonction
callback()
2021-10-20 16:47:41 +02:00
est appelée pour chaque fichier trouvé.
Fonction utilisée pour les photos et les fichiers étudiants (archives).
2020-09-26 16:19:37 +02:00
"""
# 1- build mapping etudid -> filename
exceldata = xlsfile.read()
if not exceldata:
raise ScoValueError("Fichier excel vide ou invalide")
2021-08-20 09:17:44 +02:00
_, data = sco_excel.excel_bytes_to_list(exceldata)
2021-10-20 16:47:41 +02:00
if not data:
2020-09-26 16:19:37 +02:00
raise ScoValueError("Fichier excel vide !")
# on doit avoir une colonne etudid et une colonne filename_title ('fichier_photo')
titles = data[0]
try:
etudid_idx = titles.index("etudid")
filename_idx = titles.index(filename_title)
except Exception as exc:
2020-09-26 16:19:37 +02:00
raise ScoValueError(
f"Fichier excel incorrect (il faut une colonne etudid et une colonne {filename_title}) !"
) from exc
2020-09-26 16:19:37 +02:00
filename_to_etudid = {} # filename : etudid
for line_num, l in enumerate(data[1:]):
2020-09-26 16:19:37 +02:00
filename = l[filename_idx].strip()
if filename:
try:
filename_to_etudid[_norm_zip_filename(filename)] = int(l[etudid_idx])
except ValueError as exc:
raise ScoValueError(
f"etudid invalide ({l[etudid_idx]}) sur ligne {line_num+1} !",
dest_url=back_url,
) from exc
2020-09-26 16:19:37 +02:00
# 2- Ouvre le zip et
try:
z = ZipFile(zipfile)
except BadZipfile:
2021-10-20 16:47:41 +02:00
raise ScoValueError("Fichier ZIP incorrect !") from BadZipfile
2020-09-26 16:19:37 +02:00
ignored_zipfiles = []
2021-10-20 16:47:41 +02:00
stored_etud_filename = [] # [ (etud, filename) ]
2020-09-26 16:19:37 +02:00
for name in z.namelist():
if len(name) > 4 and name[-1] != "/" and "." in name:
try:
data = z.read(name)
except BadZipfile as exc:
raise ScoValueError(
f"Fichier Zip incorrect: erreur sur {name}", dest_url=back_url
) from exc
2020-09-26 16:19:37 +02:00
# match zip filename with name given in excel
normname = _norm_zip_filename(name)
if normname in filename_to_etudid:
etudid = filename_to_etudid[normname]
2020-09-26 16:19:37 +02:00
# ok, store photo
etud: Identite = db.session.get(Identite, etudid)
if not etud:
raise ScoValueError(
f"ID étudiant invalide: {etudid}", dest_url=back_url
)
del filename_to_etudid[normname]
status, err_msg = callback(
2020-09-26 16:19:37 +02:00
etud,
data,
_norm_zip_filename(name, lowercase=False),
2020-09-26 16:19:37 +02:00
)
if not status:
raise ScoValueError(f"Erreur: {err_msg}", dest_url=back_url)
2021-10-20 16:47:41 +02:00
stored_etud_filename.append((etud, name))
2020-09-26 16:19:37 +02:00
else:
log(f"zip: zip name {name} not in excel !")
2020-09-26 16:19:37 +02:00
ignored_zipfiles.append(name)
else:
if name[-1] != "/":
ignored_zipfiles.append(name)
log(f"zip: ignoring {name}")
if filename_to_etudid:
2020-09-26 16:19:37 +02:00
# lignes excel non traitées
unmatched_files = list(filename_to_etudid.keys())
2020-09-26 16:19:37 +02:00
else:
unmatched_files = []
2021-10-20 16:47:41 +02:00
return ignored_zipfiles, unmatched_files, stored_etud_filename