--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/cubicweb/web/views/timetable.py Sat Jan 16 13:48:51 2016 +0100
@@ -0,0 +1,221 @@
+# copyright 2003-2010 LOGILAB S.A. (Paris, FRANCE), all rights reserved.
+# contact http://www.logilab.fr/ -- mailto:contact@logilab.fr
+#
+# This file is part of CubicWeb.
+#
+# CubicWeb is free software: you can redistribute it and/or modify it under the
+# terms of the GNU Lesser General Public License as published by the Free
+# Software Foundation, either version 2.1 of the License, or (at your option)
+# any later version.
+#
+# CubicWeb is distributed in the hope that it will be useful, but WITHOUT
+# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
+# FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
+# details.
+#
+# You should have received a copy of the GNU Lesser General Public License along
+# with CubicWeb. If not, see <http://www.gnu.org/licenses/>.
+"""html timetable views"""
+
+__docformat__ = "restructuredtext en"
+from cubicweb import _
+
+from six.moves import range
+
+from logilab.mtconverter import xml_escape
+from logilab.common.date import ONEDAY, date_range, todatetime
+
+from cubicweb.predicates import adaptable
+from cubicweb.view import EntityView
+
+
+class _TaskEntry(object):
+ def __init__(self, task, color, column):
+ self.task = task
+ self.color = color
+ self.column = column
+ self.lines = 1
+
+MIN_COLS = 3 # minimum number of task columns for a single user
+ALL_USERS = object()
+
+class TimeTableView(EntityView):
+ __regid__ = 'timetable'
+ title = _('timetable')
+ __select__ = adaptable('ICalendarable')
+ paginable = False
+
+ def call(self, title=None):
+ """Dumps a timetable from a resultset composed of a note (anything
+ with start/stop) and a user (anything)"""
+ self._cw.add_css('cubicweb.timetable.css')
+ dates = {}
+ users = []
+ users_max = {}
+ # XXX: try refactoring with calendar.py:OneMonthCal
+ for row in range(self.cw_rset.rowcount):
+ task = self.cw_rset.get_entity(row, 0)
+ icalendarable = task.cw_adapt_to('ICalendarable')
+ if len(self.cw_rset[row]) > 1 and self.cw_rset.description[row][1] == 'CWUser':
+ user = self.cw_rset.get_entity(row, 1)
+ else:
+ user = ALL_USERS
+ the_dates = []
+ if icalendarable.start and icalendarable.stop:
+ if icalendarable.start.toordinal() == icalendarable.stop.toordinal():
+ the_dates.append(icalendarable.start)
+ else:
+ the_dates += date_range(icalendarable.start,
+ icalendarable.stop + ONEDAY)
+ elif icalendarable.start:
+ the_dates.append(icalendarable.start)
+ elif icalendarable.stop:
+ the_dates.append(icalendarable.stop)
+ for d in the_dates:
+ d = todatetime(d)
+ d_users = dates.setdefault(d, {})
+ u_tasks = d_users.setdefault(user, set())
+ u_tasks.add( task )
+ task_max = users_max.setdefault(user, 0)
+ if len(u_tasks)>task_max:
+ users_max[user] = len(u_tasks)
+ if user not in users:
+ # keep original ordering
+ users.append(user)
+ if not dates:
+ return
+ date_min = min(dates)
+ date_max = max(dates)
+ #users = list(sorted(users, key=lambda u:u.login))
+
+ rows = []
+ # colors here are class names defined in cubicweb.css
+ colors = ["col%x" % i for i in range(12)]
+ next_color_index = 0
+
+ visited_tasks = {} # holds a description of a task for a user
+ task_colors = {} # remember a color assigned to a task
+ for date in date_range(date_min, date_max + ONEDAY):
+ columns = [date]
+ d_users = dates.get(date, {})
+ for user in users:
+ # every user has its column "splitted" in at least MIN_COLS
+ # sub-columns (for overlapping tasks)
+ user_columns = [None] * max(MIN_COLS, users_max[user])
+ # every task that is "visited" for the first time
+ # require a special treatment, so we put them in
+ # 'postpone'
+ postpone = []
+ for task in d_users.get(user, []):
+ key = (task, user)
+ if key in visited_tasks:
+ task_descr = visited_tasks[ key ]
+ user_columns[task_descr.column] = task_descr, False
+ task_descr.lines += 1
+ else:
+ postpone.append(key)
+ for key in postpone:
+ # to every 'new' task we must affect a color
+ # (which must be the same for every user concerned
+ # by the task)
+ task, user = key
+ for i, t in enumerate(user_columns):
+ if t is None:
+ if task in task_colors:
+ color = task_colors[task]
+ else:
+ color = colors[next_color_index]
+ next_color_index = (next_color_index+1)%len(colors)
+ task_colors[task] = color
+ task_descr = _TaskEntry(task, color, i)
+ user_columns[i] = task_descr, True
+ visited_tasks[key] = task_descr
+ break
+ else:
+ raise RuntimeError("is it possible we got it wrong?")
+
+ columns.append( user_columns )
+ rows.append( columns )
+
+ widths = [ len(col) for col in rows[0][1:] ]
+ self.w(u'<div class="section">')
+ if title:
+ self.w(u'<h4>%s</h4>\n' % title)
+ self.w(u'<table class="listing timetable">')
+ self.render_col_headers(users, widths)
+ self.render_rows(rows)
+ self.w(u'</table>')
+ self.w(u'</div>\n')
+
+ def render_col_headers(self, users, widths):
+ """ render column headers """
+ self.w(u'<tr class="header">\n')
+
+ self.w(u'<th class="ttdate"> </th>\n')
+ columns = []
+ for user, width in zip(users, widths):
+ self.w(u'<th colspan="%s">' % max(MIN_COLS, width))
+ if user is ALL_USERS:
+ self.w(u'*')
+ else:
+ user.view('oneline', w=self.w)
+ self.w(u'</th>')
+ self.w(u'</tr>\n')
+ return columns
+
+ def render_rows(self, rows):
+ """ render table content (row headers and central content) """
+ odd = False
+ previous_is_empty = False
+ for row in rows:
+ date = row[0]
+ empty_line = True
+ for group in row[1:]:
+ for value in group:
+ if value:
+ empty_line = False
+ break
+ else:
+ continue
+ break
+ if empty_line and previous_is_empty:
+ continue
+ previous_is_empty = False
+
+ klass = "even"
+ if date.weekday() in (5, 6) and not empty_line:
+ klass = "odd"
+ self.w(u'<tr class="%s">' % klass)
+ odd = not odd
+
+ if not empty_line:
+ self.w(u'<th class="ttdate">%s</th>' % self._cw.format_date(date) )
+ else:
+ self.w(u'<th>...</th>' )
+ previous_is_empty = True
+
+ empty_klasses = [ "ttle", "ttme", "ttre" ]
+ filled_klasses = [ "ttlf", "ttmf", "ttrf" ]
+ kj = 0 # 0: left, 1: mid, 2: right
+ for uid, group in enumerate(row[1:]):
+ for i, value in enumerate(group):
+ if i == 0:
+ kj = 0
+ elif i == len(group):
+ kj = 2
+ else:
+ kj = 1
+ if value:
+ task_descr, first_row = value
+ if first_row:
+ url = xml_escape(task_descr.task.absolute_url(vid="edition"))
+ self.w(u'<td rowspan="%d" class="%s %s" onclick="document.location=\'%s\'"> <div>' % (
+ task_descr.lines, task_descr.color, filled_klasses[kj], url))
+ task_descr.task.view('tooltip', w=self.w)
+ self.w(u'</div></td>')
+ else:
+ if empty_line:
+ self.w(u'<td class="ttempty"> </td>')
+ else:
+ self.w(u'<td class="%s"> </td>' % empty_klasses[kj] )
+ self.w(u'</tr>\n')