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