[IMP] Some minor improvements.

japinol7 · Apr 30, 2024 · 66b5728 · 66b5728
1 parent f05a3f3
commit 66b5728
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 18 deletions.
diff --git a/codemaster/models/actors/npcs/snakes.py b/codemaster/models/actors/npcs/snakes.py
@@ -298,7 +298,7 @@ def __init__(self, x, y, game, name=None, change_x=0, change_y=0,
 
 
 class SnakeRed(Snake):
-    """Represents a blue red."""
+    """Represents a red snake."""
 
     def __init__(self, x, y, game, name=None, change_x=0, change_y=0,
                  border_left=0, border_right=0,

diff --git a/codemaster/models/special_effects/vortex.py b/codemaster/models/special_effects/vortex.py
@@ -45,8 +45,8 @@ def __init__(self, position, angle, speed, color=Color.WHITE,
             self.draw_method = self.draw_method_circle
 
     def calc_move(self, dt):
-        return [math.cos(self.angle) * self.speed * dt,
-                math.sin(self.angle) * self.speed * dt]
+        return (math.cos(self.angle) * self.speed * dt,
+                math.sin(self.angle) * self.speed * dt)
 
     def set_point_towards(self, angle, rate):
         rotate_dir = ((angle - self.angle + math.pi * 3) % (math.pi * 2)) - math.pi
@@ -69,18 +69,18 @@ def move(self, dt):
             self.alive = False
 
     def draw_method_polygon(self, surface):
-        lateral_point = [
+        lateral_point = (
             self.position[0] + math.cos(self.angle + math.pi / 2) * self.speed * self.scale * 0.3,
-            self.position[1] + math.sin(self.angle + math.pi / 2) * self.speed * self.scale * 0.3]
+            self.position[1] + math.sin(self.angle + math.pi / 2) * self.speed * self.scale * 0.3)
 
-        points = [
-            [self.position[0] + math.cos(self.angle) * self.speed * self.scale,
-             self.position[1] + math.sin(self.angle) * self.speed * self.scale],
+        points = (
+            (self.position[0] + math.cos(self.angle) * self.speed * self.scale,
+             self.position[1] + math.sin(self.angle) * self.speed * self.scale),
             lateral_point,
-            [self.position[0] + math.cos(self.angle) * self.speed * self.scale * 3.5,
-             self.position[1] + math.sin(self.angle) * self.speed * self.scale * 3.5],
+            (self.position[0] + math.cos(self.angle) * self.speed * self.scale * 3.5,
+             self.position[1] + math.sin(self.angle) * self.speed * self.scale * 3.5),
             lateral_point,
-            ]
+            )
         pg.draw.polygon(surface, self.color, points)
 
     def draw_method_circle(self, surface):

diff --git a/codemaster/tools/utils/queue.py b/codemaster/tools/utils/queue.py
@@ -2,12 +2,10 @@
 
 
 class Queue:
-    def __init__(self, iterable=(), maxlen=None, name=''):
+    def __init__(self, iterable=(), maxlen=None, name='', num=0):
         self._container = deque(iterable=iterable, maxlen=maxlen)
         self.name = name
-
-    def __iter__(self):
-        return iter(self._container)
+        self.num = num
 
     @property
     def is_empty(self):
@@ -22,6 +20,9 @@ def pop(self):
     def peek(self):
         return self._container[0] if not self.is_empty else None
 
+    def __iter__(self):
+        return iter(self._container)
+
     def __len__(self):
         return len(self._container)
 

diff --git a/codemaster/tools/utils/stack.py b/codemaster/tools/utils/stack.py
@@ -7,9 +7,6 @@ def __init__(self, iterable=(), maxlen=None, name='', num=0):
         self.name = name
         self.num = num
 
-    def __iter__(self):
-        return iter(self._container)
-
     @property
     def is_empty(self):
         return not self._container
@@ -23,6 +20,9 @@ def pop(self):
     def peek(self):
         return self._container[-1] if not self.is_empty else None
 
+    def __iter__(self):
+        return iter(self._container)
+
     def __len__(self):
         return len(self._container)