Linting

squirrelbattle/mapgeneration/randomwalk.py

@@ -1,15 +1,22 @@
+# Copyright (C) 2020 by ÿnérant, eichhornchen, nicomarg, charlse
+# SPDX-License-Identifier: GPL-3.0-or-later
+
 from enum import auto, Enum
 from random import choice, random, randint
+from typing import Tuple
+
 from ..interfaces import Map, Tile
 
 
-DEFAULT_PARAMS = {"split_chance" : .15,
+DEFAULT_PARAMS = {
+    "split_chance": .15,
     "turn_chance": .5,
     "death_chance": .1,
     "max_walkers": 15,
     "width": 100,
     "height": 100,
-        "fill" : .4}
+    "fill": .4,
+}
 
 
 class Directions(Enum):
@@ -20,16 +27,15 @@ class Directions(Enum):
 
 
 class Walker:
-
+    def __init__(self, x: int, y: int):
-    def __init__(self, x, y):
         self.x = x
         self.y = y
         self.dir = choice(list(Directions))
 
-    def random_turn(self):
+    def random_turn(self) -> None:
         self.dir = choice(list(Directions))
 
-    def next_pos(self):
+    def next_pos(self) -> Tuple[int, int]:
         if self.dir == Directions.up:
             return self.x, self.y + 1
         elif self.dir == Directions.down:
@@ -39,23 +45,24 @@ class Walker:
         elif self.dir == Directions.left:
             return self.x - 1, self.y
 
-    def move_in_bounds(self, width, height):
+    def move_in_bounds(self, width: int, height: int) -> None:
         nx, ny = self.next_pos()
         if 0 < nx < width and 0 < ny < height:
             self.x, self.y = nx, ny
 
-    def split(self):
+    def split(self) -> "Walker":
         child = Walker(self.x, self.y)
         child.dir = self.dir
         return child
 
 
 class Generator:
-
+    def __init__(self, params: dict = None):
-    def __init__(self, params = DEFAULT_PARAMS):
+        if params is None:
+            params = DEFAULT_PARAMS
         self.params = params
 
-    def run(self):
+    def run(self) -> Map:
         width, height = self.params["width"], self.params["height"]
         walkers = [Walker(width // 2, height // 2)]
         grid = [[Tile.EMPTY for _ in range(width)] for _ in range(height)]
@@ -77,7 +84,7 @@ class Generator:
                     next_walker_pop.append(walker)
 
             # we make sure to never kill all walkers
-            if next_walker_pop == []:
+            if not next_walker_pop:
                 next_walker_pop.append(choice(walkers))
 
             # we use a second loop for spliting so we're not bothered by cases
@@ -90,7 +97,6 @@ class Generator:
                         next_walker_pop.append(walker.split())
             walkers = next_walker_pop
 
-
         start_x, start_y = randint(0, width - 1), randint(0, height - 1)
         while grid[start_y][start_x] != Tile.FLOOR:
             start_x, start_y = randint(0, width - 1), randint(0, height - 1)