si23traktor/main.py

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()