# -*- mode: python -*- # -*- coding: utf-8 -*- ############################################################################## # # Gestion scolarite IUT # # Copyright (c) 1999 - 2022 Emmanuel Viennet. All rights reserved. # # 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 # ############################################################################## """Fonctions sur les entreprises """ # codes anciens déplacés de ZEntreprise import datetime from operator import itemgetter import app.scodoc.sco_utils as scu import app.scodoc.notesdb as ndb from app.scodoc.notesdb import ScoDocCursor, EditableTable, DateISOtoDMY, DateDMYtoISO def _format_nom(nom): "formatte nom (filtre en entree db) d'une entreprise" if not nom: return nom return nom[0].upper() + nom[1:] class EntreprisesEditor(EditableTable): def delete(self, cnx, oid): "delete correspondants and contacts, then self" # first, delete all correspondants and contacts cursor = cnx.cursor(cursor_factory=ScoDocCursor) cursor.execute( "delete from entreprise_contact where entreprise_id=%(entreprise_id)s", {"entreprise_id": oid}, ) cursor.execute( "delete from entreprise_correspondant where entreprise_id=%(entreprise_id)s", {"entreprise_id": oid}, ) cnx.commit() EditableTable.delete(self, cnx, oid) def list( self, cnx, args={}, operator="and", test="=", sortkey=None, sort_on_contact=False, limit="", offset="", ): # list, then sort on date of last contact R = EditableTable.list( self, cnx, args=args, operator=operator, test=test, sortkey=sortkey, limit=limit, offset=offset, ) if sort_on_contact: for r in R: c = do_entreprise_contact_list( args={"entreprise_id": r["entreprise_id"]}, disable_formatting=True, ) if c: r["date"] = max([x["date"] or datetime.date.min for x in c]) else: r["date"] = datetime.date.min # sort R.sort(key=itemgetter("date")) for r in R: r["date"] = DateISOtoDMY(r["date"]) return R def list_by_etud( self, cnx, args={}, sort_on_contact=False, disable_formatting=False ): "cherche rentreprise ayant eu contact avec etudiant" cursor = cnx.cursor(cursor_factory=ScoDocCursor) cursor.execute( "select E.*, I.nom as etud_nom, I.prenom as etud_prenom, C.date from entreprises E, entreprise_contact C, identite I where C.entreprise_id = E.entreprise_id and C.etudid = I.etudid and I.nom ~* %(etud_nom)s ORDER BY E.nom", args, ) _, res = [x[0] for x in cursor.description], cursor.dictfetchall() R = [] for r in res: r["etud_prenom"] = r["etud_prenom"] or "" d = {} for key in r: v = r[key] # format value if not disable_formatting and key in self.output_formators: v = self.output_formators[key](v) d[key] = v R.append(d) # sort if sort_on_contact: R.sort(key=lambda x: (x["date"] or datetime.date.min)) for r in R: r["date"] = DateISOtoDMY(r["date"] or datetime.date.min) return R _entreprisesEditor = EntreprisesEditor( "entreprises", "entreprise_id", ( "entreprise_id", "nom", "adresse", "ville", "codepostal", "pays", "contact_origine", "secteur", "privee", "localisation", "qualite_relation", "plus10salaries", "note", "date_creation", ), filter_dept=True, sortkey="nom", input_formators={ "nom": _format_nom, "plus10salaries": bool, }, ) # ----------- Correspondants _entreprise_correspEditor = EditableTable( "entreprise_correspondant", "entreprise_corresp_id", ( "entreprise_corresp_id", "entreprise_id", "civilite", "nom", "prenom", "fonction", "phone1", "phone2", "mobile", "fax", "mail1", "mail2", "note", ), sortkey="nom", ) # ----------- Contacts _entreprise_contactEditor = EditableTable( "entreprise_contact", "entreprise_contact_id", ( "entreprise_contact_id", "date", "type_contact", "entreprise_id", "entreprise_corresp_id", "etudid", "description", "enseignant", ), sortkey="date", output_formators={"date": DateISOtoDMY}, input_formators={"date": DateDMYtoISO}, ) def do_entreprise_create(args): "entreprise_create" cnx = ndb.GetDBConnexion() r = _entreprisesEditor.create(cnx, args) return r def do_entreprise_delete(oid): "entreprise_delete" cnx = ndb.GetDBConnexion() _entreprisesEditor.delete(cnx, oid) def do_entreprise_list(**kw): "entreprise_list" cnx = ndb.GetDBConnexion() return _entreprisesEditor.list(cnx, **kw) def do_entreprise_list_by_etud(**kw): "entreprise_list_by_etud" cnx = ndb.GetDBConnexion() return _entreprisesEditor.list_by_etud(cnx, **kw) def do_entreprise_edit(*args, **kw): "entreprise_edit" cnx = ndb.GetDBConnexion() _entreprisesEditor.edit(cnx, *args, **kw) def do_entreprise_correspondant_create(args): "entreprise_correspondant_create" cnx = ndb.GetDBConnexion() r = _entreprise_correspEditor.create(cnx, args) return r def do_entreprise_correspondant_delete(oid): "entreprise_correspondant_delete" cnx = ndb.GetDBConnexion() _entreprise_correspEditor.delete(cnx, oid) def do_entreprise_correspondant_list(**kw): "entreprise_correspondant_list" cnx = ndb.GetDBConnexion() return _entreprise_correspEditor.list(cnx, **kw) def do_entreprise_correspondant_edit(*args, **kw): "entreprise_correspondant_edit" cnx = ndb.GetDBConnexion() _entreprise_correspEditor.edit(cnx, *args, **kw) def do_entreprise_correspondant_listnames(args={}): "-> liste des noms des correspondants (pour affichage menu)" C = do_entreprise_correspondant_list(args=args) return [(x["prenom"] + " " + x["nom"], str(x["entreprise_corresp_id"])) for x in C] def do_entreprise_contact_delete(oid): "entreprise_contact_delete" cnx = ndb.GetDBConnexion() _entreprise_contactEditor.delete(cnx, oid) def do_entreprise_contact_list(**kw): "entreprise_contact_list" cnx = ndb.GetDBConnexion() return _entreprise_contactEditor.list(cnx, **kw) def do_entreprise_contact_edit(*args, **kw): "entreprise_contact_edit" cnx = ndb.GetDBConnexion() _entreprise_contactEditor.edit(cnx, *args, **kw) def do_entreprise_contact_create(args): "entreprise_contact_create" cnx = ndb.GetDBConnexion() r = _entreprise_contactEditor.create(cnx, args) return r def do_entreprise_check_etudiant(etudiant): """Si etudiant est vide, ou un ETUDID valide, ou un nom unique, retourne (1, ETUDID). Sinon, retourne (0, 'message explicatif') """ etudiant = etudiant.strip().translate( str.maketrans("", "", "'()") ) # suppress parens and quote from name if not etudiant: return 1, None cnx = ndb.GetDBConnexion() cursor = cnx.cursor(cursor_factory=ScoDocCursor) cursor.execute( "select etudid, nom, prenom from identite where upper(nom) ~ upper(%(etudiant)s) or etudid=%(etudiant)s", {"etudiant": etudiant}, ) r = cursor.fetchall() if len(r) < 1: return 0, 'Aucun etudiant ne correspond à "%s"' % etudiant elif len(r) > 10: return ( 0, "%d etudiants correspondent à ce nom (utilisez le code)" % len(r), ) elif len(r) > 1: e = ['") return ( 0, "Les étudiants suivants correspondent: préciser le nom complet ou le code\n" + "\n".join(e), ) else: # une seule reponse ! return 1, r[0][0].strip()