31 lines
939 B
Python
31 lines
939 B
Python
|
import numpy as np
|
||
|
|
import glob2
|
||
|
|
import pygame
|
||
|
|
import random
|
||
|
|
RECT_SIZE = 50
|
||
|
|
|
||
|
|
|
||
|
|
class Package:
|
||
|
|
def __init__(self, window):
|
||
|
|
self.window = window
|
||
|
|
self.length = np.random.randint(1, 3) * RECT_SIZE -1
|
||
|
|
self.width = RECT_SIZE-1
|
||
|
|
self.color = list(np.random.choice(range(256), size=3))
|
||
|
|
self.x = 1251
|
||
|
|
self.y = 351
|
||
|
|
# self.mark = random.choice(['fragile', 'dont turn around', 'keep dry', 'glass'])
|
||
|
|
self.mark_image = self.get_marking_photo()
|
||
|
|
self.block = pygame.Rect(self.x, self.y, self.width, self.length)
|
||
|
|
|
||
|
|
def get_marking_photo(self):
|
||
|
|
file_path_type = ["resources/package_markings/*.jpg"]
|
||
|
|
images = glob2.glob(random.choice(file_path_type))
|
||
|
|
random_image = random.choice(images)
|
||
|
|
print(random_image)
|
||
|
|
return random_image
|
||
|
|
|
||
|
|
def draw(self):
|
||
|
|
pygame.draw.rect(self.window, self.color, self.block)
|
||
|
|
pygame.display.flip()
|
||
|
|
|