forklift image

main.py

@@ -3,36 +3,55 @@ import sys
 
 pygame.init()
 screen = pygame.display.set_mode((1280, 720))
-wozek = pygame.Rect(55,55,50,50)
 c = (0, 150, 0)
 
-while True:
-    for event in pygame.event.get():
-        if event.type == pygame.QUIT:
-            sys.exit(0)
-        elif event.type == pygame.KEYDOWN and event.key == pygame.K_ESCAPE:
-            sys.exit(0)
-        elif event.type == pygame.KEYDOWN and event.key == pygame.K_DOWN:
-            if wozek.y <= 600:
-                    wozek.y += 80
-        elif event.type == pygame.KEYDOWN and event.key == pygame.K_UP:
-            if wozek.y >= 100:
-                wozek.y -= 80
-        elif event.type == pygame.KEYDOWN and event.key == pygame.K_RIGHT:
-            if wozek.x <= 1170:
-                wozek.x += 80
-        elif event.type == pygame.KEYDOWN and event.key == pygame.K_LEFT:
-            if wozek.x >= 100:
-                wozek.x -= 80
+class Wozek:
+    def __init__(self):
+        self.x = 55
+        self.y = 55
+        self.height = 64
+        self.width = 64
+        self.image = pygame.image.load("wozek.png") 
+        # Credit: Forklift icons created by Smashicons - Flaticon
+        # https://www.flaticon.com/free-icons/forklift
+
+    def draw(self):
+        screen.blit(self.image, (self.x, self.y))
 
 
-    for x in range(0, 1280, 80):
-        pygame.draw.line(screen, c, (1, x), (1280, x), 2)
-        pygame.draw.line(screen, c, (x, 1), (x, 720), 2)
-        pygame.display.update()
+def main():
+    wozek = Wozek()
+    while True:
+        for event in pygame.event.get():
+            if event.type == pygame.QUIT:
+                sys.exit(0)
+            elif event.type == pygame.KEYDOWN and event.key == pygame.K_ESCAPE:
+                sys.exit(0)
+            elif event.type == pygame.KEYDOWN and event.key == pygame.K_DOWN:
+                if wozek.y <= 600:
+                        wozek.y += 80
+            elif event.type == pygame.KEYDOWN and event.key == pygame.K_UP:
+                if wozek.y >= 100:
+                    wozek.y -= 80
+            elif event.type == pygame.KEYDOWN and event.key == pygame.K_RIGHT:
+                if wozek.x <= 1170:
+                    wozek.x += 80
+            elif event.type == pygame.KEYDOWN and event.key == pygame.K_LEFT:
+                if wozek.x >= 100:
+                    wozek.x -= 80
+
+
+        for x in range(0, 1280, 80):
+            pygame.draw.line(screen, c, (1, x), (1280, x), 2)
+            pygame.draw.line(screen, c, (x, 1), (x, 720), 2)
+            pygame.display.update()
     
 
-    #Drawing
-    screen.fill((0,0,0)) #removes object trail
-    pygame.draw.rect(screen, (0,150,255), wozek) #resolution, color, object
-    pygame.display.flip()   #updating frames
+        #Drawing
+        screen.fill((0,0,0)) #removes object trail
+        pygame.display.flip()   #updating frames
+        wozek.draw()
+
+
+if __name__ == "__main__":
+    main()