diff --git a/images/pelny_wozek_1_crate.png b/images/pelny_wozek_1_crate.png new file mode 100644 index 0000000..a55c1b1 Binary files /dev/null and b/images/pelny_wozek_1_crate.png differ diff --git a/images/pelny_wozek_2_crates.png b/images/pelny_wozek_2_crates.png new file mode 100644 index 0000000..6cb7103 Binary files /dev/null and b/images/pelny_wozek_2_crates.png differ diff --git a/images/pelny_wozek_full_3_crates.png b/images/pelny_wozek_full_3_crates.png new file mode 100644 index 0000000..849de5d Binary files /dev/null and b/images/pelny_wozek_full_3_crates.png differ diff --git a/main.py b/main.py index 3b9df7f..6296b94 100644 --- a/main.py +++ b/main.py @@ -2,6 +2,7 @@ import sys import pygame import regal import paczka +#from wozek import Wozek pygame.init() screen = pygame.display.set_mode((980, 980)) @@ -11,7 +12,7 @@ pygame.display.set_caption("Inteligentny wozek") icon = pygame.image.load('images/icon.png') pygame.display.set_icon(icon) -class Wozek: +class Wozek(): def __init__(self): self.x = 55 self.y = 55 @@ -26,9 +27,34 @@ class Wozek: def draw(self): screen.blit(self.image, (self.x, self.y)) + #storage = ["none"] * 10 + storage = [] + max_size = 10 + + def add_element(element, storage, max_size): + if len(storage) < max_size: + storage.append(element) + else: + print("I'm full!") + def remove_element(storage): + if len(storage) > 0: + place = storage.pop() + return place + else: + print("I'm empty!") + + def dynamic_wozek_picture(wozek, storage): + if len(storage) == 0: + wozek.image = pygame.image.load("images/pusty_wozek.png") + elif ((len(storage) > 0) and (len(storage) < 4)): + wozek.image = pygame.image.load("images/pelny_wozek_1_crate.png") + elif ((len(storage) > 3) and (len(storage) < 10)): + wozek.image = pygame.image.load("images/pelny_wozek_2_crates.png") + elif (len(storage) == 10): + wozek.image = pygame.image.load("images/pelny_wozek_full_3_crates.png") + def main(): - wozek = Wozek() while True: for event in pygame.event.get(): diff --git a/wozek.py b/wozek.py new file mode 100644 index 0000000..cc083f8 --- /dev/null +++ b/wozek.py @@ -0,0 +1,44 @@ +from main import pygame, screen + +#screen nie dziaƂa + +class Wozek(): + def __init__(self): + self.x = 55 + self.y = 55 + self.x_change = 0 + self.y_change = 0 + self.height = 64 + self.width = 64 + self.image = pygame.image.load("images/pusty_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)) + + #storage = ["none"] * 10 + storage = [] + max_size = 10 + + def add_element(element, storage, max_size): + if len(storage) < max_size: + storage.append(element) + else: + print("I'm full!") + def remove_element(storage): + if len(storage) > 0: + place = storage.pop() + return place + else: + print("I'm empty!") + + def dynamic_wozek_picture(wozek, storage): + if len(storage) == 0: + wozek.image = pygame.image.load("images/pusty_wozek.png") + elif ((len(storage) > 0) and (len(storage) < 4)): + wozek.image = pygame.image.load("images/pelny_wozek_1_crate.png") + elif ((len(storage) > 3) and (len(storage) < 10)): + wozek.image = pygame.image.load("images/pelny_wozek_2_crates.png") + elif (len(storage) == 10): + wozek.image = pygame.image.load("images/pelny_wozek_full_3_crates.png") \ No newline at end of file