xenofem/python-zulip-api
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

mypy: Annotate bots/gcal-bot.

Browse source
This commit is contained in:
Heidi Dong 2016-11-30 04:05:00 +00:00 committed by Tim Abbott
parent 1ae8ac85d8
commit 836e3379c9
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
bots/gcal-bot
View file

@ -11,6 +11,7 @@ import os
sys.path.append(os.path.join(os.path.dirname(__file__), '../api')) sys.path.append(os.path.join(os.path.dirname(__file__), '../api'))
import zulip import zulip
from typing import List, Set, Tuple, Iterable, Optional
parser = optparse.OptionParser(r""" parser = optparse.OptionParser(r"""
@ -57,6 +58,7 @@ except ImportError:
parser.error('Install python-gdata') parser.error('Install python-gdata')
def get_calendar_url(): def get_calendar_url():
# type: () -> str
parts = urllib.parse.urlparse(options.calendar) parts = urllib.parse.urlparse(options.calendar)
pat = os.path.split(parts.path) pat = os.path.split(parts.path)
if pat[1] != 'basic': if pat[1] != 'basic':
@ -70,6 +72,7 @@ calendar_url = get_calendar_url()
client = zulip.init_from_options(options) client = zulip.init_from_options(options)
def get_events(): def get_events():
# type: () -> Iterable[Tuple[int, datetime.datetime, str]]
feed = CalendarClient().GetCalendarEventFeed(uri=calendar_url) feed = CalendarClient().GetCalendarEventFeed(uri=calendar_url)
for event in feed.entry: for event in feed.entry:
@ -80,12 +83,13 @@ def get_events():
yield (event.uid.value, start, event.title.text) yield (event.uid.value, start, event.title.text)
# Our cached view of the calendar, updated periodically. # Our cached view of the calendar, updated periodically.
events = [] events = [] # type: List[Tuple[int, datetime.datetime, str]]
# Unique keys for events we've already sent, so we don't remind twice. # Unique keys for events we've already sent, so we don't remind twice.
sent = set() sent = set() # type: Set[Tuple[int, datetime.datetime]]
def send_reminders(): def send_reminders():
# type: () -> Optional[None]
global sent global sent
messages = [] messages = []