1 | """SS1 -- a spreadsheet."""
|
---|
2 |
|
---|
3 | import os
|
---|
4 | import re
|
---|
5 | import sys
|
---|
6 | import cgi
|
---|
7 | import rexec
|
---|
8 | from xml.parsers import expat
|
---|
9 |
|
---|
10 | LEFT, CENTER, RIGHT = "LEFT", "CENTER", "RIGHT"
|
---|
11 |
|
---|
12 | def ljust(x, n):
|
---|
13 | return x.ljust(n)
|
---|
14 | def center(x, n):
|
---|
15 | return x.center(n)
|
---|
16 | def rjust(x, n):
|
---|
17 | return x.rjust(n)
|
---|
18 | align2action = {LEFT: ljust, CENTER: center, RIGHT: rjust}
|
---|
19 |
|
---|
20 | align2xml = {LEFT: "left", CENTER: "center", RIGHT: "right"}
|
---|
21 | xml2align = {"left": LEFT, "center": CENTER, "right": RIGHT}
|
---|
22 |
|
---|
23 | align2anchor = {LEFT: "w", CENTER: "center", RIGHT: "e"}
|
---|
24 |
|
---|
25 | def sum(seq):
|
---|
26 | total = 0
|
---|
27 | for x in seq:
|
---|
28 | if x is not None:
|
---|
29 | total += x
|
---|
30 | return total
|
---|
31 |
|
---|
32 | class Sheet:
|
---|
33 |
|
---|
34 | def __init__(self):
|
---|
35 | self.cells = {} # {(x, y): cell, ...}
|
---|
36 | self.rexec = rexec.RExec()
|
---|
37 | m = self.rexec.add_module('__main__')
|
---|
38 | m.cell = self.cellvalue
|
---|
39 | m.cells = self.multicellvalue
|
---|
40 | m.sum = sum
|
---|
41 |
|
---|
42 | def cellvalue(self, x, y):
|
---|
43 | cell = self.getcell(x, y)
|
---|
44 | if hasattr(cell, 'recalc'):
|
---|
45 | return cell.recalc(self.rexec)
|
---|
46 | else:
|
---|
47 | return cell
|
---|
48 |
|
---|
49 | def multicellvalue(self, x1, y1, x2, y2):
|
---|
50 | if x1 > x2:
|
---|
51 | x1, x2 = x2, x1
|
---|
52 | if y1 > y2:
|
---|
53 | y1, y2 = y2, y1
|
---|
54 | seq = []
|
---|
55 | for y in range(y1, y2+1):
|
---|
56 | for x in range(x1, x2+1):
|
---|
57 | seq.append(self.cellvalue(x, y))
|
---|
58 | return seq
|
---|
59 |
|
---|
60 | def getcell(self, x, y):
|
---|
61 | return self.cells.get((x, y))
|
---|
62 |
|
---|
63 | def setcell(self, x, y, cell):
|
---|
64 | assert x > 0 and y > 0
|
---|
65 | assert isinstance(cell, BaseCell)
|
---|
66 | self.cells[x, y] = cell
|
---|
67 |
|
---|
68 | def clearcell(self, x, y):
|
---|
69 | try:
|
---|
70 | del self.cells[x, y]
|
---|
71 | except KeyError:
|
---|
72 | pass
|
---|
73 |
|
---|
74 | def clearcells(self, x1, y1, x2, y2):
|
---|
75 | for xy in self.selectcells(x1, y1, x2, y2):
|
---|
76 | del self.cells[xy]
|
---|
77 |
|
---|
78 | def clearrows(self, y1, y2):
|
---|
79 | self.clearcells(0, y1, sys.maxint, y2)
|
---|
80 |
|
---|
81 | def clearcolumns(self, x1, x2):
|
---|
82 | self.clearcells(x1, 0, x2, sys.maxint)
|
---|
83 |
|
---|
84 | def selectcells(self, x1, y1, x2, y2):
|
---|
85 | if x1 > x2:
|
---|
86 | x1, x2 = x2, x1
|
---|
87 | if y1 > y2:
|
---|
88 | y1, y2 = y2, y1
|
---|
89 | return [(x, y) for x, y in self.cells
|
---|
90 | if x1 <= x <= x2 and y1 <= y <= y2]
|
---|
91 |
|
---|
92 | def movecells(self, x1, y1, x2, y2, dx, dy):
|
---|
93 | if dx == 0 and dy == 0:
|
---|
94 | return
|
---|
95 | if x1 > x2:
|
---|
96 | x1, x2 = x2, x1
|
---|
97 | if y1 > y2:
|
---|
98 | y1, y2 = y2, y1
|
---|
99 | assert x1+dx > 0 and y1+dy > 0
|
---|
100 | new = {}
|
---|
101 | for x, y in self.cells:
|
---|
102 | cell = self.cells[x, y]
|
---|
103 | if hasattr(cell, 'renumber'):
|
---|
104 | cell = cell.renumber(x1, y1, x2, y2, dx, dy)
|
---|
105 | if x1 <= x <= x2 and y1 <= y <= y2:
|
---|
106 | x += dx
|
---|
107 | y += dy
|
---|
108 | new[x, y] = cell
|
---|
109 | self.cells = new
|
---|
110 |
|
---|
111 | def insertrows(self, y, n):
|
---|
112 | assert n > 0
|
---|
113 | self.movecells(0, y, sys.maxint, sys.maxint, 0, n)
|
---|
114 |
|
---|
115 | def deleterows(self, y1, y2):
|
---|
116 | if y1 > y2:
|
---|
117 | y1, y2 = y2, y1
|
---|
118 | self.clearrows(y1, y2)
|
---|
119 | self.movecells(0, y2+1, sys.maxint, sys.maxint, 0, y1-y2-1)
|
---|
120 |
|
---|
121 | def insertcolumns(self, x, n):
|
---|
122 | assert n > 0
|
---|
123 | self.movecells(x, 0, sys.maxint, sys.maxint, n, 0)
|
---|
124 |
|
---|
125 | def deletecolumns(self, x1, x2):
|
---|
126 | if x1 > x2:
|
---|
127 | x1, x2 = x2, x1
|
---|
128 | self.clearcells(x1, x2)
|
---|
129 | self.movecells(x2+1, 0, sys.maxint, sys.maxint, x1-x2-1, 0)
|
---|
130 |
|
---|
131 | def getsize(self):
|
---|
132 | maxx = maxy = 0
|
---|
133 | for x, y in self.cells:
|
---|
134 | maxx = max(maxx, x)
|
---|
135 | maxy = max(maxy, y)
|
---|
136 | return maxx, maxy
|
---|
137 |
|
---|
138 | def reset(self):
|
---|
139 | for cell in self.cells.itervalues():
|
---|
140 | if hasattr(cell, 'reset'):
|
---|
141 | cell.reset()
|
---|
142 |
|
---|
143 | def recalc(self):
|
---|
144 | self.reset()
|
---|
145 | for cell in self.cells.itervalues():
|
---|
146 | if hasattr(cell, 'recalc'):
|
---|
147 | cell.recalc(self.rexec)
|
---|
148 |
|
---|
149 | def display(self):
|
---|
150 | maxx, maxy = self.getsize()
|
---|
151 | width, height = maxx+1, maxy+1
|
---|
152 | colwidth = [1] * width
|
---|
153 | full = {}
|
---|
154 | # Add column heading labels in row 0
|
---|
155 | for x in range(1, width):
|
---|
156 | full[x, 0] = text, alignment = colnum2name(x), RIGHT
|
---|
157 | colwidth[x] = max(colwidth[x], len(text))
|
---|
158 | # Add row labels in column 0
|
---|
159 | for y in range(1, height):
|
---|
160 | full[0, y] = text, alignment = str(y), RIGHT
|
---|
161 | colwidth[0] = max(colwidth[0], len(text))
|
---|
162 | # Add sheet cells in columns with x>0 and y>0
|
---|
163 | for (x, y), cell in self.cells.iteritems():
|
---|
164 | if x <= 0 or y <= 0:
|
---|
165 | continue
|
---|
166 | if hasattr(cell, 'recalc'):
|
---|
167 | cell.recalc(self.rexec)
|
---|
168 | if hasattr(cell, 'format'):
|
---|
169 | text, alignment = cell.format()
|
---|
170 | assert isinstance(text, str)
|
---|
171 | assert alignment in (LEFT, CENTER, RIGHT)
|
---|
172 | else:
|
---|
173 | text = str(cell)
|
---|
174 | if isinstance(cell, str):
|
---|
175 | alignment = LEFT
|
---|
176 | else:
|
---|
177 | alignment = RIGHT
|
---|
178 | full[x, y] = (text, alignment)
|
---|
179 | colwidth[x] = max(colwidth[x], len(text))
|
---|
180 | # Calculate the horizontal separator line (dashes and dots)
|
---|
181 | sep = ""
|
---|
182 | for x in range(width):
|
---|
183 | if sep:
|
---|
184 | sep += "+"
|
---|
185 | sep += "-"*colwidth[x]
|
---|
186 | # Now print The full grid
|
---|
187 | for y in range(height):
|
---|
188 | line = ""
|
---|
189 | for x in range(width):
|
---|
190 | text, alignment = full.get((x, y)) or ("", LEFT)
|
---|
191 | text = align2action[alignment](text, colwidth[x])
|
---|
192 | if line:
|
---|
193 | line += '|'
|
---|
194 | line += text
|
---|
195 | print line
|
---|
196 | if y == 0:
|
---|
197 | print sep
|
---|
198 |
|
---|
199 | def xml(self):
|
---|
200 | out = ['<spreadsheet>']
|
---|
201 | for (x, y), cell in self.cells.iteritems():
|
---|
202 | if hasattr(cell, 'xml'):
|
---|
203 | cellxml = cell.xml()
|
---|
204 | else:
|
---|
205 | cellxml = '<value>%s</value>' % cgi.escape(cell)
|
---|
206 | out.append('<cell row="%s" col="%s">\n %s\n</cell>' %
|
---|
207 | (y, x, cellxml))
|
---|
208 | out.append('</spreadsheet>')
|
---|
209 | return '\n'.join(out)
|
---|
210 |
|
---|
211 | def save(self, filename):
|
---|
212 | text = self.xml()
|
---|
213 | f = open(filename, "w")
|
---|
214 | f.write(text)
|
---|
215 | if text and not text.endswith('\n'):
|
---|
216 | f.write('\n')
|
---|
217 | f.close()
|
---|
218 |
|
---|
219 | def load(self, filename):
|
---|
220 | f = open(filename, 'r')
|
---|
221 | SheetParser(self).parsefile(f)
|
---|
222 | f.close()
|
---|
223 |
|
---|
224 | class SheetParser:
|
---|
225 |
|
---|
226 | def __init__(self, sheet):
|
---|
227 | self.sheet = sheet
|
---|
228 |
|
---|
229 | def parsefile(self, f):
|
---|
230 | parser = expat.ParserCreate()
|
---|
231 | parser.StartElementHandler = self.startelement
|
---|
232 | parser.EndElementHandler = self.endelement
|
---|
233 | parser.CharacterDataHandler = self.data
|
---|
234 | parser.ParseFile(f)
|
---|
235 |
|
---|
236 | def startelement(self, tag, attrs):
|
---|
237 | method = getattr(self, 'start_'+tag, None)
|
---|
238 | if method:
|
---|
239 | for key, value in attrs.iteritems():
|
---|
240 | attrs[key] = str(value) # XXX Convert Unicode to 8-bit
|
---|
241 | method(attrs)
|
---|
242 | self.texts = []
|
---|
243 |
|
---|
244 | def data(self, text):
|
---|
245 | text = str(text) # XXX Convert Unicode to 8-bit
|
---|
246 | self.texts.append(text)
|
---|
247 |
|
---|
248 | def endelement(self, tag):
|
---|
249 | method = getattr(self, 'end_'+tag, None)
|
---|
250 | if method:
|
---|
251 | method("".join(self.texts))
|
---|
252 |
|
---|
253 | def start_cell(self, attrs):
|
---|
254 | self.y = int(attrs.get("row"))
|
---|
255 | self.x = int(attrs.get("col"))
|
---|
256 |
|
---|
257 | def start_value(self, attrs):
|
---|
258 | self.fmt = attrs.get('format')
|
---|
259 | self.alignment = xml2align.get(attrs.get('align'))
|
---|
260 |
|
---|
261 | start_formula = start_value
|
---|
262 |
|
---|
263 | def end_int(self, text):
|
---|
264 | try:
|
---|
265 | self.value = int(text)
|
---|
266 | except:
|
---|
267 | self.value = None
|
---|
268 |
|
---|
269 | def end_long(self, text):
|
---|
270 | try:
|
---|
271 | self.value = long(text)
|
---|
272 | except:
|
---|
273 | self.value = None
|
---|
274 |
|
---|
275 | def end_double(self, text):
|
---|
276 | try:
|
---|
277 | self.value = float(text)
|
---|
278 | except:
|
---|
279 | self.value = None
|
---|
280 |
|
---|
281 | def end_complex(self, text):
|
---|
282 | try:
|
---|
283 | self.value = complex(text)
|
---|
284 | except:
|
---|
285 | self.value = None
|
---|
286 |
|
---|
287 | def end_string(self, text):
|
---|
288 | try:
|
---|
289 | self.value = text
|
---|
290 | except:
|
---|
291 | self.value = None
|
---|
292 |
|
---|
293 | def end_value(self, text):
|
---|
294 | if isinstance(self.value, BaseCell):
|
---|
295 | self.cell = self.value
|
---|
296 | elif isinstance(self.value, str):
|
---|
297 | self.cell = StringCell(self.value,
|
---|
298 | self.fmt or "%s",
|
---|
299 | self.alignment or LEFT)
|
---|
300 | else:
|
---|
301 | self.cell = NumericCell(self.value,
|
---|
302 | self.fmt or "%s",
|
---|
303 | self.alignment or RIGHT)
|
---|
304 |
|
---|
305 | def end_formula(self, text):
|
---|
306 | self.cell = FormulaCell(text,
|
---|
307 | self.fmt or "%s",
|
---|
308 | self.alignment or RIGHT)
|
---|
309 |
|
---|
310 | def end_cell(self, text):
|
---|
311 | self.sheet.setcell(self.x, self.y, self.cell)
|
---|
312 |
|
---|
313 | class BaseCell:
|
---|
314 | __init__ = None # Must provide
|
---|
315 | """Abstract base class for sheet cells.
|
---|
316 |
|
---|
317 | Subclasses may but needn't provide the following APIs:
|
---|
318 |
|
---|
319 | cell.reset() -- prepare for recalculation
|
---|
320 | cell.recalc(rexec) -> value -- recalculate formula
|
---|
321 | cell.format() -> (value, alignment) -- return formatted value
|
---|
322 | cell.xml() -> string -- return XML
|
---|
323 | """
|
---|
324 |
|
---|
325 | class NumericCell(BaseCell):
|
---|
326 |
|
---|
327 | def __init__(self, value, fmt="%s", alignment=RIGHT):
|
---|
328 | assert isinstance(value, (int, long, float, complex))
|
---|
329 | assert alignment in (LEFT, CENTER, RIGHT)
|
---|
330 | self.value = value
|
---|
331 | self.fmt = fmt
|
---|
332 | self.alignment = alignment
|
---|
333 |
|
---|
334 | def recalc(self, rexec):
|
---|
335 | return self.value
|
---|
336 |
|
---|
337 | def format(self):
|
---|
338 | try:
|
---|
339 | text = self.fmt % self.value
|
---|
340 | except:
|
---|
341 | text = str(self.value)
|
---|
342 | return text, self.alignment
|
---|
343 |
|
---|
344 | def xml(self):
|
---|
345 | method = getattr(self, '_xml_' + type(self.value).__name__)
|
---|
346 | return '<value align="%s" format="%s">%s</value>' % (
|
---|
347 | align2xml[self.alignment],
|
---|
348 | self.fmt,
|
---|
349 | method())
|
---|
350 |
|
---|
351 | def _xml_int(self):
|
---|
352 | if -2**31 <= self.value < 2**31:
|
---|
353 | return '<int>%s</int>' % self.value
|
---|
354 | else:
|
---|
355 | return self._xml_long()
|
---|
356 |
|
---|
357 | def _xml_long(self):
|
---|
358 | return '<long>%s</long>' % self.value
|
---|
359 |
|
---|
360 | def _xml_float(self):
|
---|
361 | return '<double>%s</double>' % repr(self.value)
|
---|
362 |
|
---|
363 | def _xml_complex(self):
|
---|
364 | return '<complex>%s</double>' % repr(self.value)
|
---|
365 |
|
---|
366 | class StringCell(BaseCell):
|
---|
367 |
|
---|
368 | def __init__(self, text, fmt="%s", alignment=LEFT):
|
---|
369 | assert isinstance(text, (str, unicode))
|
---|
370 | assert alignment in (LEFT, CENTER, RIGHT)
|
---|
371 | self.text = text
|
---|
372 | self.fmt = fmt
|
---|
373 | self.alignment = alignment
|
---|
374 |
|
---|
375 | def recalc(self, rexec):
|
---|
376 | return self.text
|
---|
377 |
|
---|
378 | def format(self):
|
---|
379 | return self.text, self.alignment
|
---|
380 |
|
---|
381 | def xml(self):
|
---|
382 | s = '<value align="%s" format="%s"><string>%s</string></value>'
|
---|
383 | return s % (
|
---|
384 | align2xml[self.alignment],
|
---|
385 | self.fmt,
|
---|
386 | cgi.escape(self.text))
|
---|
387 |
|
---|
388 | class FormulaCell(BaseCell):
|
---|
389 |
|
---|
390 | def __init__(self, formula, fmt="%s", alignment=RIGHT):
|
---|
391 | assert alignment in (LEFT, CENTER, RIGHT)
|
---|
392 | self.formula = formula
|
---|
393 | self.translated = translate(self.formula)
|
---|
394 | self.fmt = fmt
|
---|
395 | self.alignment = alignment
|
---|
396 | self.reset()
|
---|
397 |
|
---|
398 | def reset(self):
|
---|
399 | self.value = None
|
---|
400 |
|
---|
401 | def recalc(self, rexec):
|
---|
402 | if self.value is None:
|
---|
403 | try:
|
---|
404 | # A hack to evaluate expressions using true division
|
---|
405 | rexec.r_exec("from __future__ import division\n" +
|
---|
406 | "__value__ = eval(%s)" % repr(self.translated))
|
---|
407 | self.value = rexec.r_eval("__value__")
|
---|
408 | except:
|
---|
409 | exc = sys.exc_info()[0]
|
---|
410 | if hasattr(exc, "__name__"):
|
---|
411 | self.value = exc.__name__
|
---|
412 | else:
|
---|
413 | self.value = str(exc)
|
---|
414 | return self.value
|
---|
415 |
|
---|
416 | def format(self):
|
---|
417 | try:
|
---|
418 | text = self.fmt % self.value
|
---|
419 | except:
|
---|
420 | text = str(self.value)
|
---|
421 | return text, self.alignment
|
---|
422 |
|
---|
423 | def xml(self):
|
---|
424 | return '<formula align="%s" format="%s">%s</formula>' % (
|
---|
425 | align2xml[self.alignment],
|
---|
426 | self.fmt,
|
---|
427 | self.formula)
|
---|
428 |
|
---|
429 | def renumber(self, x1, y1, x2, y2, dx, dy):
|
---|
430 | out = []
|
---|
431 | for part in re.split('(\w+)', self.formula):
|
---|
432 | m = re.match('^([A-Z]+)([1-9][0-9]*)$', part)
|
---|
433 | if m is not None:
|
---|
434 | sx, sy = m.groups()
|
---|
435 | x = colname2num(sx)
|
---|
436 | y = int(sy)
|
---|
437 | if x1 <= x <= x2 and y1 <= y <= y2:
|
---|
438 | part = cellname(x+dx, y+dy)
|
---|
439 | out.append(part)
|
---|
440 | return FormulaCell("".join(out), self.fmt, self.alignment)
|
---|
441 |
|
---|
442 | def translate(formula):
|
---|
443 | """Translate a formula containing fancy cell names to valid Python code.
|
---|
444 |
|
---|
445 | Examples:
|
---|
446 | B4 -> cell(2, 4)
|
---|
447 | B4:Z100 -> cells(2, 4, 26, 100)
|
---|
448 | """
|
---|
449 | out = []
|
---|
450 | for part in re.split(r"(\w+(?::\w+)?)", formula):
|
---|
451 | m = re.match(r"^([A-Z]+)([1-9][0-9]*)(?::([A-Z]+)([1-9][0-9]*))?$", part)
|
---|
452 | if m is None:
|
---|
453 | out.append(part)
|
---|
454 | else:
|
---|
455 | x1, y1, x2, y2 = m.groups()
|
---|
456 | x1 = colname2num(x1)
|
---|
457 | if x2 is None:
|
---|
458 | s = "cell(%s, %s)" % (x1, y1)
|
---|
459 | else:
|
---|
460 | x2 = colname2num(x2)
|
---|
461 | s = "cells(%s, %s, %s, %s)" % (x1, y1, x2, y2)
|
---|
462 | out.append(s)
|
---|
463 | return "".join(out)
|
---|
464 |
|
---|
465 | def cellname(x, y):
|
---|
466 | "Translate a cell coordinate to a fancy cell name (e.g. (1, 1)->'A1')."
|
---|
467 | assert x > 0 # Column 0 has an empty name, so can't use that
|
---|
468 | return colnum2name(x) + str(y)
|
---|
469 |
|
---|
470 | def colname2num(s):
|
---|
471 | "Translate a column name to number (e.g. 'A'->1, 'Z'->26, 'AA'->27)."
|
---|
472 | s = s.upper()
|
---|
473 | n = 0
|
---|
474 | for c in s:
|
---|
475 | assert 'A' <= c <= 'Z'
|
---|
476 | n = n*26 + ord(c) - ord('A') + 1
|
---|
477 | return n
|
---|
478 |
|
---|
479 | def colnum2name(n):
|
---|
480 | "Translate a column number to name (e.g. 1->'A', etc.)."
|
---|
481 | assert n > 0
|
---|
482 | s = ""
|
---|
483 | while n:
|
---|
484 | n, m = divmod(n-1, 26)
|
---|
485 | s = chr(m+ord('A')) + s
|
---|
486 | return s
|
---|
487 |
|
---|
488 | import Tkinter as Tk
|
---|
489 |
|
---|
490 | class SheetGUI:
|
---|
491 |
|
---|
492 | """Beginnings of a GUI for a spreadsheet.
|
---|
493 |
|
---|
494 | TO DO:
|
---|
495 | - clear multiple cells
|
---|
496 | - Insert, clear, remove rows or columns
|
---|
497 | - Show new contents while typing
|
---|
498 | - Scroll bars
|
---|
499 | - Grow grid when window is grown
|
---|
500 | - Proper menus
|
---|
501 | - Undo, redo
|
---|
502 | - Cut, copy and paste
|
---|
503 | - Formatting and alignment
|
---|
504 | """
|
---|
505 |
|
---|
506 | def __init__(self, filename="sheet1.xml", rows=10, columns=5):
|
---|
507 | """Constructor.
|
---|
508 |
|
---|
509 | Load the sheet from the filename argument.
|
---|
510 | Set up the Tk widget tree.
|
---|
511 | """
|
---|
512 | # Create and load the sheet
|
---|
513 | self.filename = filename
|
---|
514 | self.sheet = Sheet()
|
---|
515 | if os.path.isfile(filename):
|
---|
516 | self.sheet.load(filename)
|
---|
517 | # Calculate the needed grid size
|
---|
518 | maxx, maxy = self.sheet.getsize()
|
---|
519 | rows = max(rows, maxy)
|
---|
520 | columns = max(columns, maxx)
|
---|
521 | # Create the widgets
|
---|
522 | self.root = Tk.Tk()
|
---|
523 | self.root.wm_title("Spreadsheet: %s" % self.filename)
|
---|
524 | self.beacon = Tk.Label(self.root, text="A1",
|
---|
525 | font=('helvetica', 16, 'bold'))
|
---|
526 | self.entry = Tk.Entry(self.root)
|
---|
527 | self.savebutton = Tk.Button(self.root, text="Save",
|
---|
528 | command=self.save)
|
---|
529 | self.cellgrid = Tk.Frame(self.root)
|
---|
530 | # Configure the widget lay-out
|
---|
531 | self.cellgrid.pack(side="bottom", expand=1, fill="both")
|
---|
532 | self.beacon.pack(side="left")
|
---|
533 | self.savebutton.pack(side="right")
|
---|
534 | self.entry.pack(side="left", expand=1, fill="x")
|
---|
535 | # Bind some events
|
---|
536 | self.entry.bind("<Return>", self.return_event)
|
---|
537 | self.entry.bind("<Shift-Return>", self.shift_return_event)
|
---|
538 | self.entry.bind("<Tab>", self.tab_event)
|
---|
539 | self.entry.bind("<Shift-Tab>", self.shift_tab_event)
|
---|
540 | self.entry.bind("<Delete>", self.delete_event)
|
---|
541 | self.entry.bind("<Escape>", self.escape_event)
|
---|
542 | # Now create the cell grid
|
---|
543 | self.makegrid(rows, columns)
|
---|
544 | # Select the top-left cell
|
---|
545 | self.currentxy = None
|
---|
546 | self.cornerxy = None
|
---|
547 | self.setcurrent(1, 1)
|
---|
548 | # Copy the sheet cells to the GUI cells
|
---|
549 | self.sync()
|
---|
550 |
|
---|
551 | def delete_event(self, event):
|
---|
552 | if self.cornerxy != self.currentxy and self.cornerxy is not None:
|
---|
553 | self.sheet.clearcells(*(self.currentxy + self.cornerxy))
|
---|
554 | else:
|
---|
555 | self.sheet.clearcell(*self.currentxy)
|
---|
556 | self.sync()
|
---|
557 | self.entry.delete(0, 'end')
|
---|
558 | return "break"
|
---|
559 |
|
---|
560 | def escape_event(self, event):
|
---|
561 | x, y = self.currentxy
|
---|
562 | self.load_entry(x, y)
|
---|
563 |
|
---|
564 | def load_entry(self, x, y):
|
---|
565 | cell = self.sheet.getcell(x, y)
|
---|
566 | if cell is None:
|
---|
567 | text = ""
|
---|
568 | elif isinstance(cell, FormulaCell):
|
---|
569 | text = '=' + cell.formula
|
---|
570 | else:
|
---|
571 | text, alignment = cell.format()
|
---|
572 | self.entry.delete(0, 'end')
|
---|
573 | self.entry.insert(0, text)
|
---|
574 | self.entry.selection_range(0, 'end')
|
---|
575 |
|
---|
576 | def makegrid(self, rows, columns):
|
---|
577 | """Helper to create the grid of GUI cells.
|
---|
578 |
|
---|
579 | The edge (x==0 or y==0) is filled with labels; the rest is real cells.
|
---|
580 | """
|
---|
581 | self.rows = rows
|
---|
582 | self.columns = columns
|
---|
583 | self.gridcells = {}
|
---|
584 | # Create the top left corner cell (which selects all)
|
---|
585 | cell = Tk.Label(self.cellgrid, relief='raised')
|
---|
586 | cell.grid_configure(column=0, row=0, sticky='NSWE')
|
---|
587 | cell.bind("<ButtonPress-1>", self.selectall)
|
---|
588 | # Create the top row of labels, and confiure the grid columns
|
---|
589 | for x in range(1, columns+1):
|
---|
590 | self.cellgrid.grid_columnconfigure(x, minsize=64)
|
---|
591 | cell = Tk.Label(self.cellgrid, text=colnum2name(x), relief='raised')
|
---|
592 | cell.grid_configure(column=x, row=0, sticky='WE')
|
---|
593 | self.gridcells[x, 0] = cell
|
---|
594 | cell.__x = x
|
---|
595 | cell.__y = 0
|
---|
596 | cell.bind("<ButtonPress-1>", self.selectcolumn)
|
---|
597 | cell.bind("<B1-Motion>", self.extendcolumn)
|
---|
598 | cell.bind("<ButtonRelease-1>", self.extendcolumn)
|
---|
599 | cell.bind("<Shift-Button-1>", self.extendcolumn)
|
---|
600 | # Create the leftmost column of labels
|
---|
601 | for y in range(1, rows+1):
|
---|
602 | cell = Tk.Label(self.cellgrid, text=str(y), relief='raised')
|
---|
603 | cell.grid_configure(column=0, row=y, sticky='WE')
|
---|
604 | self.gridcells[0, y] = cell
|
---|
605 | cell.__x = 0
|
---|
606 | cell.__y = y
|
---|
607 | cell.bind("<ButtonPress-1>", self.selectrow)
|
---|
608 | cell.bind("<B1-Motion>", self.extendrow)
|
---|
609 | cell.bind("<ButtonRelease-1>", self.extendrow)
|
---|
610 | cell.bind("<Shift-Button-1>", self.extendrow)
|
---|
611 | # Create the real cells
|
---|
612 | for x in range(1, columns+1):
|
---|
613 | for y in range(1, rows+1):
|
---|
614 | cell = Tk.Label(self.cellgrid, relief='sunken',
|
---|
615 | bg='white', fg='black')
|
---|
616 | cell.grid_configure(column=x, row=y, sticky='NSWE')
|
---|
617 | self.gridcells[x, y] = cell
|
---|
618 | cell.__x = x
|
---|
619 | cell.__y = y
|
---|
620 | # Bind mouse events
|
---|
621 | cell.bind("<ButtonPress-1>", self.press)
|
---|
622 | cell.bind("<B1-Motion>", self.motion)
|
---|
623 | cell.bind("<ButtonRelease-1>", self.release)
|
---|
624 | cell.bind("<Shift-Button-1>", self.release)
|
---|
625 |
|
---|
626 | def selectall(self, event):
|
---|
627 | self.setcurrent(1, 1)
|
---|
628 | self.setcorner(sys.maxint, sys.maxint)
|
---|
629 |
|
---|
630 | def selectcolumn(self, event):
|
---|
631 | x, y = self.whichxy(event)
|
---|
632 | self.setcurrent(x, 1)
|
---|
633 | self.setcorner(x, sys.maxint)
|
---|
634 |
|
---|
635 | def extendcolumn(self, event):
|
---|
636 | x, y = self.whichxy(event)
|
---|
637 | if x > 0:
|
---|
638 | self.setcurrent(self.currentxy[0], 1)
|
---|
639 | self.setcorner(x, sys.maxint)
|
---|
640 |
|
---|
641 | def selectrow(self, event):
|
---|
642 | x, y = self.whichxy(event)
|
---|
643 | self.setcurrent(1, y)
|
---|
644 | self.setcorner(sys.maxint, y)
|
---|
645 |
|
---|
646 | def extendrow(self, event):
|
---|
647 | x, y = self.whichxy(event)
|
---|
648 | if y > 0:
|
---|
649 | self.setcurrent(1, self.currentxy[1])
|
---|
650 | self.setcorner(sys.maxint, y)
|
---|
651 |
|
---|
652 | def press(self, event):
|
---|
653 | x, y = self.whichxy(event)
|
---|
654 | if x > 0 and y > 0:
|
---|
655 | self.setcurrent(x, y)
|
---|
656 |
|
---|
657 | def motion(self, event):
|
---|
658 | x, y = self.whichxy(event)
|
---|
659 | if x > 0 and y > 0:
|
---|
660 | self.setcorner(x, y)
|
---|
661 |
|
---|
662 | release = motion
|
---|
663 |
|
---|
664 | def whichxy(self, event):
|
---|
665 | w = self.cellgrid.winfo_containing(event.x_root, event.y_root)
|
---|
666 | if w is not None and isinstance(w, Tk.Label):
|
---|
667 | try:
|
---|
668 | return w.__x, w.__y
|
---|
669 | except AttributeError:
|
---|
670 | pass
|
---|
671 | return 0, 0
|
---|
672 |
|
---|
673 | def save(self):
|
---|
674 | self.sheet.save(self.filename)
|
---|
675 |
|
---|
676 | def setcurrent(self, x, y):
|
---|
677 | "Make (x, y) the current cell."
|
---|
678 | if self.currentxy is not None:
|
---|
679 | self.change_cell()
|
---|
680 | self.clearfocus()
|
---|
681 | self.beacon['text'] = cellname(x, y)
|
---|
682 | self.load_entry(x, y)
|
---|
683 | self.entry.focus_set()
|
---|
684 | self.currentxy = x, y
|
---|
685 | self.cornerxy = None
|
---|
686 | gridcell = self.gridcells.get(self.currentxy)
|
---|
687 | if gridcell is not None:
|
---|
688 | gridcell['bg'] = 'yellow'
|
---|
689 |
|
---|
690 | def setcorner(self, x, y):
|
---|
691 | if self.currentxy is None or self.currentxy == (x, y):
|
---|
692 | self.setcurrent(x, y)
|
---|
693 | return
|
---|
694 | self.clearfocus()
|
---|
695 | self.cornerxy = x, y
|
---|
696 | x1, y1 = self.currentxy
|
---|
697 | x2, y2 = self.cornerxy or self.currentxy
|
---|
698 | if x1 > x2:
|
---|
699 | x1, x2 = x2, x1
|
---|
700 | if y1 > y2:
|
---|
701 | y1, y2 = y2, y1
|
---|
702 | for (x, y), cell in self.gridcells.iteritems():
|
---|
703 | if x1 <= x <= x2 and y1 <= y <= y2:
|
---|
704 | cell['bg'] = 'lightBlue'
|
---|
705 | gridcell = self.gridcells.get(self.currentxy)
|
---|
706 | if gridcell is not None:
|
---|
707 | gridcell['bg'] = 'yellow'
|
---|
708 | self.setbeacon(x1, y1, x2, y2)
|
---|
709 |
|
---|
710 | def setbeacon(self, x1, y1, x2, y2):
|
---|
711 | if x1 == y1 == 1 and x2 == y2 == sys.maxint:
|
---|
712 | name = ":"
|
---|
713 | elif (x1, x2) == (1, sys.maxint):
|
---|
714 | if y1 == y2:
|
---|
715 | name = "%d" % y1
|
---|
716 | else:
|
---|
717 | name = "%d:%d" % (y1, y2)
|
---|
718 | elif (y1, y2) == (1, sys.maxint):
|
---|
719 | if x1 == x2:
|
---|
720 | name = "%s" % colnum2name(x1)
|
---|
721 | else:
|
---|
722 | name = "%s:%s" % (colnum2name(x1), colnum2name(x2))
|
---|
723 | else:
|
---|
724 | name1 = cellname(*self.currentxy)
|
---|
725 | name2 = cellname(*self.cornerxy)
|
---|
726 | name = "%s:%s" % (name1, name2)
|
---|
727 | self.beacon['text'] = name
|
---|
728 |
|
---|
729 |
|
---|
730 | def clearfocus(self):
|
---|
731 | if self.currentxy is not None:
|
---|
732 | x1, y1 = self.currentxy
|
---|
733 | x2, y2 = self.cornerxy or self.currentxy
|
---|
734 | if x1 > x2:
|
---|
735 | x1, x2 = x2, x1
|
---|
736 | if y1 > y2:
|
---|
737 | y1, y2 = y2, y1
|
---|
738 | for (x, y), cell in self.gridcells.iteritems():
|
---|
739 | if x1 <= x <= x2 and y1 <= y <= y2:
|
---|
740 | cell['bg'] = 'white'
|
---|
741 |
|
---|
742 | def return_event(self, event):
|
---|
743 | "Callback for the Return key."
|
---|
744 | self.change_cell()
|
---|
745 | x, y = self.currentxy
|
---|
746 | self.setcurrent(x, y+1)
|
---|
747 | return "break"
|
---|
748 |
|
---|
749 | def shift_return_event(self, event):
|
---|
750 | "Callback for the Return key with Shift modifier."
|
---|
751 | self.change_cell()
|
---|
752 | x, y = self.currentxy
|
---|
753 | self.setcurrent(x, max(1, y-1))
|
---|
754 | return "break"
|
---|
755 |
|
---|
756 | def tab_event(self, event):
|
---|
757 | "Callback for the Tab key."
|
---|
758 | self.change_cell()
|
---|
759 | x, y = self.currentxy
|
---|
760 | self.setcurrent(x+1, y)
|
---|
761 | return "break"
|
---|
762 |
|
---|
763 | def shift_tab_event(self, event):
|
---|
764 | "Callback for the Tab key with Shift modifier."
|
---|
765 | self.change_cell()
|
---|
766 | x, y = self.currentxy
|
---|
767 | self.setcurrent(max(1, x-1), y)
|
---|
768 | return "break"
|
---|
769 |
|
---|
770 | def change_cell(self):
|
---|
771 | "Set the current cell from the entry widget."
|
---|
772 | x, y = self.currentxy
|
---|
773 | text = self.entry.get()
|
---|
774 | cell = None
|
---|
775 | if text.startswith('='):
|
---|
776 | cell = FormulaCell(text[1:])
|
---|
777 | else:
|
---|
778 | for cls in int, long, float, complex:
|
---|
779 | try:
|
---|
780 | value = cls(text)
|
---|
781 | except:
|
---|
782 | continue
|
---|
783 | else:
|
---|
784 | cell = NumericCell(value)
|
---|
785 | break
|
---|
786 | if cell is None and text:
|
---|
787 | cell = StringCell(text)
|
---|
788 | if cell is None:
|
---|
789 | self.sheet.clearcell(x, y)
|
---|
790 | else:
|
---|
791 | self.sheet.setcell(x, y, cell)
|
---|
792 | self.sync()
|
---|
793 |
|
---|
794 | def sync(self):
|
---|
795 | "Fill the GUI cells from the sheet cells."
|
---|
796 | self.sheet.recalc()
|
---|
797 | for (x, y), gridcell in self.gridcells.iteritems():
|
---|
798 | if x == 0 or y == 0:
|
---|
799 | continue
|
---|
800 | cell = self.sheet.getcell(x, y)
|
---|
801 | if cell is None:
|
---|
802 | gridcell['text'] = ""
|
---|
803 | else:
|
---|
804 | if hasattr(cell, 'format'):
|
---|
805 | text, alignment = cell.format()
|
---|
806 | else:
|
---|
807 | text, alignment = str(cell), LEFT
|
---|
808 | gridcell['text'] = text
|
---|
809 | gridcell['anchor'] = align2anchor[alignment]
|
---|
810 |
|
---|
811 |
|
---|
812 | def test_basic():
|
---|
813 | "Basic non-gui self-test."
|
---|
814 | import os
|
---|
815 | a = Sheet()
|
---|
816 | for x in range(1, 11):
|
---|
817 | for y in range(1, 11):
|
---|
818 | if x == 1:
|
---|
819 | cell = NumericCell(y)
|
---|
820 | elif y == 1:
|
---|
821 | cell = NumericCell(x)
|
---|
822 | else:
|
---|
823 | c1 = cellname(x, 1)
|
---|
824 | c2 = cellname(1, y)
|
---|
825 | formula = "%s*%s" % (c1, c2)
|
---|
826 | cell = FormulaCell(formula)
|
---|
827 | a.setcell(x, y, cell)
|
---|
828 | ## if os.path.isfile("sheet1.xml"):
|
---|
829 | ## print "Loading from sheet1.xml"
|
---|
830 | ## a.load("sheet1.xml")
|
---|
831 | a.display()
|
---|
832 | a.save("sheet1.xml")
|
---|
833 |
|
---|
834 | def test_gui():
|
---|
835 | "GUI test."
|
---|
836 | if sys.argv[1:]:
|
---|
837 | filename = sys.argv[1]
|
---|
838 | else:
|
---|
839 | filename = "sheet1.xml"
|
---|
840 | g = SheetGUI(filename)
|
---|
841 | g.root.mainloop()
|
---|
842 |
|
---|
843 | if __name__ == '__main__':
|
---|
844 | #test_basic()
|
---|
845 | test_gui()
|
---|