Add what we did in class today

Author: awenelo

in-class-main.py

@@ -12,6 +12,7 @@
 class Rectangles():
     def __init__(self, topleft, wh, speed, screensize):
         self.rect = pygame.Rect(topleft, wh)
+        self.backuprect = pygame.Rect(topleft, wh)
         self.speed = speed
         self.screen = pygame.Rect((0, 0), screensize)
 
@@ -35,20 +36,80 @@ def update(self):
         if self.rect.right > self.screen.right:
             self.rect.x = self.screen.right-self.rect.width
 
+    def draw(self, screen):
+        pygame.draw.rect(screen, (0, 255, 0), self.rect)
+    
+    def goto(self, pos, walls):
+        rectcenterbackup = self.rect.center
+        self.rect.center = pos
+        for wall in walls:
+            if self.rect.colliderect(wall.rect):
+                self.rect.center = rectcenterbackup
+
+    def reset(self):
+        self.rect = self.backuprect.move(0, 0)
+
+class Wall(pygame.sprite.Sprite):
+    def __init__(self, x, y, w, h):
+        super().__init__()
+        self.rect = pygame.Rect((x, y), (w, h))
+
     def draw(self, screen):
         pygame.draw.rect(screen, (255, 0, 0), self.rect)
 
-rect = Rectangles((200, 150), (200, 300), 10, (WIDTH, HEIGHT))
+class LoseScreen():
+    def __init__(self, screensize):
+        self.screensize = screensize
+        self.screenrect = pygame.Rect((0, 0), screensize)
+        self.font = pygame.font.Font(None, 100)
+        self.text = self.font.render("You lost!", True, (255, 255, 255))
+        self.textrect = self.text.get_rect(center = (screensize[0]/2, screensize[1]/2))
+        self.count = 0
+    
+    def draw(self, screen):
+        if self.count > 0:
+            pygame.draw.rect(screen, (255, 0, 0), self.screenrect)
+            screen.blit(self.text, self.textrect)
+
+    def update(self, reset=False):
+        if reset:
+            self.count = 40
+            return 0
+        self.count -= 1
+        return 0 if self.count > 0 else 1
+
+rect = Rectangles((200, 150), (100, 100), 10, (WIDTH, HEIGHT))
+losescreen = LoseScreen((WIDTH, HEIGHT))
+level = 1
 
 clock = pygame.time.Clock()
 
+walls = pygame.sprite.Group()
+walls.add(Wall(0, 0, 100, 100))
+walls.add(Wall(400, 0, 100, 100))
+
 while True:
     clock.tick(FRAMERATE)
     for event in pygame.event.get():
         if event.type == QUIT:
             raise SystemExit
+        elif event.type == MOUSEBUTTONDOWN:
+            if event.button == 1:
+                rect.goto(event.pos, walls)
+        elif event.type == USEREVENT and event.code == "LOSE":
+            level = losescreen.update(reset=True)
+            rect.reset()
     screen.fill((255,255,255))
-    rect.update()
-    rect.draw(screen)
+    if level == 0:
+        level = losescreen.update()
+        losescreen.draw(screen)
+    else:
+        rect.update()
+        rect.draw(screen)
+        for wall in walls:
+            wall.draw(screen)
+            if rect.rect.colliderect(wall.rect):
+                pygame.event.post(pygame.event.Event(USEREVENT, code="LOSE"))
+
 
-    pygame.display.update()
+    pygame.display.update()