Snaps versions

Made all the versions of snaps in the different folders the same
version.

Ch 03 Python Program Structure/snaps.py

@@ -3,6 +3,8 @@
 # Rob Miles July 2017
 # Version 1.0
 
+import time
+import random
 import pygame
 
 surface = None
@@ -42,12 +44,12 @@ def setup(width=800, height=600, title=''):
 
     pygame.display.set_caption(title)
 
-
 def handle_events():
     '''
     Consume events that are generated by the pygame window
-    These are not presntly used for anything
+    Captures key pressed events and sets the key_pressed value 
     '''
+    global key_pressed
     setup()
     for event in pygame.event.get():
         pass
@@ -95,54 +97,65 @@ def clear_display():
     if image is not None:
         surface.blit(image, (0, 0))
 
+    pygame.display.flip()
+
+def split_lines_on_spaces(text):
+    '''
+    returns a list of words which have been
+    extracted from the text.
+    Spaces on the ends of words are
+    preserved
+    '''
+    result = []
+    got_space = False
+    word = ''
+    for ch in text:
+        if ch == ' ':
+            got_space = True
+            word = word + ch
+        else:
+            if got_space:
+                # first character of next word
+                result.append(word)
+                word=ch
+                got_space=False
+            else:
+                word = word + ch
+
+    result.append(word)
+
+    return result            
+
 
 def get_display_lines(text, font, width):
     '''
     Returns a list of strings which have been split
     to fit the given window width using the supplied font
     '''
-    space_width = font.size(' ')[0]
     result = []
-    text_lines = text.splitlines()
+    text_lines =  text.splitlines()
     for text_line in text_lines:
-        words = text_line.split()
+        words = split_lines_on_spaces(text_line)
         x = 0
         line = ''
         for word in words:
             word_width = font.size(word)[0]
             if x + word_width > width:
                 # Remove the trailing space from the line
                 # before adding to the list of lines to return
-                line = line.strip()
                 result.append(line)
-                line = word + ' '
-                x = word_width + space_width
+                line = word
+                x = word_width
             else:
-                line = line + word + ' '
-                x = x + word_width + space_width
-
-        if line != '':
-            # Got a partial line to add to the end
-            # Remove the trailing space from the line
-            # before adding to the list of lines to return
-            line = line.strip()
-            result.append(line)
-    return result
+                line = line + word
+                x = x + word_width
 
+        result.append(line)
+    return result
 
-def display_message(text, size=200, margin=20, horiz='center', vert='center',
-                    color=(255, 0, 0)):
-    '''
-    Displays the text as a message
-    Sice can be used to select the size of the
-    text
-    '''
-    global window_size
-    global surface
-
-    handle_events()
 
-    clear_display()
+def render_message(text, size=200, margin=20, horiz='center', vert='center',
+                    color=(255, 0, 0), cursor=''):
 
     # Get the text version of the input
     text = str(text)
