l4/pkg/python/contrib/Demo/tkinter/guido/ss1.py

   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()