1 #!/usr/bin/env python 2 3 """ 4 Construct free/busy records for a user, either recording that user's own 5 availability schedule or the schedule of another user (using details provided 6 when scheduling events with that user). 7 8 Copyright (C) 2014, 2015 Paul Boddie <paul@boddie.org.uk> 9 10 This program is free software; you can redistribute it and/or modify it under 11 the terms of the GNU General Public License as published by the Free Software 12 Foundation; either version 3 of the License, or (at your option) any later 13 version. 14 15 This program is distributed in the hope that it will be useful, but WITHOUT 16 ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS 17 FOR A PARTICULAR PURPOSE. See the GNU General Public License for more 18 details. 19 20 You should have received a copy of the GNU General Public License along with 21 this program. If not, see <http://www.gnu.org/licenses/>. 22 """ 23 24 from bisect import bisect_left 25 from codecs import getwriter 26 from imiptools.data import get_window_end, Object 27 from imiptools.dates import get_default_timezone, to_utc_datetime 28 from imiptools.period import insert_period 29 from imiptools.profile import Preferences 30 from imip_store import FileStore, FilePublisher 31 import sys 32 33 # Main program. 34 35 if __name__ == "__main__": 36 37 # Interpret the command line arguments. 38 39 try: 40 user = sys.argv[1] 41 args = sys.argv[2:] 42 participant = args and args[0] not in ("-n", "-s", "-v", "-r") and args[0] or user 43 store_and_publish = "-s" in args 44 include_needs_action = "-n" in args 45 verbose = "-v" in args 46 reset_updated_list = "-r" in args 47 except IndexError: 48 print >>sys.stderr, """\ 49 Need a user and an optional participant (if different from the user), 50 along with the -s option if updating the store and the published details. 51 """ 52 sys.exit(1) 53 54 preferences = Preferences(user) 55 tzid = preferences.get("TZID") or get_default_timezone() 56 57 # Get the size of the free/busy window. 58 59 try: 60 window_size = int(preferences.get("window_size")) 61 except (TypeError, ValueError): 62 window_size = 100 63 window_end = get_window_end(tzid, window_size) 64 65 store = FileStore() 66 publisher = FilePublisher() 67 68 # Get identifiers for uncancelled events either from a list of events 69 # providing free/busy periods at the end of the given time window, or from 70 # a list of all events. 71 72 all_events = not reset_updated_list and store.get_freebusy_providers(user, window_end) 73 74 if not all_events: 75 all_events = store.get_active_events(user) 76 fb = [] 77 78 # With providers of additional periods, append to the existing collection. 79 80 else: 81 if user == participant: 82 fb = store.get_freebusy(user) 83 else: 84 fb = store.get_freebusy_for_other(user, participant) 85 86 # Obtain event objects. 87 88 objs = [] 89 for uid, recurrenceid in all_events: 90 if verbose: 91 print >>sys.stderr, uid, recurrenceid 92 event = store.get_event(user, uid, recurrenceid) 93 if event: 94 objs.append(Object(event)) 95 96 # Build a free/busy collection for the given user. 97 98 for obj in objs: 99 partstat = obj.get_participation_status(participant) 100 recurrenceids = not obj.get_recurrenceid() and store.get_recurrences(user, obj.get_uid()) 101 102 if obj.get_participation(partstat, include_needs_action): 103 for p in obj.get_active_periods(recurrenceids, tzid, window_end): 104 fbp = obj.get_freebusy_period(p, partstat == "ORG") 105 insert_period(fb, fbp) 106 107 # Store and publish the free/busy collection. 108 109 if store_and_publish: 110 if user == participant: 111 store.set_freebusy(user, fb) 112 publisher.set_freebusy(user, fb) 113 store.set_freebusy_providers(user, to_utc_datetime(window_end, tzid), [obj for obj in objs if obj.possibly_active_from(window_end, tzid)]) 114 else: 115 store.set_freebusy_for_other(user, fb, participant) 116 else: 117 f = getwriter("utf-8")(sys.stdout) 118 for item in fb: 119 print >>f, "\t".join(item.as_tuple(strings_only=True)) 120 121 # vim: tabstop=4 expandtab shiftwidth=4