ntnoe_cancer/main.py

161 lines
4.6 KiB
Python
Raw Normal View History

2017-12-04 21:19:04 +00:00
import datetime
import httplib2
import urllib
import os
2017-12-05 19:05:36 +00:00
import requests
import shutil
2017-12-04 21:19:04 +00:00
from apiclient import discovery
from oauth2client import client
from oauth2client import tools
from oauth2client.file import Storage
import icalendar
2017-12-05 19:10:28 +00:00
TIMEDELTA_SYNCHRO = datetime.timedelta(days=7) # Number of days to look for
2017-12-04 21:19:04 +00:00
# for synchronization
with open('ntnoe_credentials') as f:
NTNOE_ID,NTNOE_PASS,_ = f.read().split('\n')
SCOPES = 'https://www.googleapis.com/auth/calendar'
CLIENT_SECRET_FILE = 'client_secret.json'
APPLICATION_NAME = 'Google Calendar API Python Quickstart'
class Event:
# ColorId corresponding to course code
EVENT_COLOR = {
'9' : '9', # Amphi
'11' : '10', # TL
'10' : '6', # TD
'13' : '5', # Autre
'12' : '3', # Exam
}
def __init__(self, e):
""" Initialize an event from a `icalendar.cal.Event`."""
self.summary = e.decoded('SUMMARY').decode('utf-8')
self.start = e.decoded('DTSTART')
self.end = e.decoded('DTEND')
self.location = e.decoded('LOCATION').decode('utf-8')
self.colorid = self.EVENT_COLOR.get(e.decoded('DESCRIPTION').decode('utf-8'), '1')
def __str__(self):
return str(self.as_google())
def as_google(self):
return {
'summary' : self.summary,
'location' : self.location,
'start' : {
'dateTime' : self.start.isoformat(),
'timeZone' : 'Europe/Paris',
},
'end' : {
'dateTime' : self.end.isoformat(),
'timeZone' : 'Europe/Paris',
},
'colorId' : self.colorid,
'reminders': {
'useDefault': False,
'overrides': [],
},
}
def get_credentials():
"""Gets valid user credentials from storage.
If nothing has been stored, or if the stored credentials are invalid,
the OAuth2 flow is completed to obtain the new credentials.
Returns:
Credentials, the obtained credential.
"""
home_dir = os.path.expanduser('~')
credential_dir = os.path.join(home_dir, '.credentials')
if not os.path.exists(credential_dir):
os.makedirs(credential_dir)
credential_path = os.path.join(credential_dir, 'ntnoe.json')
store = Storage(credential_path)
credentials = store.get()
if not credentials or credentials.invalid:
flow = client.flow_from_clientsecrets(CLIENT_SECRET_FILE, SCOPES)
flow.user_agent = APPLICATION_NAME
print('Storing credentials to ' + credential_path)
credentials = tools.run_flow(flow, store)
return credentials
def get_ntnoe():
url = "http://ntnoe.metz.supelec.fr/ical/EdTcustom/Eleves/edt_{}.ics"
url = url.format(NTNOE_ID)
2017-12-05 19:05:36 +00:00
data = {'envoyer':'Utf8_All','submit':'Générer'}
2017-12-04 21:19:04 +00:00
2017-12-05 19:05:36 +00:00
r = requests.get(
2017-12-04 21:19:04 +00:00
"https://ntnoe.metz.supelec.fr/ical/index.php",
2017-12-05 19:05:36 +00:00
data=urllib.parse.urlencode(data),
auth=(NTNOE_ID, NTNOE_PASS),
2017-12-04 21:19:04 +00:00
)
2017-12-05 19:05:36 +00:00
url = "https://ntnoe.metz.supelec.fr/ical/EdTcustom/Eleves/edt_{id}.ics".format(id=NTNOE_ID)
r = requests.get(url, auth=(NTNOE_ID, NTNOE_PASS), stream=True)
with open("edt_{}.ics".format(NTNOE_ID), 'wb') as f:
f.write(r.content)
2017-12-04 21:19:04 +00:00
def main():
"""Get the events on NTNOE the puts them on Google Calendar.
"""
get_ntnoe()
credentials = get_credentials()
http = credentials.authorize(httplib2.Http())
service = discovery.build('calendar', 'v3', http=http)
with open("edt_" + NTNOE_ID + ".ics") as f:
c = icalendar.Calendar.from_ical(f.read())
now = datetime.datetime.now()
then = now + TIMEDELTA_SYNCHRO
2017-12-05 19:05:36 +00:00
time_search = datetime.datetime(now.year, now.month, now.day, 1)
2017-12-04 21:19:04 +00:00
response = service.events().list(
calendarId='primary',
2017-12-05 19:05:36 +00:00
timeMin=time_search.isoformat()+'Z',
2017-12-04 21:19:04 +00:00
timeMax=then.isoformat()+'Z',
).execute()
existing_events = set()
for e in response['items']:
if 'summary' in e.keys():
existing_events.add(e['summary'])
for e in c.walk('VEVENT'):
event = Event(e)
t = (
event.summary,
event.start.isoformat(),
event.end.isoformat(),
event.location
)
if now >= event.end or event.start >= then:
continue
if existing_events.intersection({event.summary}):
continue
event = service.events().insert(
calendarId='primary',
body=event.as_google()
).execute()
print("Added event : {}".format(event.get('htmlLink')))
if __name__ == '__main__':
main()