Retour de get_tot_rooms pour régler le pb de parsing trop long (se fait uniquement pour le calendrier d'aujourd'hui, par du mois).
Ajout d'une constante TIMEZONE.
This commit is contained in:
parent
430e8938a2
commit
d79af7ef9b
201
rooms_get.py
201
rooms_get.py
@ -39,6 +39,9 @@ CACHE_SIZE = 10
|
||||
# Flag pour utiliser le cache :
|
||||
NO_CACHE = False
|
||||
|
||||
# Fuseau horaire :
|
||||
TIMEZONE = "Europe/Paris"
|
||||
|
||||
# Globales :
|
||||
last_cache_init = -999
|
||||
|
||||
@ -87,7 +90,7 @@ def trim(link) :
|
||||
|
||||
return result
|
||||
|
||||
def sched_get(date, link, enddate = None, nocache = False) :
|
||||
def sched_get(date, link, enddate = None, nocache = False, sortcal = False) :
|
||||
"""
|
||||
Récupère l'emploi du temps de toutes les salles sur ADE
|
||||
depuis le site de l'Unistra.
|
||||
@ -109,9 +112,12 @@ def sched_get(date, link, enddate = None, nocache = False) :
|
||||
Date de fin du calendrier à télécharger (par défaut, il s'agit de
|
||||
la date de début).
|
||||
|
||||
nocache : booléen
|
||||
nocache : bool
|
||||
Si mis à 'True', ne lit et ne modifie pas le dossier CACHE_DIR.
|
||||
|
||||
sortcal : bool
|
||||
Si 'True', trie le calendrier par ordre chronologique.
|
||||
|
||||
Returns
|
||||
-------
|
||||
str
|
||||
@ -143,12 +149,13 @@ def sched_get(date, link, enddate = None, nocache = False) :
|
||||
finallink = finallink.replace("$YEAR2$", year1)
|
||||
|
||||
if nocache :
|
||||
# Utilisation du module 'ics' pour le tri du calendrier dans l'ordre
|
||||
# chronologique :
|
||||
result = requests.get(finallink).text
|
||||
cal = ics.Calendar(result)
|
||||
cal.events = sorted(cal.events)
|
||||
result = cal.serialize()
|
||||
if sortcal :
|
||||
# Utilisation du module 'ics' pour le tri du calendrier dans l'ordre
|
||||
# chronologique :
|
||||
cal = ics.Calendar(result)
|
||||
cal.events = sorted(cal.events)
|
||||
result = cal.serialize()
|
||||
return result
|
||||
else :
|
||||
# Vérifie la TTL :
|
||||
@ -169,11 +176,12 @@ def sched_get(date, link, enddate = None, nocache = False) :
|
||||
return result
|
||||
else :
|
||||
result = requests.get(finallink).text
|
||||
# Utilisation du module 'ics' pour le tri du calendrier dans l'ordre
|
||||
# chronologique :
|
||||
cal = ics.Calendar(result)
|
||||
cal.events = sorted(cal.events)
|
||||
result = cal.serialize()
|
||||
if sortcal :
|
||||
# Utilisation du module 'ics' pour le tri du calendrier dans l'ordre
|
||||
# chronologique :
|
||||
cal = ics.Calendar(result)
|
||||
cal.events = sorted(cal.events)
|
||||
result = cal.serialize()
|
||||
with open(cachepath,'w') as f :
|
||||
f.write(result)
|
||||
return result
|
||||
@ -209,10 +217,9 @@ def get_depts(filename) :
|
||||
|
||||
return dept_list
|
||||
|
||||
def getrooms(datet, depts, ignore_list) :
|
||||
def get_tot_rooms(datet, depts, ignore_list) :
|
||||
"""
|
||||
Génération de la liste des salles avec heures de début et fin de dispo,
|
||||
et indicateur de dispo.
|
||||
Crée une liste de toutes les salles des départements choisis.
|
||||
|
||||
Parameters
|
||||
----------
|
||||
@ -220,7 +227,7 @@ def getrooms(datet, depts, ignore_list) :
|
||||
Date pour la recherche de salles.
|
||||
|
||||
depts : list
|
||||
Liste des départements dans lesquel chercher des salles.
|
||||
Liste des départements dans lesquels chercher des salles.
|
||||
|
||||
ignore_list : list
|
||||
Liste des noms de salles à ignorer.
|
||||
@ -229,14 +236,12 @@ def getrooms(datet, depts, ignore_list) :
|
||||
-------
|
||||
total_rooms : list
|
||||
Liste des salles.
|
||||
|
||||
"""
|
||||
total_rooms = list()
|
||||
|
||||
# Marge de temps (en mois) pour le début du calendrier.
|
||||
# Certaines salles ne sont pas utilisées tous les jours,
|
||||
# donc on télécharge l'EDT du reste du mois et du suivant
|
||||
# pour être tranquille.
|
||||
# donc on télécharge l'EDT pour 30 jours.
|
||||
margintime = 1
|
||||
|
||||
# Récupération du calendrier de chaque département,
|
||||
@ -261,12 +266,10 @@ def getrooms(datet, depts, ignore_list) :
|
||||
# Parcours de ces calendriers :
|
||||
dept_index = 0
|
||||
for cal in cals :
|
||||
for comp in cal.walk() : # Événements
|
||||
if comp.name == "VEVENT" :
|
||||
# Récupération des infos :
|
||||
for comp in cal.walk() : # Composants du calendrier
|
||||
if comp.name == "VEVENT" : # Événements
|
||||
# Récupération de l'emplacement :
|
||||
evtloc = str(comp.get("location"))
|
||||
evtstart = comp.decoded("dtstart")
|
||||
evtend = comp.decoded("dtend")
|
||||
|
||||
# Contient le nom de toutes les salles indiquées
|
||||
# dans la section "LOCATION" :
|
||||
@ -297,52 +300,120 @@ def getrooms(datet, depts, ignore_list) :
|
||||
# de disponibilité est aujourd'hui à 23:59.
|
||||
# - La salle est libre.
|
||||
r = Room(roomname,
|
||||
datet.replace(hour = 0, minute = 0, second = 0),
|
||||
datet.replace(hour = 23, minute = 59, second = 59),
|
||||
True,
|
||||
True,
|
||||
True,
|
||||
depts[dept_index].name
|
||||
datet.replace(hour = 0, minute = 0, second = 0),
|
||||
datet.replace(hour = 23, minute = 59, second = 59),
|
||||
True,
|
||||
True,
|
||||
True,
|
||||
depts[dept_index].name
|
||||
)
|
||||
|
||||
# Si l'événement se passe aujourd'hui :
|
||||
if evtstart.day == datet.day and \
|
||||
evtstart.month == datet.month and \
|
||||
evtstart.year == datet.year :
|
||||
# Si l'événement se passe maintenant
|
||||
# (salle occupée maintenant) :
|
||||
if evtstart.timestamp() <= datet.timestamp() and \
|
||||
evtend.timestamp() > datet.timestamp() :
|
||||
r.is_free = False
|
||||
r.nostart = False
|
||||
# L'heure de début de la prochaine dispo est
|
||||
# la fin de l'événement :
|
||||
r.start = evtend
|
||||
# Si l'événement se passe prochainement
|
||||
# (salle occupée à l'occasion de cet événement)
|
||||
# et que le début de l'événement est avant la date
|
||||
# de fin de dispo actuelle (on cherche la date la
|
||||
# plus proche de maintenant) :
|
||||
elif evtstart.timestamp() > datet.timestamp() and \
|
||||
evtstart.timestamp() < r.end.timestamp() :
|
||||
if evtstart.timestamp() == r.start.timestamp() :
|
||||
r.nostart = False
|
||||
# Dans ce cas, l'événement en cours suit
|
||||
# celui qui a défini le début de dispo
|
||||
# de la salle. Alors, c'est la fin de cet
|
||||
# événement qui marque le début de la dispo.
|
||||
r.start = evtend
|
||||
else :
|
||||
r.noend = False
|
||||
r.end = evtstart
|
||||
# Réglage du fuseau horaire :
|
||||
r.start = r.start.astimezone(pytz.timezone(TIMEZONE))
|
||||
r.end = r.end.astimezone(pytz.timezone(TIMEZONE))
|
||||
|
||||
# Réglage du fuseau horaire :
|
||||
r.start = r.start.astimezone(pytz.timezone('Europe/Paris'))
|
||||
r.end = r.end.astimezone(pytz.timezone('Europe/Paris'))
|
||||
|
||||
if not exists :
|
||||
total_rooms.append(r)
|
||||
|
||||
dept_index += 1
|
||||
|
||||
return total_rooms
|
||||
|
||||
def getrooms(datet, depts, ignore_list) :
|
||||
"""
|
||||
Ajout des heures de début et fin de dispo aux salles,
|
||||
et indicateur de dispo.
|
||||
|
||||
Parameters
|
||||
----------
|
||||
datet : datetime.datetime()
|
||||
Date pour la recherche de salles.
|
||||
|
||||
depts : list
|
||||
Liste des départements dans lesquel chercher des salles.
|
||||
|
||||
ignore_list : list
|
||||
Liste des noms de salles à ignorer.
|
||||
|
||||
Returns
|
||||
-------
|
||||
total_rooms : list
|
||||
Liste des salles.
|
||||
|
||||
"""
|
||||
total_rooms = get_tot_rooms(datet, depts, ignore_list)
|
||||
|
||||
# Récupération du calendrier de chaque département :
|
||||
cals = list() # Liste des EDT des départements choisis
|
||||
for d in depts :
|
||||
result = sched_get(datet, d.link, datet, NO_CACHE, True)
|
||||
cals.append(icalendar.Calendar.from_ical(result))
|
||||
# cals.append(icalendar.Calendar(result.decode("utf-8")))
|
||||
|
||||
# Parcours de ces calendriers :
|
||||
dept_index = 0
|
||||
for cal in cals :
|
||||
for comp in cal.walk() : # Événements
|
||||
if comp.name == "VEVENT" :
|
||||
# Récupération des infos :
|
||||
evtloc = str(comp.get("location"))
|
||||
evtstart = comp.decoded("dtstart")
|
||||
evtend = comp.decoded("dtend")
|
||||
|
||||
# Contient le nom de toutes les salles indiquées
|
||||
# dans la section "LOCATION" :
|
||||
rnamelist = list()
|
||||
|
||||
# Séparation des salles multiples, le cas échéant :
|
||||
if "," in evtloc :
|
||||
rnamelist = evtloc.split(",")
|
||||
else :
|
||||
rnamelist.append(evtloc)
|
||||
|
||||
for roomname in rnamelist :
|
||||
roomname = roomname.strip()
|
||||
if roomname not in ignore_list :
|
||||
exists = False
|
||||
for room in total_rooms :
|
||||
if room.name == roomname :
|
||||
exists = True
|
||||
r = room
|
||||
|
||||
if exists :
|
||||
# Si l'événement se passe aujourd'hui :
|
||||
if evtstart.day == datet.day and \
|
||||
evtstart.month == datet.month and \
|
||||
evtstart.year == datet.year :
|
||||
# Si l'événement se passe maintenant
|
||||
# (salle occupée maintenant) :
|
||||
if evtstart.timestamp() <= datet.timestamp() and \
|
||||
evtend.timestamp() > datet.timestamp() :
|
||||
r.is_free = False
|
||||
r.nostart = False
|
||||
# L'heure de début de la prochaine dispo est
|
||||
# la fin de l'événement :
|
||||
r.start = evtend
|
||||
# Si l'événement se passe prochainement
|
||||
# (salle occupée à l'occasion de cet événement)
|
||||
# et que le début de l'événement est avant la date
|
||||
# de fin de dispo actuelle (on cherche la date la
|
||||
# plus proche de maintenant) :
|
||||
elif evtstart.timestamp() > datet.timestamp() and \
|
||||
evtstart.timestamp() < r.end.timestamp() :
|
||||
if evtstart.timestamp() == r.start.timestamp() :
|
||||
r.nostart = False
|
||||
# Dans ce cas, l'événement en cours suit
|
||||
# celui qui a défini le début de dispo
|
||||
# de la salle. Alors, c'est la fin de cet
|
||||
# événement qui marque le début de la dispo.
|
||||
r.start = evtend
|
||||
else :
|
||||
r.noend = False
|
||||
r.end = evtstart
|
||||
|
||||
# Réglage du fuseau horaire :
|
||||
r.start = r.start.astimezone(pytz.timezone(TIMEZONE))
|
||||
r.end = r.end.astimezone(pytz.timezone(TIMEZONE))
|
||||
|
||||
dept_index += 1
|
||||
|
||||
return total_rooms
|
||||
Loading…
Reference in New Issue
Block a user