mirror of
https://github.com/kristoferssolo/School.git
synced 2025-10-21 20:10:38 +00:00
added snake
This commit is contained in:
parent
785c7c08ee
commit
d9ea2405fe
1
pygame/snake/requirements.txt
Normal file
1
pygame/snake/requirements.txt
Normal file
@ -0,0 +1 @@
|
||||
pygame==2.1.2
|
||||
217
pygame/snake/snake.py
Executable file
217
pygame/snake/snake.py
Executable file
@ -0,0 +1,217 @@
|
||||
# Author - Kristiāns Francis Cagulis
|
||||
# Date - 11.03.2022
|
||||
# Title - Snake
|
||||
|
||||
from random import randrange, randint
|
||||
import pygame
|
||||
import tkinter as tk
|
||||
from tkinter import messagebox
|
||||
|
||||
RED = (255, 0, 0)
|
||||
BLACK = (0, 0, 0)
|
||||
WHITE = (255, 255, 255)
|
||||
GREEN = (0, 128, 30)
|
||||
PURPLE = (170, 0, 255)
|
||||
RANDOM_COLOR = (randint(0, 255), randint(0, 255), randint(0, 255))
|
||||
|
||||
WIDTH = 500
|
||||
ROWS = 20
|
||||
|
||||
|
||||
class Cube(object):
|
||||
width = WIDTH
|
||||
rows = ROWS
|
||||
|
||||
def __init__(self, start, color=RANDOM_COLOR):
|
||||
self.pos = start
|
||||
self.dirnx = 1
|
||||
self.dirny = 0
|
||||
self.color = color
|
||||
|
||||
def move(self, dirnx, dirny):
|
||||
self.dirnx = dirnx
|
||||
self.dirny = dirny
|
||||
self.pos = (self.pos[0] + self.dirnx, self.pos[1] + self.dirny)
|
||||
|
||||
# eyes
|
||||
def draw(self, surface, eyes=False):
|
||||
distance = self.width // self.rows
|
||||
i = self.pos[0]
|
||||
j = self.pos[1]
|
||||
|
||||
pygame.draw.rect(surface, self.color, (i * distance + 1, j * distance + 1, distance - 2, distance - 2))
|
||||
if eyes:
|
||||
center = distance // 2
|
||||
radius = 3
|
||||
circle_middle = (i * distance + center - radius, j * distance + 8)
|
||||
circle_middle_2 = (i * distance + distance - radius * 2, j * distance + 8)
|
||||
pygame.draw.circle(surface, BLACK, circle_middle, radius)
|
||||
pygame.draw.circle(surface, BLACK, circle_middle_2, radius)
|
||||
|
||||
|
||||
class Snake(object):
|
||||
body = []
|
||||
turns = {}
|
||||
|
||||
def __init__(self, color, pos):
|
||||
self.color = color
|
||||
self.head = Cube(pos, self.color)
|
||||
self.body.append(self.head)
|
||||
self.dirnx = 0
|
||||
self.dirny = 1
|
||||
|
||||
def move(self):
|
||||
for event in pygame.event.get():
|
||||
if event.type == pygame.QUIT:
|
||||
pygame.quit()
|
||||
keys = pygame.key.get_pressed()
|
||||
|
||||
for _ in keys:
|
||||
if keys[pygame.K_LEFT]:
|
||||
self.dirnx = -1
|
||||
self.dirny = 0
|
||||
self.turns[self.head.pos[:]] = [self.dirnx, self.dirny]
|
||||
|
||||
elif keys[pygame.K_RIGHT]:
|
||||
self.dirnx = 1
|
||||
self.dirny = 0
|
||||
self.turns[self.head.pos[:]] = [self.dirnx, self.dirny]
|
||||
|
||||
elif keys[pygame.K_UP]:
|
||||
self.dirnx = 0
|
||||
self.dirny = -1
|
||||
self.turns[self.head.pos[:]] = [self.dirnx, self.dirny]
|
||||
|
||||
elif keys[pygame.K_DOWN]:
|
||||
self.dirnx = 0
|
||||
self.dirny = 1
|
||||
self.turns[self.head.pos[:]] = [self.dirnx, self.dirny]
|
||||
|
||||
for i, c in enumerate(self.body):
|
||||
p = c.pos[:]
|
||||
if p in self.turns:
|
||||
turn = self.turns[p]
|
||||
c.move(turn[0], turn[1])
|
||||
if i == len(self.body) - 1:
|
||||
self.turns.pop(p)
|
||||
|
||||
else:
|
||||
if c.dirnx == -1 and c.pos[0] <= 0:
|
||||
c.pos = (c.rows - 1, c.pos[1])
|
||||
|
||||
elif c.dirnx == 1 and c.pos[0] >= c.rows - 1:
|
||||
c.pos = (0, c.pos[1])
|
||||
|
||||
elif c.dirny == 1 and c.pos[1] >= c.rows - 1:
|
||||
c.pos = (c.pos[0], 0)
|
||||
|
||||
elif c.dirny == -1 and c.pos[1] <= 0:
|
||||
c.pos = (c.pos[0], c.rows - 1)
|
||||
|
||||
else:
|
||||
c.move(c.dirnx, c.dirny)
|
||||
|
||||
def reset(self, pos):
|
||||
self.head = Cube(pos)
|
||||
self.body = []
|
||||
self.turns = {}
|
||||
self.dirnx = 0
|
||||
self.dirny = 1
|
||||
|
||||
def add_cube(self):
|
||||
tail = self.body[-1]
|
||||
dx, dy = tail.dirnx, tail.dirny
|
||||
|
||||
if dx == 1 and dy == 0:
|
||||
self.body.append(Cube((tail.pos[0] - 1, tail.pos[1])))
|
||||
elif dx == -1 and dy == 0:
|
||||
self.body.append(Cube((tail.pos[0] + 1, tail.pos[1])))
|
||||
elif dx == 0 and dy == 1:
|
||||
self.body.append(Cube((tail.pos[0], tail.pos[1] - 1)))
|
||||
elif dx == 0 and dy == -1:
|
||||
self.body.append(Cube((tail.pos[0], tail.pos[1] + 1)))
|
||||
|
||||
self.body[-1].dirnx = dx
|
||||
self.body[-1].dirny = dy
|
||||
|
||||
def draw(self, surface):
|
||||
for i, c in enumerate(self.body):
|
||||
if i == 0:
|
||||
c.draw(surface, True)
|
||||
|
||||
else:
|
||||
c.draw(surface)
|
||||
|
||||
|
||||
def draw_grid(width, rows, surface):
|
||||
size_between = width // rows
|
||||
x = 0
|
||||
y = 0
|
||||
for _ in range(rows):
|
||||
x += size_between
|
||||
y += size_between
|
||||
|
||||
pygame.draw.line(surface, WHITE, (x, 0), (x, width))
|
||||
pygame.draw.line(surface, WHITE, (0, y), (width, y))
|
||||
|
||||
|
||||
def redraw_window(surface):
|
||||
surface.fill(BLACK)
|
||||
s.draw(surface)
|
||||
snack.draw(surface)
|
||||
draw_grid(WIDTH, ROWS, surface)
|
||||
pygame.display.update()
|
||||
|
||||
|
||||
def random_snack(rows, item):
|
||||
positions = item.body
|
||||
while True:
|
||||
x = randrange(rows)
|
||||
y = randrange(rows)
|
||||
if len(list(filter(lambda z: z.pos == (x, y), positions))) > 0:
|
||||
continue
|
||||
else:
|
||||
break
|
||||
return x, y
|
||||
|
||||
|
||||
def message_box(subject, content):
|
||||
root = tk.Tk()
|
||||
root.attributes("-topmost", True)
|
||||
root.withdraw()
|
||||
messagebox.showinfo(subject, content)
|
||||
try:
|
||||
root.destroy()
|
||||
except:
|
||||
pass
|
||||
|
||||
|
||||
def main():
|
||||
win = pygame.display.set_mode((WIDTH, WIDTH))
|
||||
flag = True
|
||||
clock = pygame.time.Clock()
|
||||
global snack
|
||||
snack = Cube(random_snack(ROWS, s), color=RED)
|
||||
while flag:
|
||||
pygame.time.delay(50)
|
||||
clock.tick(10)
|
||||
s.move()
|
||||
if s.body[0].pos == snack.pos:
|
||||
s.add_cube()
|
||||
snack = Cube(random_snack(ROWS, s), color=RED)
|
||||
|
||||
for i in range(len(s.body)):
|
||||
if s.body[i].pos in list(map(lambda z: z.pos, s.body[i + 1:])):
|
||||
print(f"Score: {len(s.body)}")
|
||||
message_box("You Lost!", f"Your score was {len(s.body)}.\nPlay again")
|
||||
s.reset((10, 10))
|
||||
break
|
||||
redraw_window(win)
|
||||
|
||||
pass
|
||||
|
||||
|
||||
s = Snake(PURPLE, (10, 10))
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
@ -107,7 +107,7 @@ class Ship:
|
||||
|
||||
class Player(Ship):
|
||||
|
||||
def __init__(self, x: int, y: int, lives: int = 3) -> None:
|
||||
def __init__(self, x: int, y: int, lives: int = 3000) -> None:
|
||||
super().__init__(x, y, lives)
|
||||
self.ship_img = SPACESHIP
|
||||
self.missile_img = PLAYER_MISSILE
|
||||
@ -167,8 +167,8 @@ def main() -> None:
|
||||
enemies = []
|
||||
player_vel = 7
|
||||
player_missile_vel = 15
|
||||
enemy_x_vel = 2
|
||||
enemy_y_vel = 2
|
||||
enemy_x_vel = 5
|
||||
enemy_y_vel = 5
|
||||
vel_x = enemy_x_vel
|
||||
enemy_missile_vel = 6
|
||||
last_enemy_shot = 0
|
||||
@ -252,7 +252,7 @@ def main() -> None:
|
||||
vel_x = -enemy_x_vel
|
||||
|
||||
elif enemy.x <= 0:
|
||||
move(0, enemy_y_vel, enemies)
|
||||
move(0, -enemy_y_vel, enemies)
|
||||
vel_x = enemy_x_vel
|
||||
|
||||
enemy.move_missiles(enemy_missile_vel, player)
|
||||
|
||||
Loading…
Reference in New Issue
Block a user