Calendar for Penn Mobile Server (#66)

* calendar begun

* pull method finished

* range_parse functionality

* calendar.py had to be renamed due to conflicts with built-in calendar module

* adding functionality for year scraping

* year change added, code cleanup coming

* merge master

* init

* more docs and code cleanup

* tests added, summer bug fix

* silly mistake

* tests added

* docs and comments

* iCal used for scraping instead

* tests modified

* every time...

* static methods are no longer necessary

* more tests

Author: esqu1

penn/__init__.py

@@ -9,3 +9,4 @@
 from .laundry import Laundry
 from .map import Map
 from .studyspaces import StudySpaces
+from .calendar3year import Calendar

penn/calendar3year.py

@@ -0,0 +1,44 @@
+import requests
+import datetime
+
+
+BASE_URL = "https://www.google.com/calendar/ical/[email protected]/public/basic.ics"
+
+
+class Calendar(object):
+    def __init__(self):
+        pass
+
+    def pull_3year(self):
+        """Returns a list (in JSON format) containing all the events from the Penn iCal Calendar.
+
+        List contains events in chronological order.
+
+        Each element of the list is a dictionary, containing:
+            - Name of the event 'name'
+            - Start date 'start'
+            - End date 'end'
+        """
+        events = []
+        r = requests.get(BASE_URL).text
+        l = r.split("\r\n")
+        d = {}
+        for line in l:
+            if line == "BEGIN:VEVENT":
+                d = {}
+            elif line[:7] == "DTSTART":
+                raw_date = line.split(":")[1]
+                start_date = datetime.datetime.strptime(raw_date, '%Y%m%d').date()
+                d['start'] = start_date
+            elif line[:5] == "DTEND":
+                raw_date = line.split(":")[1]
+                end_date = datetime.datetime.strptime(raw_date,'%Y%m%d').date()
+                d['end'] = end_date
+            elif line[:7] == "SUMMARY":
+                name = line.split(":")[1]
+                d['name'] = name.encode('utf-8').strip()
+            elif line == "END:VEVENT":
+                events.append(d)
+
+        events.sort(key=lambda d: d['start'])
+        return events

tests/calendar_test.py

@@ -0,0 +1,39 @@
+from nose.tools import ok_
+from penn import Calendar
+import datetime
+
+class TestCalendar():
+
+    def setUp(self):
+        self.calendar = Calendar()
+
+    def test_pull(self):
+        l = self.calendar.pull_3year()
+        ok_(len(l) > 0)
+        for event in l:
+            ok_(len(event) == 3)
+
+    def test_date(self):
+        l = self.calendar.pull_3year()
+        ok_(len(l) > 0)
+        for event in l:
+            if event['name'] == "Independence Day Observed (no classes)":
+                independence = event['start']
+                ok_(isinstance(independence, datetime.date))
+                ok_(independence.month == 7)
+
+    def test_name(self):
+        l = self.calendar.pull_3year()
+        ok_(len(l) > 0)
+        for event in l:
+            start = event['start']
+            end = event['end']
+            ok_((end - start).total_seconds() >= 0)
+
+    def test_chrono(self):
+        l = self.calendar.pull_3year()
+        ok_(len(l) > 0)        
+        for i, event in enumerate(l[:-1]):            
+            start = event['start']
+            nextstart = l[i]['start']
+            ok_((nextstart - start).total_seconds() >= 0)