Mercurial > mamba
annotate mamba/widgets/messagebox.py @ 227:c068190c0b1d
Cleanup copy-n-paste left-over junk
author | Neil Muller <drnlmuller@gmail.com> |
---|---|
date | Wed, 14 Sep 2011 23:06:14 +0200 |
parents | b92f705bd8ea |
children | af202e15167d |
rev | line source |
---|---|
190 | 1 import pygame |
210
b92f705bd8ea
Modal dialogs, and focus stealing bugfixes
Stefano Rivera <stefano@rivera.za.net>
parents:
190
diff
changeset
|
2 from pygame.constants import SRCALPHA |
190 | 3 |
4 from mamba.widgets.base import Container | |
5 from mamba.widgets.text import TextWidget, TextButton | |
6 | |
7 | |
8 class MessageBox(Container): | |
9 | |
10 def __init__(self, rect, text, post_callback=None, color='white'): | |
11 super(MessageBox, self).__init__(rect) | |
12 self.text = text | |
13 self.post_callback = post_callback | |
14 self.color = color | |
15 self.prepare() | |
210
b92f705bd8ea
Modal dialogs, and focus stealing bugfixes
Stefano Rivera <stefano@rivera.za.net>
parents:
190
diff
changeset
|
16 self.modal = True |
190 | 17 |
18 def prepare(self): | |
19 cont = TextWidget((0, 0), "Press [OK] or Enter to continue") | |
20 message = TextWidget((0, 0), self.text, color=self.color) | |
21 width = max(cont.rect.width, message.rect.width) | |
22 if width > message.rect.width: | |
23 cont_pos = (self.rect.left + 5, | |
24 self.rect.top + 10 + message.rect.height) | |
25 message_pos = (self.rect.left + 5 + width / 2 | |
26 - message.rect.width / 2, self.rect.top + 5) | |
27 else: | |
28 message_pos = (self.rect.left + 5, self.rect.top + 5) | |
29 cont_pos = (self.rect.left + 5 + width / 2 | |
30 - message.rect.width / 2, | |
31 self.rect.top + 10 + message.rect.height) | |
32 message.rect.topleft = message_pos | |
33 cont.rect.topleft = cont_pos | |
34 self.add(message) | |
35 self.add(cont) | |
210
b92f705bd8ea
Modal dialogs, and focus stealing bugfixes
Stefano Rivera <stefano@rivera.za.net>
parents:
190
diff
changeset
|
36 self.ok_button = ok_button = TextButton((0, 0), 'OK') |
190 | 37 ok_pos = (self.rect.left + 5 + width / 2 - ok_button.rect.width / 2, |
38 cont_pos[1] + 5 + cont.rect.height) | |
39 ok_button.rect.topleft = ok_pos | |
40 ok_button.add_callback('clicked', self.close) | |
41 self.add(ok_button) | |
42 | |
43 def draw(self, surface): | |
44 background = pygame.Surface(self.rect.size, SRCALPHA) | |
45 background.fill(pygame.Color('gray')) | |
46 surface.blit(background, self.rect) | |
47 super(MessageBox, self).draw(surface) | |
48 | |
49 def close(self, ev, widget): | |
50 self.parent.remove(self) | |
51 if self.post_callback: | |
52 self.post_callback() | |
210
b92f705bd8ea
Modal dialogs, and focus stealing bugfixes
Stefano Rivera <stefano@rivera.za.net>
parents:
190
diff
changeset
|
53 |
b92f705bd8ea
Modal dialogs, and focus stealing bugfixes
Stefano Rivera <stefano@rivera.za.net>
parents:
190
diff
changeset
|
54 def grab_focus(self): |
b92f705bd8ea
Modal dialogs, and focus stealing bugfixes
Stefano Rivera <stefano@rivera.za.net>
parents:
190
diff
changeset
|
55 return self.ok_button.grab_focus() |