|
1 """html calendar views |
|
2 |
|
3 :organization: Logilab |
|
4 :copyright: 2001-2008 LOGILAB S.A. (Paris, FRANCE), all rights reserved. |
|
5 :contact: http://www.logilab.fr/ -- mailto:contact@logilab.fr |
|
6 """ |
|
7 |
|
8 from logilab.mtconverter import html_escape |
|
9 |
|
10 from cubicweb.interfaces import ITimetableViews |
|
11 from cubicweb.common.utils import date_range |
|
12 from cubicweb.common.selectors import interface_selector, anyrset_selector |
|
13 from cubicweb.common.view import AnyRsetView |
|
14 |
|
15 |
|
16 class _TaskEntry(object): |
|
17 def __init__(self, task, color, column): |
|
18 self.task = task |
|
19 self.color = color |
|
20 self.column = column |
|
21 self.lines = 1 |
|
22 |
|
23 MIN_COLS = 3 # minimum number of task columns for a single user |
|
24 |
|
25 class TimeTableView(AnyRsetView): |
|
26 id = 'timetable' |
|
27 title = _('timetable') |
|
28 __selectors__ = (interface_selector, anyrset_selector) |
|
29 accepts_interfaces = (ITimetableViews,) |
|
30 need_navigation = False |
|
31 |
|
32 def call(self, title=None): |
|
33 """Dumps a timetable from a resultset composed of a note (anything |
|
34 with start/stop) and a user (anything)""" |
|
35 self.req.add_css('cubicweb.timetable.css') |
|
36 dates = {} |
|
37 users = [] |
|
38 users_max = {} |
|
39 |
|
40 # XXX: try refactoring with calendar.py:OneMonthCal |
|
41 for row in xrange(self.rset.rowcount): |
|
42 task = self.rset.get_entity(row,0) |
|
43 if len(self.rset[row])>1: |
|
44 user = self.rset.get_entity(row,1) |
|
45 else: |
|
46 user = u"*" |
|
47 the_dates = [] |
|
48 if task.start and task.stop: |
|
49 if task.start.absdate == task.stop.absdate: |
|
50 the_dates.append(task.start) |
|
51 else: |
|
52 the_dates += date_range( task.start, task.stop ) |
|
53 elif task.start: |
|
54 the_dates.append(task.start) |
|
55 elif task.stop: |
|
56 the_dates.append(task.stop) |
|
57 for d in the_dates: |
|
58 d_users = dates.setdefault(d, {}) |
|
59 u_tasks = d_users.setdefault(user,set()) |
|
60 u_tasks.add( task ) |
|
61 task_max = users_max.setdefault(user,0) |
|
62 if len(u_tasks)>task_max: |
|
63 users_max[user] = len(u_tasks) |
|
64 if user not in users: |
|
65 # keep original ordering |
|
66 users.append(user) |
|
67 if not dates: |
|
68 return |
|
69 date_min = min(dates) |
|
70 date_max = max(dates) |
|
71 #users = list(sorted(users, key=lambda u:u.login)) |
|
72 |
|
73 rows = [] |
|
74 # colors here are class names defined in cubicweb.css |
|
75 colors = [ "col%x"%i for i in range(12) ] |
|
76 next_color_index = 0 |
|
77 |
|
78 visited_tasks = {} # holds a description of a task for a user |
|
79 task_colors = {} # remember a color assigned to a task |
|
80 for date in date_range(date_min, date_max): |
|
81 columns = [date] |
|
82 d_users = dates.get(date, {}) |
|
83 for user in users: |
|
84 # every user has its column "splitted" in at least MIN_COLS |
|
85 # sub-columns (for overlapping tasks) |
|
86 user_columns = [None] * max(MIN_COLS, users_max[user]) |
|
87 # every task that is "visited" for the first time |
|
88 # require a special treatment, so we put them in |
|
89 # 'postpone' |
|
90 postpone = [] |
|
91 for task in d_users.get(user, []): |
|
92 key = (task, user) |
|
93 if key in visited_tasks: |
|
94 task_descr = visited_tasks[ key ] |
|
95 user_columns[task_descr.column] = task_descr, False |
|
96 task_descr.lines+=1 |
|
97 else: |
|
98 postpone.append(key) |
|
99 for key in postpone: |
|
100 # to every 'new' task we must affect a color |
|
101 # (which must be the same for every user concerned |
|
102 # by the task) |
|
103 task, user = key |
|
104 for i,t in enumerate(user_columns): |
|
105 if t is None: |
|
106 if task in task_colors: |
|
107 color = task_colors[task] |
|
108 else: |
|
109 color = colors[next_color_index] |
|
110 next_color_index = (next_color_index+1)%len(colors) |
|
111 task_colors[task] = color |
|
112 task_descr = _TaskEntry(task, color, i) |
|
113 user_columns[i] = task_descr, True |
|
114 visited_tasks[key] = task_descr |
|
115 break |
|
116 else: |
|
117 raise RuntimeError("is it possible we got it wrong?") |
|
118 |
|
119 columns.append( user_columns ) |
|
120 rows.append( columns ) |
|
121 |
|
122 widths = [ len(col) for col in rows[0][1:] ] |
|
123 self.w(u'<div class="section">') |
|
124 if title: |
|
125 self.w(u'<h4>%s</h4>\n' % title) |
|
126 self.w(u'<table class="listing timetable">') |
|
127 self.render_col_headers(users, widths) |
|
128 self.render_rows(rows) |
|
129 self.w(u'</table>') |
|
130 self.w(u'</div>\n') |
|
131 |
|
132 def render_col_headers(self,users,widths): |
|
133 """ render column headers """ |
|
134 self.w(u'<tr class="header">\n') |
|
135 |
|
136 self.w(u'<th class="ttdate"> </th>\n') |
|
137 columns = [] |
|
138 for user,width in zip(users,widths): |
|
139 self.w(u'<th colspan="%s">' % max(MIN_COLS,width)) |
|
140 if user!=u"*": |
|
141 user.view('secondary',w=self.w) |
|
142 else: |
|
143 self.w(user) |
|
144 self.w(u'</th>') |
|
145 self.w(u'</tr>\n') |
|
146 return columns |
|
147 |
|
148 def render_rows(self, rows): |
|
149 """ render table content (row headers and central content) """ |
|
150 odd = False |
|
151 previous_is_empty = False |
|
152 for row in rows: |
|
153 date = row[0] |
|
154 empty_line = True |
|
155 for group in row[1:]: |
|
156 for value in group: |
|
157 if value: |
|
158 empty_line = False |
|
159 break |
|
160 else: |
|
161 continue |
|
162 break |
|
163 if empty_line and previous_is_empty: |
|
164 continue |
|
165 previous_is_empty = False |
|
166 |
|
167 klass = "even" |
|
168 if date.day_of_week in (5,6) and not empty_line: |
|
169 klass = "odd" |
|
170 self.w(u'<tr class="%s">' % klass) |
|
171 odd = not odd |
|
172 |
|
173 if not empty_line: |
|
174 self.w(u'<th class="ttdate">%s</th>' % self.format_date(date) ) |
|
175 else: |
|
176 self.w(u'<th>...</th>' ) |
|
177 previous_is_empty = True |
|
178 |
|
179 empty_klasses = [ "ttle", "ttme", "ttre" ] |
|
180 filled_klasses = [ "ttlf", "ttmf", "ttrf" ] |
|
181 kj = 0 # 0: left, 1: mid, 2: right |
|
182 for uid, group in enumerate(row[1:]): |
|
183 for i, value in enumerate(group): |
|
184 if i == 0: |
|
185 kj = 0 |
|
186 elif i == len(group): |
|
187 kj = 2 |
|
188 else: |
|
189 kj = 1 |
|
190 if value: |
|
191 task_descr, first_row = value |
|
192 if first_row: |
|
193 url = html_escape(task_descr.task.absolute_url(vid="edition")) |
|
194 self.w(u'<td rowspan="%d" class="%s %s" onclick="document.location=\'%s\'"> <div>' % ( |
|
195 task_descr.lines, task_descr.color, filled_klasses[kj], url)) |
|
196 task_descr.task.view('tooltip', w=self.w) |
|
197 self.w(u'</div></td>') |
|
198 else: |
|
199 if empty_line: |
|
200 self.w(u'<td class="ttempty"> </td>') |
|
201 else: |
|
202 self.w(u'<td class="%s"> </td>' % empty_klasses[kj] ) |
|
203 self.w(u'</tr>\n') |