forked from CatalinStefan/python-design-patterns
-
Notifications
You must be signed in to change notification settings - Fork 0
/
flyweight.py
69 lines (49 loc) · 1.35 KB
/
flyweight.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
import random
from abc import ABC, abstractmethod
class Sprite(ABC):
@abstractmethod
def draw(self):
pass
@abstractmethod
def move(self, x: int, y: int):
pass
class FighterRank:
private = 0
sergeant = 1
major = 2
class Fighter(Sprite):
def __init__(self, rank: FighterRank):
self.rank = rank
def draw(self):
print(f"Drawing fighter {self}")
def move(self, x: int, y: int):
print(f"Moving fighter {self} to position {x}, {y}")
class FighterFactory:
def __init__(self):
self.fighters = []
def get_fighter(self, rank: FighterRank):
f = self.fighters[rank]
if not f:
f = Fighter(rank)
self.fighters[rank] = f
return f
class Army:
army = []
def spawn_fighter(self, rank: FighterRank):
self.army.append(Fighter(rank))
def draw_army(self):
for fighter in self.army:
if fighter.rank == FighterRank.major:
print("M ", end="")
elif fighter.rank == FighterRank.sergeant:
print("S ", end="")
else:
print("P ", end="")
if __name__ == '__main__':
# army_size = 1000
army_size = 1000000
army = Army()
for i in range(army_size):
r = random.randrange(3)
army.spawn_fighter(r)
army.draw_army()