pep8 cleanup
[erdslangetjie.git] / erdslangetjie / player.py
1 # Player and Player-like objects
2
3 from erdslangetjie.data import load_image
4
5
6 class GameSprite(object):
7
8     def __init__(self):
9         self.pos = (0, 0)
10         self.sprite = None
11
12     def get_image(self):
13         return self.sprite
14
15     def get_texture(self):
16         return self.sprite.texture
17
18
19 class FigureSprite(GameSprite):
20
21     def can_move(self, direction, level):
22         '''Check if we can move in the given direction'''
23         cand_pos = (self.pos[0] + direction[0], self.pos[1] + direction[1])
24         return not level.blocked(cand_pos)
25
26
27 class ThePlayer(FigureSprite):
28
29     def __init__(self):
30         super(ThePlayer, self).__init__()
31         self.sprite = load_image('sprites/player.png')
32
33     def move(self, direction, level):
34         if self.can_move(direction, level):
35             self.pos = (self.pos[0] + direction[0], self.pos[1] + direction[1])
36             return True
37         return False
38
39
40 class Nemesis(FigureSprite):
41
42     def __init__(self):
43         super(Nemesis, self).__init__()
44         self.sprite = load_image('sprites/nemesis.png')
45         self.reset_pos()
46
47     def move(self, level):
48         if not self.on_board():
49             # Advance towards the map a step at a time
50             self.pos = (self.pos[0] + 1, self.pos[1])
51             if self.on_board():
52                 self.pos = level.enter_pos
53             return False
54         # AI goes here
55         return True
56
57     def reset_pos(self):
58         self.pos = (-10, 0)
59
60     def on_board(self):
61         return self.pos[0] >= 0