refactor: kleine Korrekturen
This commit is contained in:
+2
-1
@@ -1 +1,2 @@
|
||||
/venv/
|
||||
/.venv/
|
||||
/.idea/
|
||||
@@ -1,5 +1,4 @@
|
||||
import enum
|
||||
from typing import Any
|
||||
|
||||
|
||||
class ZoneNames(enum.Enum):
|
||||
|
||||
@@ -15,18 +15,19 @@ class CsvDialect (csv.Dialect):
|
||||
class ComponentReader:
|
||||
|
||||
def __init__(self):
|
||||
my_dirname = str(os.path.dirname(__file__))
|
||||
self._inner_structure_data = {}
|
||||
self._reactor_data = {}
|
||||
self._weapon_data = {}
|
||||
with open(os.path.join(os.path.dirname(__file__), 'tabellen', 'bauteile.csv')) as csvfile:
|
||||
with open(os.path.join(my_dirname, 'tabellen', 'bauteile.csv')) as csvfile:
|
||||
component_reader = csv.DictReader(csvfile, dialect=CsvDialect)
|
||||
for row in component_reader:
|
||||
self._inner_structure_data[float(row['GTonnage'])] = row
|
||||
with open(os.path.join(os.path.dirname(__file__), 'tabellen', 'reaktoren.csv')) as csvfile:
|
||||
with open(os.path.join(my_dirname, 'tabellen', 'reaktoren.csv')) as csvfile:
|
||||
component_reader = csv.DictReader(csvfile, dialect=CsvDialect)
|
||||
for row in component_reader:
|
||||
self._reactor_data[float(row['RW'])] = row
|
||||
with open(os.path.join(os.path.dirname(__file__), 'tabellen', 'waffen.csv')) as csvfile:
|
||||
with open(os.path.join(my_dirname, 'tabellen', 'waffen.csv')) as csvfile:
|
||||
component_reader = csv.DictReader(csvfile, dialect=CsvDialect)
|
||||
for row in component_reader:
|
||||
self._weapon_data[row['Waffentyp']] = row
|
||||
@@ -43,13 +44,9 @@ class ComponentReader:
|
||||
return 3
|
||||
case AreaNames.MIDDLE_TORSO:
|
||||
return int(self._inner_structure_data.get(tonnage, {}).get('TorsoMitte', 0))
|
||||
pass
|
||||
case AreaNames.RIGHT_TORSO | AreaNames.LEFT_TORSO:
|
||||
return int(self._inner_structure_data.get(tonnage, {}).get('TorsoSeite', 0))
|
||||
pass
|
||||
case AreaNames.RIGHT_ARM | AreaNames.LEFT_ARM:
|
||||
return int(self._inner_structure_data.get(tonnage, {}).get('Arm', 0))
|
||||
pass
|
||||
case AreaNames.RIGHT_LEG | AreaNames.LEFT_LEG:
|
||||
return int(self._inner_structure_data.get(tonnage, {}).get('Bein', 0))
|
||||
return 0
|
||||
|
||||
@@ -1,5 +1,3 @@
|
||||
import unittest
|
||||
|
||||
from test_mech import AbstractTestMechAmor, INNER_ARMOR, NEXT_OUTER_ZONE, OUTER_ARMOR
|
||||
from MechFabrik.bauteile import Area, AreaStatus, OuterAmor, StrikeEvent, ZoneNames
|
||||
|
||||
|
||||
Reference in New Issue
Block a user