Inteligentny_Wozek/paczka.py

48 lines
1.5 KiB
Python

from etykieta import Etykieta
from main import pygame
class Paczka:
def __init__(self, rozmiar, waga, kategoria, czy_krucha, nadawca, adres, imie, nazwisko, telefon):
self.rozmiar = rozmiar
self.waga = waga
self.kategoria = kategoria
self.czy_krucha = czy_krucha
self.x = 0
self.y = 0
self.szerokosc = 0
self.wysokosc = 0
self.image = pygame.image.load("images/paczka.png")
self.label = Etykieta(nadawca, adres, imie, nazwisko, telefon)
# zmienia rozmiar obrazka w zaleznosci od rozmiaru
def __dobierz_rozmiar_obrazka(self):
if self.rozmiar == "duzy":
self.image = pygame.transform.scale(self.image, (50, 50))
self.szerokosc = 50
self.wysokosc = 50
return 1
elif self.rozmiar == "sredni":
self.image = pygame.transform.scale(self.image, (35, 35))
self.szerokosc = 35
self.wysokosc = 35
return 1
elif self.rozmiar == "maly":
self.image = pygame.transform.scale(self.image, (20, 20))
self.szerokosc = 20
self.wysokosc = 20
return 1
else:
return 0
def narysuj(self, x, y, screen):
self.x = x
self.y = y
if self.__dobierz_rozmiar_obrazka() == 1:
screen.blit(self.image, (self.x, self.y))
else:
print("Zmien rozmiar paczki")
print("Paczka moze miec rozmiar duzy, sredni lub maly")
exit(0)