game_reader.py

import xml.etree.ElementTree as ET

from constants import Const
from place import Place
from character import Enemy
from objects import Weapon
from objects import Item
from objects import Shield


class GameReader:

    def parse_game(self, file):
        tree = ET.parse(file)
        root = tree.getroot()

        # Get all places
        for place in root.iter('place'):
            self.place_list.append(self.get_place(place))

    def get_place(self, place):
        place_name = place.find(self.CONST.NAME_TAG).text
        description = place.find(self.CONST.DESCRIPTION_TAG).text
        place_id = int(place.find('id').text)

        connection_list = []
        for connection in place.iter('connection'):
            aux = int(connection.text)
            connection_list.append(aux)

        objects = place.find('objects')
        if objects is not None:
            object_list = self.get_objects(objects)
        else:
            object_list = []
        enemy_list = []
        for enemy in place.iter('enemy'):
            enemy_list.append(self.get_enemy(enemy))

        p = Place(place_name, description, place_id, connection_list, object_list, enemy_list)
        return p

    @staticmethod
    def get_objects(place):
        object_list = []
        for obj in place.iter('weapon'):
            attributes = obj.attrib
            name = attributes['name']
            description = attributes['description']
            attack = int(attributes['attack'])
            object_list.append(Weapon(name, description, attack))
        for obj in place.iter('object'):
            attributes = obj.attrib
            name = attributes['name']
            description = attributes['description']
            object_list.append(Item(name, description))
        for obj in place.iter('shield'):
            attributes = obj.attrib
            name = attributes['name']
            description = attributes['description']
            defense = int(attributes['defense'])
            object_list.append(Shield(name, description, defense))
        return object_list

    def get_enemy(self, enemy):
        name = enemy.find(self.CONST.NAME_TAG).text
        description = enemy.find(self.CONST.DESCRIPTION_TAG).text
        level = int(enemy.find(self.CONST.LEVEL_TAG).text)
        # location = int(enemy.find(self.CONST.LOCATION_TAG).text)
        object_list = self.get_objects(enemy)
        return Enemy(name, description, level, object_list)

    def __init__(self):
        self.place_list = []
        self.enemy_list = []
        self.CONST = Const()