import pygame
import sys

pygame.init()
screen = pygame.display.set_mode((1280, 720))
c = (0, 150, 0)


def draw_grid():
    for y in range(80, 720, 80):  # horizontal lines
        pygame.draw.line(screen, c, (80, y), (1280 - 80, y), 1)
    for x in range(80, 1280, 80):  # vertical lines
        pygame.draw.line(screen, c, (x, 80), (x, 720 - 80), 1)


class Wozek:
    def __init__(self):
        self.x = 55
        self.y = 55
        self.height = 64
        self.width = 64
        self.image = pygame.image.load("wozek.png")
        # Credit: Forklift icons created by Smashicons - Flaticon
        # https://www.flaticon.com/free-icons/forklift

    def draw(self):
        screen.blit(self.image, (self.x, self.y))


def main():
    wozek = Wozek()
    while True:
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                sys.exit(0)
            elif event.type == pygame.KEYDOWN and event.key == pygame.K_ESCAPE:
                sys.exit(0)
            elif event.type == pygame.KEYDOWN and event.key == pygame.K_DOWN:
                if wozek.y <= 600:
                    wozek.y += 80
            elif event.type == pygame.KEYDOWN and event.key == pygame.K_UP:
                if wozek.y >= 100:
                    wozek.y -= 80
            elif event.type == pygame.KEYDOWN and event.key == pygame.K_RIGHT:
                if wozek.x <= 1170:
                    wozek.x += 80
            elif event.type == pygame.KEYDOWN and event.key == pygame.K_LEFT:
                if wozek.x >= 100:
                    wozek.x -= 80

        # Drawing
        screen.fill((0, 0, 0))  # removes object trail
        draw_grid()
        wozek.draw()
        pygame.display.flip()  # updating frames


if __name__ == "__main__":
    main()