111 lines
3.5 KiB
Python
111 lines
3.5 KiB
Python
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()
|