refactor: kleine Korrekturen
This commit is contained in:
+2
-1
@@ -1 +1,2 @@
|
|||||||
/venv/
|
/.venv/
|
||||||
|
/.idea/
|
||||||
@@ -1,5 +1,4 @@
|
|||||||
import enum
|
import enum
|
||||||
from typing import Any
|
|
||||||
|
|
||||||
|
|
||||||
class ZoneNames(enum.Enum):
|
class ZoneNames(enum.Enum):
|
||||||
@@ -103,8 +102,8 @@ class Zone:
|
|||||||
|
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
self.__dict__['_armor'] = 0
|
self.__dict__['_armor'] = 0
|
||||||
self.__dict__['_parent']= None
|
self.__dict__['_parent'] = None
|
||||||
self.__dict__['_state']= AreaStatus.FINE
|
self.__dict__['_state'] = AreaStatus.FINE
|
||||||
|
|
||||||
def __setattr__(self, key, value):
|
def __setattr__(self, key, value):
|
||||||
self.__dict__['_' + key] = value
|
self.__dict__['_' + key] = value
|
||||||
@@ -117,17 +116,17 @@ class Area(Zone):
|
|||||||
|
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
super().__init__()
|
super().__init__()
|
||||||
self.__dict__['_outer_amor']= list()
|
self.__dict__['_outer_amor'] = list()
|
||||||
self.__dict__['_children']= list()
|
self.__dict__['_children'] = list()
|
||||||
self.__dict__['_items']= dict()
|
self.__dict__['_items'] = dict()
|
||||||
|
|
||||||
|
|
||||||
class OuterAmor(Zone):
|
class OuterAmor(Zone):
|
||||||
|
|
||||||
def __init__(self, area: Area):
|
def __init__(self, area: Area):
|
||||||
super().__init__()
|
super().__init__()
|
||||||
self.__dict__['_parent']= area
|
self.__dict__['_parent'] = area
|
||||||
self.__dict__['_area']= area
|
self.__dict__['_area'] = area
|
||||||
|
|
||||||
|
|
||||||
class StrikeEvent:
|
class StrikeEvent:
|
||||||
|
|||||||
@@ -15,18 +15,19 @@ class CsvDialect (csv.Dialect):
|
|||||||
class ComponentReader:
|
class ComponentReader:
|
||||||
|
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
|
my_dirname = str(os.path.dirname(__file__))
|
||||||
self._inner_structure_data = {}
|
self._inner_structure_data = {}
|
||||||
self._reactor_data = {}
|
self._reactor_data = {}
|
||||||
self._weapon_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)
|
component_reader = csv.DictReader(csvfile, dialect=CsvDialect)
|
||||||
for row in component_reader:
|
for row in component_reader:
|
||||||
self._inner_structure_data[float(row['GTonnage'])] = row
|
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)
|
component_reader = csv.DictReader(csvfile, dialect=CsvDialect)
|
||||||
for row in component_reader:
|
for row in component_reader:
|
||||||
self._reactor_data[float(row['RW'])] = row
|
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)
|
component_reader = csv.DictReader(csvfile, dialect=CsvDialect)
|
||||||
for row in component_reader:
|
for row in component_reader:
|
||||||
self._weapon_data[row['Waffentyp']] = row
|
self._weapon_data[row['Waffentyp']] = row
|
||||||
@@ -43,13 +44,9 @@ class ComponentReader:
|
|||||||
return 3
|
return 3
|
||||||
case AreaNames.MIDDLE_TORSO:
|
case AreaNames.MIDDLE_TORSO:
|
||||||
return int(self._inner_structure_data.get(tonnage, {}).get('TorsoMitte', 0))
|
return int(self._inner_structure_data.get(tonnage, {}).get('TorsoMitte', 0))
|
||||||
pass
|
|
||||||
case AreaNames.RIGHT_TORSO | AreaNames.LEFT_TORSO:
|
case AreaNames.RIGHT_TORSO | AreaNames.LEFT_TORSO:
|
||||||
return int(self._inner_structure_data.get(tonnage, {}).get('TorsoSeite', 0))
|
return int(self._inner_structure_data.get(tonnage, {}).get('TorsoSeite', 0))
|
||||||
pass
|
|
||||||
case AreaNames.RIGHT_ARM | AreaNames.LEFT_ARM:
|
case AreaNames.RIGHT_ARM | AreaNames.LEFT_ARM:
|
||||||
return int(self._inner_structure_data.get(tonnage, {}).get('Arm', 0))
|
return int(self._inner_structure_data.get(tonnage, {}).get('Arm', 0))
|
||||||
pass
|
|
||||||
case AreaNames.RIGHT_LEG | AreaNames.LEFT_LEG:
|
case AreaNames.RIGHT_LEG | AreaNames.LEFT_LEG:
|
||||||
return int(self._inner_structure_data.get(tonnage, {}).get('Bein', 0))
|
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 test_mech import AbstractTestMechAmor, INNER_ARMOR, NEXT_OUTER_ZONE, OUTER_ARMOR
|
||||||
from MechFabrik.bauteile import Area, AreaStatus, OuterAmor, StrikeEvent, ZoneNames
|
from MechFabrik.bauteile import Area, AreaStatus, OuterAmor, StrikeEvent, ZoneNames
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user