Package flumotion :: Package component :: Package bouncers :: Module icalbouncer
[hide private]

Source Code for Module flumotion.component.bouncers.icalbouncer

 1  # -*- Mode: Python -*- 
 2  # vi:si:et:sw=4:sts=4:ts=4 
 3  # 
 4  # Flumotion - a streaming media server 
 5  # Copyright (C) 2004,2005,2006,2007 Fluendo, S.L. (www.fluendo.com). 
 6  # All rights reserved. 
 7   
 8  # This file may be distributed and/or modified under the terms of 
 9  # the GNU General Public License version 2 as published by 
10  # the Free Software Foundation. 
11  # This file is distributed without any warranty; without even the implied 
12  # warranty of merchantability or fitness for a particular purpose. 
13  # See "LICENSE.GPL" in the source distribution for more information. 
14   
15  # Licensees having purchased or holding a valid Flumotion Advanced 
16  # Streaming Server license may use this file in accordance with the 
17  # Flumotion Advanced Streaming Server Commercial License Agreement. 
18  # See "LICENSE.Flumotion" in the source distribution for more information. 
19   
20  # Headers in this file shall remain intact. 
21   
22  """ 
23  A bouncer that only lets in during an event scheduled with an ical file. 
24  """ 
25   
26  from twisted.internet import defer 
27   
28  from flumotion.common import keycards, errors 
29  from flumotion.component.bouncers import bouncer 
30  from flumotion.common.keycards import KeycardGeneric 
31  from datetime import datetime 
32  from flumotion.component.base import scheduler 
33   
34  __all__ = ['IcalBouncer'] 
35  __version__ = "$Rev: 6995 $" 
36   
37   
38  try: 
39      # icalendar and dateutil modules needed for ical parsing 
40      from icalendar import Calendar 
41      from dateutil import rrule 
42      HAS_ICAL = True 
43  except ImportError: 
44      HAS_ICAL = False 
45   
46 -class IcalBouncer(bouncer.Bouncer):
47 48 logCategory = 'icalbouncer' 49 keycardClasses = (KeycardGeneric) 50 events = [] 51
52 - def do_setup(self):
53 if not HAS_ICAL: 54 return defer.fail( 55 errors.ConfigError( 56 "Please install icalendar and dateutil modules")) 57 props = self.config['properties'] 58 self._icsfile = props['file'] 59 self.icalScheduler = scheduler.ICalScheduler(open( 60 self._icsfile, 'r')) 61 62 return True
63
64 - def do_authenticate(self, keycard):
65 self.debug('authenticating keycard') 66 67 # need to check if inside an event time 68 # FIXME: think of a strategy for handling overlapping events 69 currentEvents = self.icalScheduler.getCurrentEvents() 70 if currentEvents: 71 event = currentEvents[0] 72 keycard.state = keycards.AUTHENTICATED 73 now = datetime.now() 74 nowInTz = datetime(now.year, now.month, now.day, 75 now.hour, now.minute, now.second, 76 tzinfo=scheduler.LOCAL) 77 end = event.currentEnd 78 duration = end - nowInTz 79 durationSecs = duration.days * 86400 + duration.seconds 80 keycard.duration = durationSecs 81 self.addKeycard(keycard) 82 self.info("authenticated login") 83 return keycard 84 self.info("failed in authentication, outside hours") 85 return None
86
87 - def do_stop(self):
88 self.icalScheduler.stopWatchingIcalFile() 89 for event in self.icalScheduler.getCurrentEvents(): 90 self.icalScheduler.removeEvent(event)
91