mirror of
https://git.davidovski.xyz/tkinter-space-game.git
synced 2024-08-15 00:43:41 +00:00
89 lines
2.6 KiB
Python
89 lines
2.6 KiB
Python
|
from dataclasses import dataclass
|
||
|
|
from typing import List
|
||
|
|
|
||
|
|
from game import DamageableSprite, Game, GameSprite
|
||
|
|
from sprite import Sprite
|
||
|
|
|
||
|
|
|
||
|
|
class Lazer(GameSprite):
|
||
|
|
"""Lazer object that is shot by a shooter"""
|
||
|
|
|
||
|
|
def __init__(self, game: Game, velocity=-4, color="white"):
|
||
|
|
"""Initialise the lazer
|
||
|
|
|
||
|
|
:param game: The game which this belongs to
|
||
|
|
:type game: Game
|
||
|
|
:param velocity: Velocity to move the lazer at
|
||
|
|
:param color: name of the colour of the lazer
|
||
|
|
"""
|
||
|
|
self.velocity = velocity
|
||
|
|
self.game = game
|
||
|
|
super().__init__(game, game.texture_factory.get_image(
|
||
|
|
f"lazer:{color}"))
|
||
|
|
|
||
|
|
def tick(self):
|
||
|
|
"""Update this object"""
|
||
|
|
self.move(0, self.velocity)
|
||
|
|
if self.y + self.h > self.game.h or self.y < 0:
|
||
|
|
self.destroy()
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class ShooterAttributes:
|
||
|
|
"""Attributes for a shooter object"""
|
||
|
|
|
||
|
|
lazer_color: str = "white"
|
||
|
|
cooldown: int = 40
|
||
|
|
velocity: int = 1
|
||
|
|
hp: int = 3
|
||
|
|
|
||
|
|
|
||
|
|
class Shooter(DamageableSprite):
|
||
|
|
"""A game object that is able to shoot lazers"""
|
||
|
|
|
||
|
|
def __init__(self, game: Game,
|
||
|
|
image_name: str, attributes: ShooterAttributes):
|
||
|
|
"""Initialise the shooter
|
||
|
|
|
||
|
|
:param game: The game which this belongs to
|
||
|
|
:type game: Game
|
||
|
|
:param image_name: The name of the image to use for this sprite
|
||
|
|
:type image_name: str
|
||
|
|
:param attributes: The attributes to use for this object
|
||
|
|
:type attributes: ShooterAttributes
|
||
|
|
"""
|
||
|
|
super().__init__(game, image_name, hp=attributes.hp)
|
||
|
|
self.lazers: List[Lazer] = []
|
||
|
|
self.attributes = attributes
|
||
|
|
self.last_shot = self.game.alpha
|
||
|
|
|
||
|
|
def shoot(self):
|
||
|
|
"""Soot a lazer if possible"""
|
||
|
|
next_shot = self.last_shot + self.attributes.cooldown
|
||
|
|
|
||
|
|
if not self.destroyed \
|
||
|
|
and self.game.alpha > next_shot:
|
||
|
|
self.last_shot = self.game.alpha
|
||
|
|
|
||
|
|
lazer = Lazer(self.game,
|
||
|
|
velocity=self.attributes.velocity,
|
||
|
|
color=self.attributes.lazer_color)
|
||
|
|
lazer.set_pos((self.x + self.w//2 - 1, self.y +
|
||
|
|
self.h//2 - 1))
|
||
|
|
lazer.show()
|
||
|
|
self.lazers.append(lazer)
|
||
|
|
|
||
|
|
def tick(self):
|
||
|
|
"""Update this object"""
|
||
|
|
super().tick()
|
||
|
|
for lazer in self.lazers:
|
||
|
|
lazer.tick()
|
||
|
|
|
||
|
|
self.lazers = Sprite.remove_destroyed(self.lazers)
|
||
|
|
|
||
|
|
def destroy(self):
|
||
|
|
"""Remove all the associated objects"""
|
||
|
|
super().destroy()
|
||
|
|
for lazer in self.lazers:
|
||
|
|
self.game.sprites.append(lazer)
|