UniSquat_Python/rooms_get.py

115 lines
2.6 KiB
Python

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Thu Feb 24 08:51:58 2022
@author: antoine
"""
################
### UniSquat ###
################
"""
Indique toutes les salles disponibles dans les différents départements de
l'Université de Strasbourg.
"""
# Modules :
import requests
import icalendar
# Variables générales :
rooms_filen = "rooms.txt"
# Fonctions :
def sched_get(date) :
"""
Récupère l'emploi du temps de toutes les salles (pour le moment, juste
de l'UFR) sur ADE depuis le site de l'Unistra.
Parameters
----------
date : datetime.datetime()
Date au format datetime.
Returns
-------
Le texte du résultat de la requête.
"""
day = str(date.day)
month = str(date.month)
year = str(date.year)
r = requests.get("https://adecons.unistra.fr/jsp/custom/modules/plannings/anonymous_cal.jsp?resources=30626&projectId=8&calType=ical&firstDate="+year+"-"+month+"-"+day+"&lastDate="+year+"-"+month+"-"+day)
return r.content
def used_rooms_get(datet) :
"""
Créé la liste des salles occupées à la date 'datet' d'après
l'emploi du temps des salles.
Parameters
----------
datet : datetime.datetime()
Date au format datetime.
Returns
-------
used_rooms : list
Liste des salles occupées. total_rooms : list
Toutes les salles mentionnées dans le fichier
"""
# Récupération des informations sur l'EDT téléchargé :
cal = icalendar.Calendar.from_ical(sched_get(datet))
used_rooms = []
total_rooms = []
for comp in cal.walk():
if comp.name == "VEVENT" :
ev_dstart = comp.decoded("dtstart")
ev_dend = comp.decoded("dtend")
roomname = str(comp.get("location"))
if not roomname in total_rooms:
total_rooms.append(roomname)
if ev_dstart.timestamp() <= datet.timestamp() and ev_dend.timestamp() > datet.timestamp() :
if not roomname in used_rooms :
used_rooms.append(roomname)
return total_rooms, used_rooms
def available_rooms_get(total_rooms, used_rooms) :
"""
Créé la liste des salles disponibles d'après la liste des salles occupées
'used_rooms'.
Parameters
----------
used_rooms : list
Liste des salles occupées.
total_rooms : list
Liste de toute les salles
Returns
-------
available_rooms : list
Liste des salles disponibles.
"""
available_rooms = []
for room in total_rooms :
if room not in used_rooms :
available_rooms.append(room)
return available_rooms