Add mobreport.py
This commit is contained in:
@@ -1,16 +1,26 @@
|
||||
from tungston.core.fauna.mob import Mob
|
||||
from collections.abc import Iterator
|
||||
from tungston.core.fauna.mobplacement import MobPlacement
|
||||
|
||||
|
||||
# Class ############################################################################################
|
||||
|
||||
class MobCatalog:
|
||||
|
||||
def __init__(self, mobs):
|
||||
self.mobs = mobs
|
||||
def __init__(self, mobs:tuple[MobPlacement]):
|
||||
self._mobs = {}
|
||||
|
||||
def findByName(self, name):
|
||||
for mob in self.mobs:
|
||||
if mob.name == name:
|
||||
return mob
|
||||
for mob in mobs:
|
||||
self.add(mob)
|
||||
|
||||
return None
|
||||
def add(self, placement:MobPlacement) -> "MobCatalog":
|
||||
if placement.mob.name in self._mobs:
|
||||
raise Exception("Catalog already has an entry for {mob.name}")
|
||||
|
||||
self._mobs[placement.mob.name] = placement
|
||||
|
||||
def all(self) -> Iterator[MobPlacement]:
|
||||
for placement in self._mobs.values():
|
||||
yield placement
|
||||
|
||||
def get(self, name:str) -> MobPlacement:
|
||||
return self._mobs.get(name, None)
|
||||
|
||||
Reference in New Issue
Block a user