Animation (01.05.2014 - 01.10.2014)

This week I got animation down.

import pygame, sys, time
from pygame.locals import *

#Set up pygame
pygame.init

#set up window
WINDOWWIDTH=400
WINDOWHEIGHT=400
windowSurface=pygame.display.set_mode((WINDOWWIDTH, WINDOWHEIGHT), 0, 32)
pygame.display.set_caption('Animation')

#set up direction variables
DOWNLEFT=1
DOWNRIGHT=3
UPLEFT=7
UPRIGHT=9

MOVESPEED=4

#set up colors
BLACK=(0, 0, 0)
RED=(255, 0, 0)
GREEN=(0, 255, 0)
BLUE=(0, 0, 255)

#Set up block data structure
b1={'rect':pygame.Rect(300, 80, 50, 100), 'color':RED, 'dir':UPRIGHT}
b2={'rect':pygame.Rect(200, 200, 20, 20), 'color':GREEN, 'dir':UPLEFT}
b3={'rect':pygame.Rect(100, 150, 60, 60), 'color':BLUE, 'dir':DOWNLEFT}

blocks=[b1, b2, b3]

#run the game loop
while True:
    #check for the QUIT event
    for event in pygame.event.get():
        if event.type==QUIT:
            pygame.quit()
            sys.exit()

    #draw the black background onto the surface
    windowSurface.fill(BLACK)

    for b in blocks:
        #move the block data structure
        if b['dir']==DOWNLEFT:
            b['rect'].left-=MOVESPEED
            b['rect'].top+=MOVESPEED
        if b['dir']==DOWNRIGHT:
            b['rect'].left+=MOVESPEED
            b['rect'].top+=MOVESPEED
        if b['dir']==UPLEFT:
            b['rect'].left-=MOVESPEED
            b['rect'].top-=MOVESPEED
        if b['dir']==UPRIGHT:
            b['rect'].left+=MOVESPEED
            b['rect'].top-=MOVESPEED

        #Check if the block has moved out of the window
        if b['rect'].top<0:
            #block has moved past top
            if b['dir']==UPLEFT:
                b['dir']=DOWNLEFT
            if b['dir']==UPRIGHT:
                b['dir']=DOWNRIGHT
        if b['rect'].bottom>WINDOWHEIGHT:
            #block has moved past bottom
            if b['dir']==DOWNLEFT:
                b['dir']=UPLEFT
            if b['dir']==DOWNRIGHT:
                b['dir']=UPRIGHT
        if b['rect'].left<0:
            #block has moved past left side
            if b['dir']==DOWNLEFT:
                b['dir']=DOWNRIGHT
            if b['dir']==UPLEFT:
                b['dir']=UPRIGHT
        if b['rect'].right>WINDOWHEIGHT:
            #block has moved past right side
            if b['dir']==DOWNRIGHT:
                b['dir']=DOWNLEFT
            if b['dir']==UPRIGHT:
                b['dir']=UPLEFT

        #Draw the block onto the surface
        pygame.draw.rect(windowSurface, b['color'], b['rect'])

    #Draw the window onto the screen
    pygame.display.update()
    time.sleep(0.02)