@@ -167,20 +180,239 @@ def display_message(text, size=200, margin=20, horiz='center', vert='center',
 
     if vert == 'center':
         y = (window_size[1] - height) / 2.0
-    elif vert == 'top':
-        y = margin
     elif vert == 'bottom':
         y=(window_size[1]-margin) - height
+    else:
+        # default vertical cursor position is top
+        y = margin
+
 
     for rendered_line in rendered_lines:
         width = rendered_line.get_width()
         height = rendered_line.get_height()
         if horiz == 'center':
             x = (available_width - width) / 2.0 + margin
-        elif horiz == 'left':
-            x = margin
         elif horiz == 'right':
-            x = self.window_size[0] - width - margin
+            x = window_size[0] - width - margin
+        else:
+            # default position is left margin
+            x = margin
         surface.blit(rendered_line, (x, y))
         y += height
+
+    if cursor:
+        cursor_size = font.size(cursor)
+        cursor_width = cursor_size[0]
+        cursor_height = cursor_size[1]
+        if len(rendered_lines):
+            # put the cursor on the end of an existing line
+            y -= height
+            x += width
+        else:
+            # put the cursor in the start position for this
+            # orientation
+            # default x position is the margin
+            x = margin
+
+            if horiz == 'center':
+                x = (available_width - cursor_width) / 2.0 + margin
+            elif horiz == 'right':
+                x = window_size[0] - cursor_width - margin
+            else:
+                # default position is left margin
+                x = margin
+
+            if vert == 'center':
+                y = (window_size[1] - cursor_height) / 2.0
+            elif vert == 'bottom':
+                y=(window_size[1]-margin) - cursor_height
+            else:
+                # default vertical cursor position is top
+                y = margin
+
+        cursor_image = font.render(cursor, 1, color)
+        surface.blit(cursor_image, (x, y))
+
+    pygame.display.flip()
+
+
+def display_message(text, size=200, margin=20, horiz='center', vert='center',
+                    color=(255, 0, 0)):
+    '''
+    Displays the text as a message
+    Size can be used to select the size of the
+    text
+    '''
+    global window_size
+    global surface
+
+    handle_events()
+
+    clear_display()
+
+    render_message(text, size=size, margin=margin, horiz=horiz, vert=vert, color=color)
+
+def get_dot():
+    '''
+    Waits for a mouse movement and then returns it
+    as a tuple of x and y coordinates
+    '''
+    setup()
+    while True:
+        event = pygame.event.wait()
+        if event.type == 4:
+            # Event 4 is mouse motion
+            pos = event.dict['pos']
+            return pos
+
+BLACK = (  0,   0,   0)
+WHITE = (255, 255, 255)
+BLUE =  (  0,   0, 255)
+GREEN = (  0, 255,   0)
+RED =   (255,   0,   0)
+YELLOW =   (255,   255,   0)
+MAGENTA =   (255,   0,   255)
+CYAN =   (0,   255,   255)
+
+dot_color = WHITE
+
+def set_color(r,g,b):
+    dot_color = (r,g,b)
+
+def set_random_color():
+    global dot_color
+    dot_color = (random.randint(0,255),
+                 random.randint(0,255),
+                 random.randint(0,255))
+
+def get_mouse_pressed():
+    return pygame.mouse.get_pressed()[0]
+
+
+def draw_dot(pos, radius):
+    setup()
+    pygame.draw.circle(surface, dot_color, pos, radius)
     pygame.display.flip()
+
+
+def get_key():
+    '''
+    Waits for a keypress and then returns it as a string
+    Only characters are returned, not control keys
+    '''
+    setup()
+    while True:
+        event = pygame.event.wait()
+        if event.type == 2:
+            # Event 2 is keydown
+            key_code = event.dict['unicode']
+            if key_code:
+                return key_code
+
+
+def get_string(prompt, size=50, margin=20,
+               color=(255, 0, 0), horiz='left', vert='center',
+               max_line_length=20):
+    '''
+    Reads a string from the user
+    '''
+
+    setup()
+
+    result = ''
+    cursor_char = '*'
+    cursor = None
+
+    def redraw():
+        clear_display()
+
+        render_message(prompt+result, margin=margin, size=size,
+                       horiz=horiz, vert=vert, color=color, cursor=cursor)
+
+    def cursor_flip():
+        nonlocal cursor
+
+    # create a timer for the cursor
+
+    cursor_event = pygame.USEREVENT+1
+
+    pygame.time.set_timer(cursor_event,500)
+
+    while True:
+        event = pygame.event.wait()
+
+        if event.type == cursor_event:
+            if cursor:
+                cursor = None
+            else:
+                cursor = cursor_char
+            redraw()
+        elif event.type == 2:
+            # Event 2 is keydown
+            key_code = event.dict['unicode']
+            if key_code is None:
+                continue
+            if key_code == '\r':
+                break
+            elif key_code == '\x08':
+                if len(result) > 0:
+                    result=result[:-1]
+                    redraw()
+            else:
+                if len(result) < max_line_length:
+                    result += key_code
+                    redraw()
+
+    # disable the timer for the cursor
+    pygame.time.set_timer(cursor_event,0)
+    return result
+
+import urllib.request
+import xml.etree.ElementTree
+
+def get_weather_temp(latitude,longitude):
+    '''
+    Uses forecast.weather.gov to get the weather
+    for the specified latitude and longitude
+    '''
+    url="http://forecast.weather.gov/MapClick.php?lat={0}&lon={1}&unit=0&lg=english&FcstType=dwml".format(latitude,longitude)
+    req=urllib.request.urlopen(url)
+    page=req.read()
+    doc=xml.etree.ElementTree.fromstring(page)
+    # I'm not proud of this, but by gum it works...
+    for child in doc:
+        if child.tag == 'data':
+            if child.attrib['type'] == 'current observations':
+                for item in child:
+                    if item.tag == 'parameters':
+                        for i in item:
+                            if i.tag == 'temperature':
+                                if i.attrib['type'] == 'apparent':
+                                    for t in i:
+                                        if t.tag =='value':
+                                            return int(t.text)
+
+
+def get_weather_desciption(latitude,longitude):
+    '''
+    Uses forecast.weather.gov to get the weather
+    for the specified latitude and longitude
+    '''
+    url="http://forecast.weather.gov/MapClick.php?lat={0}&lon={1}&unit=0&lg=english&FcstType=dwml".format(latitude,longitude)
+    req=urllib.request.urlopen(url)
+    page=req.read()
+    doc=xml.etree.ElementTree.fromstring(page)
+    # I'm not proud of this, but by gum it works...
+    for child in doc:
+        if child.tag == 'data':
+            if child.attrib['type'] == 'current observations':
+                for item in child:
+                    if item.tag == 'parameters':
+                        for i in item:
+                            if i.tag == 'weather':
+                                for t in i:
+                                    if t.tag == 'weather-conditions':
+                                        if t.get('weather-summary') is not None:
+                                            return t.get('weather-summary')
+
+