44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
|
import pygame
|
||
|
|
import random
|
||
|
|
|
||
|
|
class Tractor:
|
||
|
|
def __init__(self, parent_screen, cell_size):
|
||
|
|
self.parent_screen = parent_screen
|
||
|
|
self.image = pygame.image.load(r'resources/tractor.png').convert_alpha()
|
||
|
|
self.image = pygame.transform.scale(self.image, (cell_size, cell_size+5))
|
||
|
|
self.x = cell_size*2
|
||
|
|
self.y = cell_size*2
|
||
|
|
self.angle = 0
|
||
|
|
self.direction = 'up'
|
||
|
|
|
||
|
|
def draw(self):
|
||
|
|
self.parent_screen.blit(pygame.transform.rotate(self.image, self.angle), (self.x, self.y)) # rotate tractor
|
||
|
|
|
||
|
|
def move(self, direction, cell_size):
|
||
|
|
if direction == 'up':
|
||
|
|
self.y -= cell_size
|
||
|
|
self.angle = 0
|
||
|
|
if direction == 'down':
|
||
|
|
self.y += cell_size
|
||
|
|
self.angle = 180
|
||
|
|
if direction == 'left':
|
||
|
|
self.x -= cell_size
|
||
|
|
self.angle = 90
|
||
|
|
if direction == 'right':
|
||
|
|
self.x += cell_size
|
||
|
|
self.angle = 270
|
||
|
|
|
||
|
|
def walk(self):
|
||
|
|
choice = ['up', 'down', 'left', 'right']
|
||
|
|
|
||
|
|
if self.x == 450:
|
||
|
|
choice.pop(3)
|
||
|
|
if self.x == 0:
|
||
|
|
choice.pop(2)
|
||
|
|
if self.y == 0:
|
||
|
|
choice.pop(0)
|
||
|
|
if self.y == 450:
|
||
|
|
choice.pop(1)
|
||
|
|
|
||
|
|
self.direction = random.choice(choice)
|
||
|
|
self.move(self.direction)
|