Martin v. Löwis | 97cf99f | 2008-06-10 04:44:07 +0000 | [diff] [blame] | 1 | #!/usr/bin/python |
| 2 | import sys |
| 3 | import os |
| 4 | |
| 5 | from tkinter import * |
| 6 | from idlelib.Percolator import Percolator |
| 7 | from idlelib.ColorDelegator import ColorDelegator |
| 8 | from idlelib.textView import view_file # TextViewer |
| 9 | from imp import reload |
| 10 | |
| 11 | from tkinter import turtle |
| 12 | import time |
| 13 | |
| 14 | STARTUP = 1 |
| 15 | READY = 2 |
| 16 | RUNNING = 3 |
| 17 | DONE = 4 |
| 18 | EVENTDRIVEN = 5 |
| 19 | |
| 20 | menufont = ("Arial", 12, NORMAL) |
| 21 | btnfont = ("Arial", 12, 'bold') |
| 22 | txtfont = ('Lucida Console', 8, 'normal') |
| 23 | |
| 24 | def getExampleEntries(): |
| 25 | cwd = os.getcwd() |
| 26 | #print(cwd, os.listdir(cwd)) |
| 27 | if "turtleDemo.py" not in os.listdir(cwd): |
| 28 | print("Directory of turtleDemo must be current working directory!") |
| 29 | print("But in your case this is", cwd) |
| 30 | sys.exit() |
| 31 | entries1 = [entry for entry in os.listdir(cwd) if |
| 32 | entry.startswith("tdemo_") and |
| 33 | not entry.endswith(".pyc")] |
| 34 | entries2 = [] |
| 35 | for entry in entries1: |
| 36 | if entry.endswith(".py"): |
| 37 | entries2.append(entry) |
| 38 | else: |
| 39 | path = os.path.join(cwd,entry) |
| 40 | sys.path.append(path) |
| 41 | subdir = [entry] |
| 42 | scripts = [script for script in os.listdir(path) if |
| 43 | script.startswith("tdemo_") and |
| 44 | script.endswith(".py")] |
| 45 | entries2.append(subdir+scripts) |
| 46 | return entries2 |
| 47 | |
| 48 | def showDemoHelp(): |
| 49 | view_file(demo.root, "Help on turtleDemo", "demohelp.txt") |
| 50 | |
| 51 | def showAboutDemo(): |
| 52 | view_file(demo.root, "About turtleDemo", "about_turtledemo.txt") |
| 53 | |
| 54 | def showAboutTurtle(): |
| 55 | view_file(demo.root, "About the new turtle module.", "about_turtle.txt") |
| 56 | |
| 57 | class DemoWindow(object): |
| 58 | |
| 59 | def __init__(self, filename=None): #, root=None): |
| 60 | self.root = root = turtle._root = Tk() |
| 61 | root.wm_protocol("WM_DELETE_WINDOW", self._destroy) |
| 62 | |
| 63 | ################# |
| 64 | self.mBar = Frame(root, relief=RAISED, borderwidth=2) |
| 65 | self.mBar.pack(fill=X) |
| 66 | |
| 67 | self.ExamplesBtn = self.makeLoadDemoMenu() |
| 68 | self.OptionsBtn = self.makeHelpMenu() |
| 69 | self.mBar.tk_menuBar(self.ExamplesBtn, self.OptionsBtn) #, QuitBtn) |
| 70 | |
| 71 | root.title('Python turtle-graphics examples') |
| 72 | ################# |
| 73 | self.left_frame = left_frame = Frame(root) |
| 74 | self.text_frame = text_frame = Frame(left_frame) |
| 75 | self.vbar = vbar =Scrollbar(text_frame, name='vbar') |
| 76 | self.text = text = Text(text_frame, |
| 77 | name='text', padx=5, wrap='none', |
| 78 | width=45) |
| 79 | vbar['command'] = text.yview |
| 80 | vbar.pack(side=LEFT, fill=Y) |
| 81 | ##################### |
| 82 | self.hbar = hbar =Scrollbar(text_frame, name='hbar', orient=HORIZONTAL) |
| 83 | hbar['command'] = text.xview |
| 84 | hbar.pack(side=BOTTOM, fill=X) |
| 85 | ##################### |
| 86 | text['yscrollcommand'] = vbar.set |
| 87 | text.config(font=txtfont) |
| 88 | text.config(xscrollcommand=hbar.set) |
| 89 | text.pack(side=LEFT, fill=Y, expand=1) |
| 90 | ##################### |
| 91 | self.output_lbl = Label(left_frame, height= 1,text=" --- ", bg = "#ddf", |
| 92 | font = ("Arial", 16, 'normal')) |
| 93 | self.output_lbl.pack(side=BOTTOM, expand=0, fill=X) |
| 94 | ##################### |
| 95 | text_frame.pack(side=LEFT, fill=BOTH, expand=0) |
| 96 | left_frame.pack(side=LEFT, fill=BOTH, expand=0) |
| 97 | self.graph_frame = g_frame = Frame(root) |
| 98 | |
| 99 | turtle.Screen._root = g_frame |
| 100 | turtle.Screen._canvas = turtle.ScrolledCanvas(g_frame, 800, 600, 1000, 800) |
| 101 | #xturtle.Screen._canvas.pack(expand=1, fill="both") |
| 102 | self.screen = _s_ = turtle.Screen() |
| 103 | ##### |
| 104 | turtle.TurtleScreen.__init__(_s_, _s_._canvas) |
| 105 | ##### |
| 106 | self.scanvas = _s_._canvas |
| 107 | #xturtle.RawTurtle.canvases = [self.scanvas] |
| 108 | turtle.RawTurtle.screens = [_s_] |
| 109 | |
| 110 | self.scanvas.pack(side=TOP, fill=BOTH, expand=1) |
| 111 | |
| 112 | self.btn_frame = btn_frame = Frame(g_frame, height=100) |
| 113 | self.start_btn = Button(btn_frame, text=" START ", font=btnfont, fg = "white", |
| 114 | disabledforeground = "#fed", command=self.startDemo) |
| 115 | self.start_btn.pack(side=LEFT, fill=X, expand=1) |
| 116 | self.stop_btn = Button(btn_frame, text=" STOP ", font=btnfont, fg = "white", |
| 117 | disabledforeground = "#fed", command = self.stopIt) |
| 118 | self.stop_btn.pack(side=LEFT, fill=X, expand=1) |
| 119 | self.clear_btn = Button(btn_frame, text=" CLEAR ", font=btnfont, fg = "white", |
| 120 | disabledforeground = "#fed", command = self.clearCanvas) |
| 121 | self.clear_btn.pack(side=LEFT, fill=X, expand=1) |
| 122 | |
| 123 | self.btn_frame.pack(side=TOP, fill=BOTH, expand=0) |
| 124 | self.graph_frame.pack(side=TOP, fill=BOTH, expand=1) |
| 125 | |
| 126 | Percolator(text).insertfilter(ColorDelegator()) |
| 127 | self.dirty = False |
| 128 | self.exitflag = False |
| 129 | if filename: |
| 130 | self.loadfile(filename) |
| 131 | self.configGUI(NORMAL, DISABLED, DISABLED, DISABLED, |
| 132 | "Choose example from menu", "black") |
| 133 | self.state = STARTUP |
| 134 | |
| 135 | def _destroy(self): |
| 136 | self.root.destroy() |
| 137 | sys.exit() |
| 138 | |
| 139 | def configGUI(self, menu, start, stop, clear, txt="", color="blue"): |
| 140 | self.ExamplesBtn.config(state=menu) |
| 141 | |
| 142 | self.start_btn.config(state=start) |
| 143 | if start==NORMAL: |
| 144 | self.start_btn.config(bg="#d00") |
| 145 | else: |
| 146 | self.start_btn.config(bg="#fca") |
| 147 | |
| 148 | self.stop_btn.config(state=stop) |
| 149 | if stop==NORMAL: |
| 150 | self.stop_btn.config(bg="#d00") |
| 151 | else: |
| 152 | self.stop_btn.config(bg="#fca") |
| 153 | self.clear_btn.config(state=clear) |
| 154 | |
| 155 | self.clear_btn.config(state=clear) |
| 156 | if clear==NORMAL: |
| 157 | self.clear_btn.config(bg="#d00") |
| 158 | else: |
| 159 | self.clear_btn.config(bg="#fca") |
| 160 | |
| 161 | self.output_lbl.config(text=txt, fg=color) |
| 162 | |
| 163 | |
| 164 | def makeLoadDemoMenu(self): |
| 165 | CmdBtn = Menubutton(self.mBar, text='Examples', underline=0, font=menufont) |
| 166 | CmdBtn.pack(side=LEFT, padx="2m") |
| 167 | CmdBtn.menu = Menu(CmdBtn) |
| 168 | |
| 169 | for entry in getExampleEntries(): |
| 170 | def loadexample(x): |
| 171 | def emit(): |
| 172 | self.loadfile(x) |
| 173 | return emit |
| 174 | if isinstance(entry,str): |
| 175 | CmdBtn.menu.add_command(label=entry[6:-3], underline=0, font=menufont, |
| 176 | command=loadexample(entry)) |
| 177 | else: |
| 178 | _dir, entries = entry[0], entry[1:] |
| 179 | CmdBtn.menu.choices = Menu(CmdBtn.menu) |
| 180 | for e in entries: |
| 181 | CmdBtn.menu.choices.add_command(label=e[6:-3], underline=0, font=menufont, |
| 182 | command = loadexample(os.path.join(_dir,e))) |
| 183 | |
| 184 | CmdBtn.menu.add_cascade(label=_dir[6:], |
| 185 | menu = CmdBtn.menu.choices, font=menufont ) |
| 186 | |
| 187 | CmdBtn['menu'] = CmdBtn.menu |
| 188 | return CmdBtn |
| 189 | |
| 190 | |
| 191 | def makeHelpMenu(self): |
| 192 | CmdBtn = Menubutton(self.mBar, text='Help', underline=0, font = menufont) |
| 193 | CmdBtn.pack(side=LEFT, padx='2m') |
| 194 | CmdBtn.menu = Menu(CmdBtn) |
| 195 | |
| 196 | CmdBtn.menu.add_command(label='About turtle.py', font=menufont, command=showAboutTurtle) |
| 197 | CmdBtn.menu.add_command(label='turtleDemo - Help', font=menufont, command=showDemoHelp) |
| 198 | CmdBtn.menu.add_command(label='About turtleDemo', font=menufont, command=showAboutDemo) |
| 199 | |
| 200 | CmdBtn['menu'] = CmdBtn.menu |
| 201 | return CmdBtn |
| 202 | |
| 203 | def refreshCanvas(self): |
| 204 | if not self.dirty: return |
| 205 | self.screen.clear() |
| 206 | #self.screen.mode("standard") |
| 207 | self.dirty=False |
| 208 | |
| 209 | def loadfile(self,filename): |
| 210 | self.refreshCanvas() |
| 211 | if os.path.exists(filename) and not os.path.isdir(filename): |
| 212 | # load and display file text |
| 213 | f = open(filename,'r') |
| 214 | chars = f.read() |
| 215 | f.close() |
| 216 | self.text.delete("1.0", "end") |
| 217 | self.text.insert("1.0",chars) |
| 218 | direc, fname = os.path.split(filename) |
| 219 | self.root.title(fname[6:-3]+" - a Python turtle graphics example") |
| 220 | self.module = __import__(fname[:-3]) |
| 221 | reload(self.module) |
| 222 | self.configGUI(NORMAL, NORMAL, DISABLED, DISABLED, |
| 223 | "Press start button", "red") |
| 224 | self.state = READY |
| 225 | |
| 226 | def startDemo(self): |
| 227 | self.refreshCanvas() |
| 228 | self.dirty = True |
| 229 | turtle.TurtleScreen._RUNNING = True |
| 230 | self.configGUI(DISABLED, DISABLED, NORMAL, DISABLED, |
| 231 | "demo running...", "black") |
| 232 | self.screen.clear() |
| 233 | self.screen.mode("standard") |
| 234 | self.state = RUNNING |
| 235 | |
| 236 | try: |
| 237 | result = self.module.main() |
| 238 | if result == "EVENTLOOP": |
| 239 | self.state = EVENTDRIVEN |
| 240 | else: |
| 241 | self.state = DONE |
| 242 | except turtle.Terminator: |
| 243 | self.state = DONE |
| 244 | result = "stopped!" |
| 245 | if self.state == DONE: |
| 246 | self.configGUI(NORMAL, NORMAL, DISABLED, NORMAL, |
| 247 | result) |
| 248 | elif self.state == EVENTDRIVEN: |
| 249 | self.exitflag = True |
| 250 | self.configGUI(DISABLED, DISABLED, NORMAL, DISABLED, |
| 251 | "use mouse/keys or STOP", "red") |
| 252 | |
| 253 | def clearCanvas(self): |
| 254 | self.refreshCanvas() |
| 255 | self.screen._delete("all") |
| 256 | self.scanvas.config(cursor="") |
| 257 | self.configGUI(NORMAL, NORMAL, DISABLED, DISABLED) |
| 258 | |
| 259 | def stopIt(self): |
| 260 | if self.exitflag: |
| 261 | self.clearCanvas() |
| 262 | self.exitflag = False |
| 263 | self.configGUI(NORMAL, NORMAL, DISABLED, DISABLED, |
| 264 | "STOPPED!", "red") |
| 265 | turtle.TurtleScreen._RUNNING = False |
| 266 | #print "stopIT: exitflag = True" |
| 267 | else: |
| 268 | turtle.TurtleScreen._RUNNING = False |
| 269 | #print "stopIt: exitflag = False" |
| 270 | |
| 271 | if __name__ == '__main__': |
| 272 | demo = DemoWindow() |
| 273 | RUN = True |
| 274 | while RUN: |
| 275 | try: |
| 276 | #print("ENTERING mainloop") |
| 277 | demo.root.mainloop() |
| 278 | except AttributeError: |
| 279 | #print("AttributeError!- WAIT A MOMENT!") |
| 280 | time.sleep(0.3) |
| 281 | print("GOING ON ..") |
| 282 | demo.ckearCanvas() |
| 283 | except TypeError: |
| 284 | demo.screen._delete("all") |
| 285 | #print("CRASH!!!- WAIT A MOMENT!") |
| 286 | time.sleep(0.3) |
| 287 | #print("GOING ON ..") |
| 288 | demo.clearCanvas() |
| 289 | except: |
| 290 | print("BYE!") |
| 291 | RUN = False |