enum_and_parser code prettified

Author: Y3drk

data/Records.csv

@@ -335,4 +335,10 @@ ferrari458,map1,8000,Lec
 ferrari458,map1,10051,Lec
 ferrari458,map1,7630,Lec
 ferrari458,map1,7872,Lec
-ferrari458,map1,13118,Lec
+ferrari458,map1,13118,Lec
+supra_new,map1,10830,Gucio
+supra_new,map1,10128,Gucio
+supra_new,map1,10362,Gucio
+supra_new,map1,21638,Gucio
+supra_new,map1,11979,Gucio
+supra_new,map1,15231,Gucio

enums_and_parser/CSVParser.py

@@ -1,4 +1,3 @@
-
 from back_elements.wall import Wall
 from back_elements.surface import Surface
 from enums_and_parser.surfaceType import SurfaceType
@@ -22,14 +21,14 @@ def draw_map(self, map):
         for row in csv_reader:
             rows.append(row)
 
-        #planned CSV structure ->
-        #0 type(string -> what is it, wall / surface),
-        #1 position x,
-        #2 position y,
-        #3 width,
-        #4 height,
-        #5 surfaceType or with_tires to specify
-        #6 rotation -> if it has image, how much is it rotated
+        # planned CSV structure ->
+        # 0 type(string -> what is it, wall / surface),
+        # 1 position x,
+        # 2 position y,
+        # 3 width,
+        # 4 height,
+        # 5 surfaceType or with_tires to specify
+        # 6 rotation -> if it has image, how much is it rotated
         for row in rows:
             position = Vector2D(int(row[1]), int(row[2]))
             width = int(row[3])
@@ -45,7 +44,8 @@ def draw_map(self, map):
             elif row[0] == "SURFACE":
                 if row[5] == "ASPHALT":
                     map.all_surfaces.add(Surface(position, width, height, SurfaceType.ASPHALT, rotation))
-                    map.places_for_boosters.append([position.x + 40, position.y + 40, position.x + width - 40, position.y + height - 40])
+                    map.places_for_boosters.append(
+                        [position.x + 40, position.y + 40, position.x + width - 40, position.y + height - 40])
 
                 elif row[5] == "SNOW":
                     map.all_surfaces.add(Surface(position, width, height, SurfaceType.SNOW, rotation))
@@ -74,12 +74,12 @@ def draw_map(self, map):
 
         file.close()
 
-    def write_to_leaderboard(self,result, name, map, car):
-        #assuming the following schema of the CSV file:
-        #0 - player name
-        #1 - player result(time)
-        #2 - map
-        #3 - car model
+    def write_to_leaderboard(self, result, name, map, car):
+        # assuming the following schema of the CSV file:
+        # 0 - player name
+        # 1 - player result(time)
+        # 2 - map
+        # 3 - car model
 
         file = open(self.leaderboard)
         csv_writer = csv.writer(file)
@@ -96,18 +96,16 @@ def read_leaderboard(self):
         rows = []
 
         for row in csv_reader:
-            row.append(row)
-
-        #now we can do what we want with the rows - it's up to us what
+            rows.append(row)
 
         file.close()
         pass
 
-    def read_car_statistics(self, car_id:int):
-        #planned cars file structure:
-        #[0] - id
-        #[1] - NAME
-        #[2] - engine
+    def read_car_statistics(self, car_id: int):
+        # planned cars file structure:
+        # [0] - id
+        # [1] - NAME
+        # [2] - engine
 
         file = open(self.cars)
         csv_reader = csv.reader(file)
@@ -118,5 +116,4 @@ def read_car_statistics(self, car_id:int):
         for row in csv_reader:
             rows.append(row)
 
-        #we return this data to engine or whatever and it will create a car
         return int(rows[car_id][0]), rows[car_id][1], int(rows[car_id][2])

enums_and_parser/boosterType.py

@@ -2,14 +2,14 @@
 
 
 class BoosterType(Enum):
-    #both positive and negative
+    # both positive and negative
     SPEED = (1, "sp_resized")
     TURNING = (2, "tr_resized")
 
-    #only positive
+    # only positive
     NO_COLLISIONS = (3, "nw_resized")
     DECREASE_TIMER = (4, "dt_resized")
 
-    #only negative
+    # only negative
     NO_TURNING = (5, "nt_resized")
     FREEZE = (6, "fr_resized")

enums_and_parser/surfaceType.py

@@ -2,15 +2,12 @@
 
 
 class SurfaceType(Enum):
-    #in car.move there is friction hardcoded so it will be easy to add these values
-    #but we can also consider adding air friction - wind (given we have enough time)
-    #it would be similar but it would depend on direction of movement 
     ASPHALT = (0.9, (0, 0, 0))  # (fraction, colour) respectively
     SNOW = (0.35, (245, 245, 245))
     ICE = (0.1, (153, 204, 255))
     GRAVEL = (0.4, (204, 102, 0))
     GRASS = (0.5, (0, 204, 0))
     SAND = (0.25, (255, 255, 0))
-    FINISHLINE = (0.9, "./data/finish_line.png") #special case where there will be an image
+    FINISHLINE = (0.9, "./data/finish_line.png")
     SIDE = (0.9, "./data/side.png")
     CHECKPOINT = (0.9, (255, 0, 0))