Merge pull request #61 from yannickulrich/master
Adding Reminder support
This commit is contained in:
Adam Coddington
commit
e26202c93e
3 changed files with 155 additions and 1 deletions
|
|
@ -16,7 +16,8 @@ from pyicloud.services import (
|
|||
FindMyiPhoneServiceManager,
|
||||
CalendarService,
|
||||
UbiquityService,
|
||||
ContactsService
|
||||
ContactsService,
|
||||
RemindersService
|
||||
)
|
||||
|
||||
|
||||
|
|
@ -177,6 +178,11 @@ class PyiCloudService(object):
|
|||
service_root = self.webservices['contacts']['url']
|
||||
return ContactsService(service_root, self.session, self.params)
|
||||
|
||||
@property
|
||||
def reminders(self):
|
||||
service_root = self.webservices['reminders']['url']
|
||||
return RemindersService(service_root, self.session, self.params)
|
||||
|
||||
def __unicode__(self):
|
||||
return 'iCloud API: %s' % self.user.get('apple_id')
|
||||
|
||||
|
|
|
|||
|
|
@ -2,3 +2,4 @@ from pyicloud.services.calendar import CalendarService
|
|||
from pyicloud.services.findmyiphone import FindMyiPhoneServiceManager
|
||||
from pyicloud.services.ubiquity import UbiquityService
|
||||
from pyicloud.services.contacts import ContactsService
|
||||
from pyicloud.services.reminders import RemindersService
|
||||
|
|
|
|||
147
pyicloud/services/reminders.py
Normal file
147
pyicloud/services/reminders.py
Normal file
|
|
@ -0,0 +1,147 @@
|
|||
from __future__ import absolute_import
|
||||
from datetime import datetime, timedelta
|
||||
import time
|
||||
import uuid
|
||||
import pytz
|
||||
import json
|
||||
|
||||
|
||||
class RemindersService(object):
|
||||
def __init__(self, service_root, session, params):
|
||||
self.session = session
|
||||
self.params = params
|
||||
self._service_root = service_root
|
||||
self.lists = {}
|
||||
self.collections = {}
|
||||
|
||||
self.refresh()
|
||||
|
||||
def get_all_possible_timezones_of_local_machine(self):
|
||||
"""
|
||||
Return all possible timezones in Olson TZ notation
|
||||
This has been taken from
|
||||
http://stackoverflow.com/questions/7669938
|
||||
"""
|
||||
local_names = []
|
||||
if time.daylight:
|
||||
local_offset = time.altzone
|
||||
localtz = time.tzname[1]
|
||||
else:
|
||||
local_offset = time.timezone
|
||||
localtz = time.tzname[0]
|
||||
|
||||
local_offset = timedelta(seconds=-local_offset)
|
||||
|
||||
for name in pytz.all_timezones:
|
||||
timezone = pytz.timezone(name)
|
||||
if not hasattr(timezone, '_tzinfos'):
|
||||
continue
|
||||
for (utcoffset, daylight, tzname), _ in timezone._tzinfos.items():
|
||||
if utcoffset == local_offset and tzname == localtz:
|
||||
local_names.append(name)
|
||||
return local_names
|
||||
|
||||
def get_system_tz(self):
|
||||
"""
|
||||
Retrieves the system's timezone from a list of possible options.
|
||||
Just take the first one
|
||||
"""
|
||||
return self.get_all_possible_timezones_of_local_machine()[0]
|
||||
|
||||
def refresh(self):
|
||||
host = self._service_root.split('//')[1].split(':')[0]
|
||||
self.session.headers.update({'host': host})
|
||||
|
||||
params_reminders = dict(self.params)
|
||||
params_reminders.update({
|
||||
'clientVersion': '4.0',
|
||||
'lang': 'en-us',
|
||||
'usertz': self.get_system_tz()
|
||||
})
|
||||
|
||||
# Open reminders
|
||||
req = self.session.get(
|
||||
self._service_root + '/rd/startup',
|
||||
params=params_reminders
|
||||
)
|
||||
|
||||
startup = req.json()
|
||||
|
||||
self.lists = {}
|
||||
self.collections = {}
|
||||
for collection in startup['Collections']:
|
||||
temp = []
|
||||
self.collections[collection['title']] = {
|
||||
'guid': collection['guid'],
|
||||
'ctag': collection['ctag']
|
||||
}
|
||||
for reminder in startup['Reminders']:
|
||||
|
||||
if reminder['pGuid'] != collection['guid']:
|
||||
continue
|
||||
if 'dueDate' in reminder:
|
||||
if reminder['dueDate']:
|
||||
due = datetime(
|
||||
reminder['dueDate'][1],
|
||||
reminder['dueDate'][2], reminder['dueDate'][3],
|
||||
reminder['dueDate'][4], reminder['dueDate'][5]
|
||||
)
|
||||
else:
|
||||
due = None
|
||||
else:
|
||||
due = None
|
||||
if reminder['description']:
|
||||
desc = reminder['description']
|
||||
else:
|
||||
desc = ""
|
||||
temp.append({
|
||||
"title": reminder['title'],
|
||||
"desc": desc,
|
||||
"due": due
|
||||
})
|
||||
self.lists[collection['title']] = temp
|
||||
|
||||
def post(self, title, description="", collection=None):
|
||||
pguid = 'tasks'
|
||||
if collection:
|
||||
if collection in self.collections:
|
||||
pguid = self.collections[collection]['guid']
|
||||
|
||||
host = self._service_root.split('//')[1].split(':')[0]
|
||||
self.session.headers.update({'host': host})
|
||||
|
||||
params_reminders = dict(self.params)
|
||||
params_reminders.update({
|
||||
'clientVersion': '4.0',
|
||||
'lang': 'en-us',
|
||||
'usertz': self.get_system_tz()
|
||||
})
|
||||
|
||||
req = self.session.post(
|
||||
self._service_root + '/rd/reminders/tasks',
|
||||
data=json.dumps({
|
||||
"Reminders": {
|
||||
'title': title,
|
||||
"description": description,
|
||||
"pGuid": pguid,
|
||||
"etag": None,
|
||||
"order": None,
|
||||
"priority": 0,
|
||||
"recurrence": None,
|
||||
"alarms": [],
|
||||
"startDate": None,
|
||||
"startDateTz": None,
|
||||
"startDateIsAllDay": False,
|
||||
"completedDate": None,
|
||||
"dueDate": None,
|
||||
"dueDateIsAllDay": False,
|
||||
"lastModifiedDate": None,
|
||||
"createdDate": None,
|
||||
"isFamily": None,
|
||||
"createdDateExtended": int(time.time()*1000),
|
||||
"guid": str(uuid.uuid4())
|
||||
},
|
||||
"ClientState": {"Collections": self.collections.values()}
|
||||
}),
|
||||
params=params_reminders)
|
||||
return req.ok
|
||||
Loading…
Reference in a new issue