]> git.saurik.com Git - wxWidgets.git/blame - wxPython/samples/hangman/hangman.py
Corrected line endings
[wxWidgets.git] / wxPython / samples / hangman / hangman.py
CommitLineData
cf694132
RD
1"""Hangman.py, a simple wxPython game, inspired by the
2old bsd game by Ken Arnold.
3From the original man page:
4
5 In hangman, the computer picks a word from the on-line
6 word list and you must try to guess it. The computer
7 keeps track of which letters have been guessed and how
8 many wrong guesses you have made on the screen in a
9 graphic fashion.
10
11That says it all, doesn't it?
12
13Have fun with it,
14
15Harm van der Heijden (H.v.d.Heijden@phys.tue.nl)"""
16
1e4a197e 17import random,re
cf694132
RD
18from wxPython.wx import *
19
20
21
22class WordFetcher:
23 builtin_words = ' albatros banana electrometer eggshell'
24
25 def __init__(self, filename, min_length = 5):
26 self.min_length = min_length
27 print "Trying to open file %s" % (filename,)
28 try:
29 f = open(filename, "r")
30 except:
31 print "Couldn't open dictionary file %s, using builtins" % (filename,)
32 self.words = self.builtin_words
33 self.filename = None
34 return
35 self.words = f.read()
36 self.filename = filename
37 print "Got %d bytes." % (len(self.words),)
38
39 def SetMinLength(min_length):
40 self.min_length = min_length
41
42 def Get(self):
43 reg = re.compile('\s+([a-zA-Z]+)\s+')
44 n = 50 # safety valve; maximum number of tries to find a suitable word
45 while n:
46 index = int(random.random()*len(self.words))
47 m = reg.search(self.words[index:])
48 if m and len(m.groups()[0]) >= self.min_length: break
49 n = n - 1
1e4a197e 50 if n: return m.groups()[0].lower()
cf694132
RD
51 return "error"
52
53
54
55def stdprint(x):
56 print x
57
58
59
60class URLWordFetcher(WordFetcher):
61 def __init__(self, url):
62 self.OpenURL(url)
63 WordFetcher.__init__(self, "hangman_dict.txt")
64
65 def logprint(self,x):
66 print x
67
68 def RetrieveAsFile(self, host, path=''):
69 from httplib import HTTP
70 try:
71 h = HTTP(host)
72 except:
73 self.logprint("Failed to create HTTP connection to %s... is the network available?" % (host))
74 return None
75 h.putrequest('GET',path)
76 h.putheader('Accept','text/html')
77 h.putheader('Accept','text/plain')
78 h.endheaders()
79 errcode, errmsg, headers = h.getreply()
80 if errcode != 200:
81 self.logprint("HTTP error code %d: %s" % (errcode, errmsg))
82 return None
83 f = h.getfile()
84 return f
85
86 def OpenURL(self,url):
87 from htmllib import HTMLParser
88 import formatter
89 self.url = url
90 m = re.match('http://([^/]+)(/\S*)\s*', url)
91 if m:
92 host = m.groups()[0]
93 path = m.groups()[1]
94 else:
95 m = re.match('http://(\S+)\s*', url)
96 if not m:
97 # Invalid URL
98 self.logprint("Invalid or unsupported URL: %s" % (url))
99 return
100 host = m.groups()[0]
101 path = ''
102 f = self.RetrieveAsFile(host,path)
103 if not f:
104 self.logprint("Could not open %s" % (url))
105 return
106 self.logprint("Receiving data...")
107 data = f.read()
108 tmp = open('hangman_dict.txt','w')
109 fmt = formatter.AbstractFormatter(formatter.DumbWriter(tmp))
110 p = HTMLParser(fmt)
111 self.logprint("Parsing data...")
112 p.feed(data)
113 p.close()
114 tmp.close()
115
116
117
118class HangmanWnd(wxWindow):
119 def __init__(self, parent, id, pos=wxDefaultPosition, size=wxDefaultSize):
120 wxWindow.__init__(self, parent, id, pos, size)
121 self.SetBackgroundColour(wxNamedColour('white'))
122 if wxPlatform == '__WXGTK__':
123 self.font = wxFont(12, wxMODERN, wxNORMAL, wxNORMAL)
124 else:
125 self.font = wxFont(10, wxMODERN, wxNORMAL, wxNORMAL)
126 self.SetFocus()
f6bcfd97
BP
127 EVT_PAINT(self, self.OnPaint)
128
cf694132
RD
129
130 def StartGame(self, word):
131 self.word = word
132 self.guess = []
133 self.tries = 0
134 self.misses = 0
135 self.Draw()
136
137 def EndGame(self):
138 self.misses = 7;
139 self.guess = map(chr, range(ord('a'),ord('z')+1))
140 self.Draw()
141
142 def HandleKey(self, key):
143 self.message = ""
144 if self.guess.count(key):
145 self.message = 'Already guessed %s' % (key,)
146 return 0
147 self.guess.append(key)
148 self.guess.sort()
149 self.tries = self.tries+1
150 if not key in self.word:
151 self.misses = self.misses+1
152 if self.misses == 7:
153 self.EndGame()
154 return 1
155 has_won = 1
156 for letter in self.word:
157 if not self.guess.count(letter):
158 has_won = 0
159 break
160 if has_won:
161 self.Draw()
162 return 2
163 self.Draw()
164 return 0
165
166 def Draw(self, dc = None):
167 if not dc:
168 dc = wxClientDC(self)
169 dc.SetFont(self.font)
170 dc.Clear()
171 (x,y) = self.GetSizeTuple()
172 x1 = x-200; y1 = 20
173 for letter in self.word:
174 if self.guess.count(letter):
175 dc.DrawText(letter, x1, y1)
176 else:
177 dc.DrawText('.', x1, y1)
178 x1 = x1 + 10
179 x1 = x-200
180 dc.DrawText("tries %d misses %d" % (self.tries,self.misses),x1,50)
181 guesses = ""
182 for letter in self.guess:
183 guesses = guesses + letter
184 dc.DrawText("guessed:", x1, 70)
185 dc.DrawText(guesses[:13], x1+80, 70)
186 dc.DrawText(guesses[13:], x1+80, 90)
187 dc.SetUserScale(x/1000.0, y/1000.0)
188 self.DrawVictim(dc)
189
190 def DrawVictim(self, dc):
191 dc.SetPen(wxPen(wxNamedColour('black'), 20))
192 dc.DrawLines([(10, 980), (10,900), (700,900), (700,940), (720,940),
193 (720,980), (900,980)])
194 dc.DrawLines([(100,900), (100, 100), (300,100)])
195 dc.DrawLine(100,200,200,100)
196 if ( self.misses == 0 ): return
197 dc.SetPen(wxPen(wxNamedColour('blue'), 10))
198 dc.DrawLine(300,100,300,200)
199 if ( self.misses == 1 ): return
200 dc.DrawEllipse(250,200,100,100)
201 if ( self.misses == 2 ): return
202 dc.DrawLine(300,300,300,600)
203 if ( self.misses == 3) : return
204 dc.DrawLine(300,300,250,550)
205 if ( self.misses == 4) : return
206 dc.DrawLine(300,300,350,550)
207 if ( self.misses == 5) : return
208 dc.DrawLine(300,600,350,850)
209 if ( self.misses == 6) : return
210 dc.DrawLine(300,600,250,850)
211
212 def OnPaint(self, event):
213 dc = wxPaintDC(self)
214 self.Draw(dc)
215
216
217
218class HangmanDemo(HangmanWnd):
219 def __init__(self, wf, parent, id, pos, size):
220 HangmanWnd.__init__(self, parent, id, pos, size)
221 self.StartGame("dummy")
222 self.start_new = 1
223 self.wf = wf
224 self.delay = 500
225 self.timer = self.PlayTimer(self.MakeMove)
226
227 def MakeMove(self):
228 self.timer.Stop()
229 if self.start_new:
230 self.StartGame(self.wf.Get())
231 self.start_new = 0
232 self.left = list('aaaabcdeeeeefghiiiiijklmnnnoooopqrssssttttuuuuvwxyz')
233 else:
234 key = self.left[int(random.random()*len(self.left))]
235 while self.left.count(key): self.left.remove(key)
236 self.start_new = self.HandleKey(key)
237 self.timer.Start(self.delay)
238
239 def Stop(self):
240 self.timer.Stop()
241
242 class PlayTimer(wxTimer):
243 def __init__(self,func):
244 wxTimer.__init__(self)
245 self.func = func
246 self.Start(1000)
247
248 def Notify(self):
249 apply(self.func, ())
250
251
252
253class HangmanDemoFrame(wxFrame):
254 def __init__(self, wf, parent, id, pos, size):
255 wxFrame.__init__(self, parent, id, "Hangman demo", pos, size)
256 self.demo = HangmanDemo(wf, self, -1, wxDefaultPosition, wxDefaultSize)
f6bcfd97 257 EVT_CLOSE(self, self.OnCloseWindow)
cf694132
RD
258
259 def OnCloseWindow(self, event):
260 self.demo.timer.Stop()
261 self.Destroy()
262
263
264
265class AboutBox(wxDialog):
266 def __init__(self, parent,wf):
267 wxDialog.__init__(self, parent, -1, "About Hangman", wxDefaultPosition, wxSize(350,450))
268 self.wnd = HangmanDemo(wf, self, -1, wxPoint(1,1), wxSize(350,150))
269 self.static = wxStaticText(self, -1, __doc__, wxPoint(1,160), wxSize(350, 250))
270 self.button = wxButton(self, 2001, "OK", wxPoint(150,420), wxSize(50,-1))
271 EVT_BUTTON(self, 2001, self.OnOK)
272
273 def OnOK(self, event):
274 self.wnd.Stop()
275 self.EndModal(wxID_OK)
276
277
278
279class MyFrame(wxFrame):
280 def __init__(self, parent, wf):
281 self.wf = wf
282 wxFrame.__init__(self, parent, -1, "hangman", wxDefaultPosition, wxSize(400,300))
283 self.wnd = HangmanWnd(self, -1)
284 menu = wxMenu()
285 menu.Append(1001, "New")
286 menu.Append(1002, "End")
287 menu.AppendSeparator()
288 menu.Append(1003, "Reset")
289 menu.Append(1004, "Demo...")
290 menu.AppendSeparator()
291 menu.Append(1005, "Exit")
292 menubar = wxMenuBar()
293 menubar.Append(menu, "Game")
294 menu = wxMenu()
1e4a197e 295 #menu.Append(1010, "Internal", "Use internal dictionary", True)
cf694132 296 menu.Append(1011, "ASCII File...")
6c5ae2d2 297 urls = [ 'wxPython home', 'http://wxPython.org/',
cf694132
RD
298 'slashdot.org', 'http://slashdot.org/',
299 'cnn.com', 'http://cnn.com',
300 'The New York Times', 'http://www.nytimes.com',
301 'De Volkskrant', 'http://www.volkskrant.nl/frameless/25000006.html',
302 'Gnu GPL', 'http://www.fsf.org/copyleft/gpl.html',
303 'Bijbel: Genesis', 'http://www.coas.com/bijbel/gn1.htm']
304 urlmenu = wxMenu()
305 for item in range(0,len(urls),2):
306 urlmenu.Append(1020+item/2, urls[item], urls[item+1])
307 urlmenu.Append(1080, 'Other...', 'Enter an URL')
308 menu.AppendMenu(1012, 'URL', urlmenu, 'Use a webpage')
309 menu.Append(1013, 'Dump', 'Write contents to stdout')
310 menubar.Append(menu, "Dictionary")
311 self.urls = urls
312 self.urloffset = 1020
313 menu = wxMenu()
314 menu.Append(1090, "About...")
315 menubar.Append(menu, "Help")
316 self.SetMenuBar(menubar)
317 self.CreateStatusBar(2)
318 EVT_MENU(self, 1001, self.OnGameNew)
319 EVT_MENU(self, 1002, self.OnGameEnd)
320 EVT_MENU(self, 1003, self.OnGameReset)
321 EVT_MENU(self, 1004, self.OnGameDemo)
322 EVT_MENU(self, 1005, self.OnWindowClose)
323 EVT_MENU(self, 1011, self.OnDictFile)
324 EVT_MENU_RANGE(self, 1020, 1020+len(urls)/2, self.OnDictURL)
325 EVT_MENU(self, 1080, self.OnDictURLSel)
326 EVT_MENU(self, 1013, self.OnDictDump)
327 EVT_MENU(self, 1090, self.OnHelpAbout)
328 EVT_CHAR(self.wnd, self.OnChar)
329 self.OnGameReset()
330
331 def OnGameNew(self, event):
332 word = self.wf.Get()
333 self.in_progress = 1
334 self.SetStatusText("",0)
335 self.wnd.StartGame(word)
336
337 def OnGameEnd(self, event):
338 self.UpdateAverages(0)
339 self.in_progress = 0
340 self.SetStatusText("",0)
341 self.wnd.EndGame()
342
343 def OnGameReset(self, event=None):
344 self.played = 0
345 self.won = 0
346 self.history = []
347 self.average = 0.0
348 self.OnGameNew(None)
349
350 def OnGameDemo(self, event):
351 frame = HangmanDemoFrame(self.wf, self, -1, wxDefaultPosition, self.GetSize())
1e4a197e 352 frame.Show(True)
cf694132
RD
353
354 def OnDictFile(self, event):
355 fd = wxFileDialog(self)
356 if (self.wf.filename):
357 fd.SetFilename(self.wf.filename)
358 if fd.ShowModal() == wxID_OK:
359 file = fd.GetPath()
360 self.wf = WordFetcher(file)
361
362 def OnDictURL(self, event):
363 item = (event.GetId() - self.urloffset)*2
364 print "Trying to open %s at %s" % (self.urls[item], self.urls[item+1])
365 self.wf = URLWordFetcher(self.urls[item+1])
366
367 def OnDictURLSel(self, event):
368 msg = wxTextEntryDialog(self, "Enter the URL of the dictionary document", "Enter URL")
369 if msg.ShowModal() == wxID_OK:
370 url = msg.GetValue()
371 self.wf = URLWordFetcher(url)
372 def OnDictDump(self, event):
373 print self.wf.words
374
375 def OnHelpAbout(self, event):
376 about = AboutBox(self, self.wf)
377 about.ShowModal()
378 about.wnd.Stop() # that damn timer won't stop!
379
380 def UpdateAverages(self, has_won):
381 if has_won:
382 self.won = self.won + 1
383 self.played = self.played+1
384 self.history.append(self.wnd.misses) # ugly
385 total = 0.0
386 for m in self.history:
387 total = total + m
388 self.average = float(total/len(self.history))
389
390 def OnChar(self, event):
391 if not self.in_progress:
bb0054cd 392 #print "new"
cf694132
RD
393 self.OnGameNew(None)
394 return
395 key = event.KeyCode();
bb0054cd 396 #print key
cf694132
RD
397 if key >= ord('A') and key <= ord('Z'):
398 key = key + ord('a') - ord('A')
399 key = chr(key)
400 if key < 'a' or key > 'z':
401 event.Skip()
402 return
403 res = self.wnd.HandleKey(key)
404 if res == 0:
405 self.SetStatusText(self.wnd.message)
406 elif res == 1:
407 self.UpdateAverages(0)
408 self.SetStatusText("Too bad, you're dead!",0)
409 self.in_progress = 0
410 elif res == 2:
411 self.in_progress = 0
412 self.UpdateAverages(1)
413 self.SetStatusText("Congratulations!",0)
414 if self.played:
415 percent = (100.*self.won)/self.played
416 else:
417 percent = 0.0
418 self.SetStatusText("p %d, w %d (%g %%), av %g" % (self.played,self.won, percent, self.average),1)
419
420 def OnWindowClose(self, event):
421 self.Destroy()
422
423
424
425class MyApp(wxApp):
426 def OnInit(self):
427 if wxPlatform == '__WXGTK__':
428 defaultfile = "/usr/share/games/hangman-words"
429 elif wxPlatform == '__WXMSW__':
430 defaultfile = "c:\\windows\\hardware.txt"
431 else:
432 defaultfile = ""
433 wf = WordFetcher(defaultfile)
493f1553 434 frame = MyFrame(None, wf)
cf694132 435 self.SetTopWindow(frame)
1e4a197e
RD
436 frame.Show(True)
437 return True
cf694132
RD
438
439
440
441if __name__ == '__main__':
442 app = MyApp(0)
443 app.MainLoop()
444
445
446#----------------------------------------------------------------------
447
448overview = __doc__
449
450
451def runTest(frame, nb, log):
452 if wxPlatform == '__WXGTK__' or wxPlatform == '__WXMOTIF__':
453 defaultfile = "/usr/share/games/hangman-words"
454 elif wxPlatform == '__WXMSW__':
455 defaultfile = "c:\\windows\\hardware.txt"
456 else:
457 defaultfile = ""
458 wf = WordFetcher(defaultfile)
459 win = MyFrame(frame, wf)
460 frame.otherWin = win
1e4a197e 461 win.Show(True)
cf694132
RD
462
463
464#----------------------------------------------------------------------
465
466
467
468