import pygame from pygame.locals import * import random from datetime import datetime class Lines: def __init__(self, parent_scr): self.parent_scr = parent_scr def draw_lines(self): # background lines for i in range(1, 10): pygame.draw.line(self.parent_scr, (228, 253, 227), (50 * i, 0), (50 * i, 500), 1) pygame.draw.line(self.parent_scr, (228, 253, 227), (0, 50 * i), (500, 50 * i), 1) pygame.display.flip() class Tractor: def __init__(self, parent_screen): self.parent_screen = parent_screen self.block = pygame.image.load(r'resources/arrow.png').convert() self.x = 100 self.y = 100 self.angle = 0 self.direction = 'up' def draw(self): self.parent_screen.fill((120, 120, 0)) # background color self.parent_screen.blit(self.block, (self.x, self.y)) self.parent_screen.blit(pygame.transform.rotate(self.block, self.angle), (self.x, self.y)) # rotate tractor pygame.display.flip() # updating screen def move(self, direction): if direction == 'up': self.y -= 50 self.angle = 0 if direction == 'down': self.y += 50 self.angle = 180 if direction == 'left': self.x -= 50 self.angle = 90 if direction == 'right': self.x += 50 self.angle = 270 self.draw() 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) class Game: def __init__(self): pygame.init() self.screenWidth = 500 self.screenHeight = 500 self.surface = pygame.display.set_mode((self.screenWidth, self.screenHeight)) # initialize a window # self.surface.fill((255, 255, 255)) # background color (overwritten by tractor) self.tractor = Tractor(self.surface) self.tractor.draw() self.lines = Lines(self.surface) def run(self): running = True last_time = datetime.now() self.lines.draw_lines() while running: time_now = datetime.now() for event in pygame.event.get(): if event.type == KEYDOWN: if pygame.key.get_pressed()[K_ESCAPE]: running = False # in case we want to use keyboard # if pygame.key.get_pressed()[K_UP]: # self.tractor.move('up') # if pygame.key.get_pressed()[K_DOWN]: # self.tractor.move('down') # if pygame.key.get_pressed()[K_LEFT]: # self.tractor.move('left') # if pygame.key.get_pressed()[K_RIGHT]: # self.tractor.move('right') elif event.type == QUIT: running = False if (time_now - last_time).total_seconds() > 1: # tractor moves every 1 sec last_time = datetime.now() self.tractor.walk() self.lines.draw_lines() print(f'x, y = ({int(self.tractor.x / 50)}, {int(self.tractor.y / 50)})') if __name__ == '__main__': game = Game() game.run()