1 # -*- coding: iso-8859-1 -*- 2 """ 3 MoinMoin - EventAggregator Macro 4 5 @copyright: 2008, 2009 by Paul Boddie <paul@boddie.org.uk> 6 @copyright: 2000-2004 Juergen Hermann <jh@web.de>, 7 2005-2008 MoinMoin:ThomasWaldmann. 8 @license: GNU GPL (v2 or later), see COPYING.txt for details. 9 """ 10 11 from MoinMoin import wikiutil 12 import EventAggregatorSupport 13 import calendar 14 15 try: 16 set 17 except NameError: 18 from sets import Set as set 19 20 Dependencies = ['pages'] 21 22 # Date labels. 23 24 month_labels = ["January", "February", "March", "April", "May", "June", "July", "August", "September", "October", "November", "December"] 25 weekday_labels = ["Mon", "Tue", "Wed", "Thu", "Fri", "Sat", "Sun"] 26 27 # HTML-related functions. 28 29 def getColour(s): 30 colour = [0, 0, 0] 31 digit = 0 32 for c in s: 33 colour[digit] += ord(c) 34 colour[digit] = colour[digit] % 256 35 digit += 1 36 digit = digit % 3 37 return tuple(colour) 38 39 def getBlackOrWhite(colour): 40 if sum(colour) / 3.0 > 127: 41 return (0, 0, 0) 42 else: 43 return (255, 255, 255) 44 45 # Macro functions. 46 47 def getMonth(arg): 48 if arg.startswith("current"): 49 date = EventAggregatorSupport.getCurrentMonth() 50 if len(arg) > 8: 51 n = int(arg[7:]) 52 date = EventAggregatorSupport.monthupdate(date, n) 53 else: 54 date = EventAggregatorSupport.getMonth(arg) 55 return date 56 57 def getFormMonth(request, calendar_name, argname): 58 arg = request.form.get("%s-%s" % (calendar_name, argname), [None])[0] 59 if arg is not None: 60 return getMonth(arg) 61 else: 62 return None 63 64 def execute(macro, args): 65 66 """ 67 Execute the 'macro' with the given 'args': an optional list of selected 68 category names (categories whose pages are to be shown), together with 69 optional named arguments of the following forms: 70 71 start=YYYY-MM shows event details starting from the specified month 72 start=current-N shows event details relative to the current month 73 end=YYYY-MM shows event details ending at the specified month 74 end=current+N shows event details relative to the current month 75 76 mode=calendar shows a calendar view of events 77 mode=list shows a list of events by month 78 mode=ics provides iCalendar data for the events 79 80 names=daily shows the name of an event on every day of that event 81 names=weekly shows the name of an event once per week 82 83 calendar=NAME uses the given NAME to provide request parameters which 84 can be used to control the calendar view 85 """ 86 87 request = macro.request 88 fmt = macro.formatter 89 page = fmt.page 90 _ = request.getText 91 92 # Interpret the arguments. 93 94 try: 95 parsed_args = args and wikiutil.parse_quoted_separated(args, name_value=False) or [] 96 except AttributeError: 97 parsed_args = args.split(",") 98 99 parsed_args = [arg for arg in parsed_args if arg] 100 101 # Get special arguments. 102 103 category_names = [] 104 calendar_start = None 105 calendar_end = None 106 mode = "calendar" 107 name_usage = "daily" 108 calendar_name = None 109 110 for arg in parsed_args: 111 if arg.startswith("start="): 112 calendar_start = getMonth(arg[6:]) 113 114 elif arg.startswith("end="): 115 calendar_end = getMonth(arg[4:]) 116 117 elif arg.startswith("mode="): 118 mode = arg[5:] 119 120 elif arg.startswith("names="): 121 name_usage = arg[6:] 122 123 elif arg.startswith("calendar="): 124 calendar_name = arg[9:] 125 126 else: 127 category_names.append(arg) 128 129 # Find request parameters to override settings. 130 131 if calendar_name is not None: 132 calendar_start = getFormMonth(request, calendar_name, "start") or calendar_start 133 calendar_end = getFormMonth(request, calendar_name, "end") or calendar_end 134 135 # Get the events. 136 137 events, shown_events, all_shown_events, earliest, latest = \ 138 EventAggregatorSupport.getEvents(request, category_names, calendar_start, calendar_end) 139 140 # Get a concrete period of time. 141 142 first, last = EventAggregatorSupport.getConcretePeriod(calendar_start, calendar_end, earliest, latest) 143 144 # Define some useful navigation months. 145 146 if calendar_name is not None: 147 span = EventAggregatorSupport.span(first, last) 148 number_of_months = span[0] * 12 + span[1] + 1 149 150 previous_month_start = EventAggregatorSupport.prevmonth(first) 151 next_month_start = EventAggregatorSupport.nextmonth(first) 152 previous_month_end = EventAggregatorSupport.prevmonth(last) 153 next_month_end = EventAggregatorSupport.nextmonth(last) 154 155 previous_set_start = EventAggregatorSupport.monthupdate(first, -number_of_months) 156 next_set_start = EventAggregatorSupport.monthupdate(first, number_of_months) 157 previous_set_end = EventAggregatorSupport.monthupdate(last, -number_of_months) 158 next_set_end = EventAggregatorSupport.monthupdate(last, number_of_months) 159 160 # Make a calendar. 161 162 output = [] 163 164 # Output top-level information. 165 166 if mode == "list": 167 output.append(fmt.bullet_list(on=1, attr={"class" : "event-listings"})) 168 169 # Visit all months in the requested range, or across known events. 170 171 for year, month in EventAggregatorSupport.daterange(first, last): 172 173 # Either output a calendar view... 174 175 if mode == "calendar": 176 177 # Output a month. 178 179 output.append(fmt.table(on=1, attrs={"tableclass" : "event-month"})) 180 181 output.append(fmt.table_row(on=1)) 182 output.append(fmt.table_cell(on=1, attrs={"class" : "event-month-heading", "colspan" : "7"})) 183 184 # Either write a month heading or produce a link for navigable 185 # calendars. 186 187 month_label = _(month_labels[month - 1]) # zero-based labels 188 189 if calendar_name is not None: 190 previous_set_link = "%s-start=%04d-%02d&%s-end=%04d-%02d" % ( 191 (calendar_name,) + previous_set_start + (calendar_name,) + previous_set_end 192 ) 193 previous_month_link = "%s-start=%04d-%02d&%s-end=%04d-%02d" % ( 194 (calendar_name,) + previous_month_start + (calendar_name,) + previous_month_end 195 ) 196 197 output.append(fmt.span(on=1, css_class="previous-month")) 198 output.append(page.link_to_raw(request, wikiutil.escape("<<"), previous_set_link)) 199 output.append(fmt.text(" ")) 200 output.append(page.link_to_raw(request, wikiutil.escape("<"), previous_month_link)) 201 output.append(fmt.span(on=0)) 202 203 next_set_link = "%s-start=%04d-%02d&%s-end=%04d-%02d" % ( 204 (calendar_name,) + next_set_start + (calendar_name,) + next_set_end 205 ) 206 next_month_link = "%s-start=%04d-%02d&%s-end=%04d-%02d" % ( 207 (calendar_name,) + next_month_start + (calendar_name,) + next_month_end 208 ) 209 210 output.append(fmt.span(on=1, css_class="next-month")) 211 output.append(page.link_to_raw(request, wikiutil.escape(">"), next_month_link)) 212 output.append(fmt.text(" ")) 213 output.append(page.link_to_raw(request, wikiutil.escape(">>"), next_set_link)) 214 output.append(fmt.span(on=0)) 215 216 full_month_label = "%s %s" % (month_label, year) 217 month_link = "%s-start=%04d-%02d" % (calendar_name, year, month) 218 output.append(page.link_to_raw(request, wikiutil.escape(full_month_label), month_link)) 219 220 else: 221 output.append(fmt.span(on=1)) 222 output.append(fmt.text(month_label)) 223 output.append(fmt.span(on=0)) 224 output.append(fmt.text(" ")) 225 output.append(fmt.span(on=1)) 226 output.append(fmt.text(year)) 227 output.append(fmt.span(on=0)) 228 229 output.append(fmt.table_cell(on=0)) 230 output.append(fmt.table_row(on=0)) 231 232 # Weekday headings. 233 234 output.append(fmt.table_row(on=1)) 235 236 for weekday in range(0, 7): 237 output.append(fmt.table_cell(on=1, attrs={"class" : "event-weekday-heading"})) 238 output.append(fmt.text(_(weekday_labels[weekday]))) 239 output.append(fmt.table_cell(on=0)) 240 241 output.append(fmt.table_row(on=0)) 242 243 # Process the days of the month. 244 245 start_weekday, number_of_days = calendar.monthrange(year, month) 246 247 # The start weekday is the weekday of day number 1. 248 # Find the first day of the week, counting from below zero, if 249 # necessary, in order to land on the first day of the month as 250 # day number 1. 251 252 first_day = 1 - start_weekday 253 254 while first_day <= number_of_days: 255 256 # Find events in this week and determine how to mark them on the 257 # calendar. 258 259 week_events = [] 260 week_coverage = set() 261 262 week_start = (year, month, max(first_day, 1)) 263 week_end = (year, month, min(first_day + 6, number_of_days)) 264 265 # Get event details. 266 267 for event in shown_events.get((year, month), []): 268 event_page, event_details = event 269 270 # Test for the event in the current week. 271 272 if event_details["start"] <= week_end and event_details["end"] >= week_start: 273 274 # Find the coverage of this week for the event. 275 276 event_start = max(event_details["start"], week_start) 277 event_end = min(event_details["end"], week_end) 278 event_coverage = set(EventAggregatorSupport.daterange(event_start, event_end)) 279 280 # Update the overall coverage. 281 282 week_coverage.update(event_coverage) 283 284 # Try and fit the event into the events list. 285 286 for i, (coverage, events) in enumerate(week_events): 287 288 # Where the event does not overlap with the current 289 # element, add it alongside existing events. 290 291 if not coverage.intersection(event_coverage): 292 events.append(event) 293 week_events[i] = coverage.union(event_coverage), events 294 break 295 296 # Make a new element in the list if the event cannot be 297 # marked alongside existing events. 298 299 else: 300 week_events.append((event_coverage, [event])) 301 302 # Output a week, starting with the day numbers. 303 304 output.append(fmt.table_row(on=1)) 305 306 for weekday in range(0, 7): 307 day = first_day + weekday 308 date = (year, month, day) 309 310 # Output out-of-month days. 311 312 if day < 1 or day > number_of_days: 313 output.append(fmt.table_cell(on=1, attrs={"class" : "event-day-heading event-day-excluded"})) 314 output.append(fmt.table_cell(on=0)) 315 316 # Output normal days. 317 318 else: 319 if date in week_coverage: 320 output.append(fmt.table_cell(on=1, attrs={"class" : "event-day-heading event-day-busy"})) 321 else: 322 output.append(fmt.table_cell(on=1, attrs={"class" : "event-day-heading event-day-empty"})) 323 324 output.append(fmt.div(on=1)) 325 output.append(fmt.span(on=1, css_class="event-day-number")) 326 output.append(fmt.text(day)) 327 output.append(fmt.span(on=0)) 328 output.append(fmt.div(on=0)) 329 330 # End of day. 331 332 output.append(fmt.table_cell(on=0)) 333 334 # End of day numbers. 335 336 output.append(fmt.table_row(on=0)) 337 338 # Either generate empty days... 339 340 if not week_events: 341 output.append(fmt.table_row(on=1)) 342 343 for weekday in range(0, 7): 344 day = first_day + weekday 345 date = (year, month, day) 346 347 # Output out-of-month days. 348 349 if day < 1 or day > number_of_days: 350 output.append(fmt.table_cell(on=1, attrs={"class" : "event-day event-day-excluded"})) 351 output.append(fmt.table_cell(on=0)) 352 353 # Output empty days. 354 355 else: 356 output.append(fmt.table_cell(on=1, attrs={"class" : "event-day event-day-empty"})) 357 358 output.append(fmt.table_row(on=0)) 359 360 # Or visit each set of scheduled events... 361 362 else: 363 for coverage, events in week_events: 364 365 # Output each set. 366 367 output.append(fmt.table_row(on=1)) 368 369 # Then, output day details. 370 371 for weekday in range(0, 7): 372 day = first_day + weekday 373 date = (year, month, day) 374 375 # Skip out-of-month days. 376 377 if day < 1 or day > number_of_days: 378 output.append(fmt.table_cell(on=1, attrs={"class" : "event-day event-day-excluded"})) 379 output.append(fmt.table_cell(on=0)) 380 continue 381 382 # Output the day. 383 384 if date in coverage: 385 output.append(fmt.table_cell(on=1, attrs={"class" : "event-day event-day-busy"})) 386 else: 387 output.append(fmt.table_cell(on=1, attrs={"class" : "event-day event-day-empty"})) 388 389 # Get event details for the current day. 390 391 for event_page, event_details in events: 392 if not (event_details["start"] <= date <= event_details["end"]): 393 continue 394 395 # Get a pretty version of the page name. 396 397 pretty_pagename = EventAggregatorSupport.getPrettyPageName(event_page) 398 399 # Generate a colour for the event. 400 401 bg = getColour(event_page.page_name) 402 fg = getBlackOrWhite(bg) 403 404 css_classes = ["event-summary"] 405 406 if event_details["start"] == date: 407 css_classes.append("event-starts") 408 start_of_event = 1 409 else: 410 start_of_event = 0 411 412 if event_details["end"] == date: 413 css_classes.append("event-ends") 414 415 # Output the event. 416 417 if name_usage == "daily" or start_of_event or weekday == 0 or day == 1: 418 hide_text = 0 419 else: 420 hide_text = 1 421 422 output.append(fmt.div(on=1, css_class=(" ".join(css_classes)), 423 style=("background-color: rgb(%d, %d, %d); color: rgb(%d, %d, %d);" % (bg + fg)))) 424 425 if not hide_text: 426 output.append(event_page.link_to_raw(request, wikiutil.escape(pretty_pagename))) 427 428 output.append(fmt.div(on=0)) 429 430 # End of day. 431 432 output.append(fmt.table_cell(on=0)) 433 434 # End of set. 435 436 output.append(fmt.table_row(on=0)) 437 438 # Process the next week... 439 440 first_day += 7 441 442 # End of month. 443 444 output.append(fmt.table(on=0)) 445 446 # Or output a summary view... 447 448 elif mode == "list": 449 450 output.append(fmt.listitem(on=1, attr={"class" : "event-listings-month"})) 451 output.append(fmt.div(on=1, attr={"class" : "event-listings-month-heading"})) 452 output.append(fmt.span(on=1)) 453 output.append(fmt.text(_(month_labels[month - 1]))) # zero-based labels 454 output.append(fmt.span(on=0)) 455 output.append(fmt.text(" ")) 456 output.append(fmt.span(on=1)) 457 output.append(fmt.text(year)) 458 output.append(fmt.span(on=0)) 459 output.append(fmt.div(on=0)) 460 461 output.append(fmt.bullet_list(on=1, attr={"class" : "event-month-listings"})) 462 463 for event_page, event_details in shown_events.get((year, month), []): 464 465 # Get a pretty version of the page name. 466 467 pretty_pagename = EventAggregatorSupport.getPrettyPageName(event_page) 468 469 output.append(fmt.listitem(on=1, attr={"class" : "event-listing"})) 470 471 # Link to the page using the pretty name. 472 473 output.append(event_page.link_to_raw(request, wikiutil.escape(pretty_pagename))) 474 475 # Add the event details. 476 477 output.append(fmt.definition_list(on=1, attr={"class" : "event-details"})) 478 479 for key, value in event_details.items(): 480 output.append(fmt.definition_term(on=1)) 481 output.append(fmt.text(key)) 482 output.append(fmt.definition_term(on=0)) 483 output.append(fmt.definition_desc(on=1)) 484 output.append(fmt.text(value)) 485 output.append(fmt.definition_desc(on=0)) 486 487 output.append(fmt.definition_list(on=0)) 488 output.append(fmt.listitem(on=0)) 489 490 output.append(fmt.bullet_list(on=0)) 491 492 # Output top-level information. 493 494 # End of list view output. 495 496 if mode == "list": 497 output.append(fmt.bullet_list(on=0)) 498 499 return ''.join(output) 500 501 # vim: tabstop=4 expandtab shiftwidth=4