TorstenHettstedt/BattleTech
1
0
Fork 0
Code Issues 3 Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: kleine Korrekturen

Browse Source
This commit is contained in:
TorstenHettstedt
2026-06-29 16:25:03 +02:00
parent d64a3eb5ff
commit 472b928950
4 changed files with 13 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.gitignore
+2 -1
View File
@@ -1 +1,2 @@
/venv/ /.venv/
/.idea/
MechFabrik/bauteile/__init__.py
-1
View File
@@ -1,5 +1,4 @@
import enum import enum
from typing import Any
class ZoneNames(enum.Enum): class ZoneNames(enum.Enum):
MechFabrik/reader.py
+4 -7
View File
@@ -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
tests/test_mech_process.py
-2
View File
@@ -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