Class AdversarySchema

app/objects/c_adversary.py:12–51

class AdversarySchema(ma.Schema):

    class Meta:
        unknown = ma.EXCLUDE

    adversary_id = ma.fields.String()
    name = ma.fields.String()
    description = ma.fields.String()
    atomic_ordering = ma.fields.List(ma.fields.String())
    objective = ma.fields.String()
    tags = ma.fields.List(ma.fields.String(), allow_none=True)
    has_repeatable_abilities = ma.fields.Boolean(dump_only=True)
    plugin = ma.fields.String(load_default=None)

    @ma.pre_load
    def fix_id(self, adversary, **_):
        if 'id' in adversary:
            adversary['adversary_id'] = adversary.pop('id')
        return adversary

    @ma.pre_load
    def phase_to_atomic_ordering(self, adversary, **_):
        """
        Convert legacy adversary phases to atomic ordering
        """
        if 'phases' in adversary and 'atomic_ordering' in adversary:
            raise ma.ValidationError('atomic_ordering and phases cannot be used at the same time', 'phases', adversary)
        elif 'phases' in adversary:
            adversary['atomic_ordering'] = [ab_id for phase in adversary.get('phases', {}).values() for ab_id in phase]
            del adversary['phases']
        return adversary

    @ma.pre_load
    def remove_properties(self, data, **_):
        data.pop('has_repeatable_abilities', None)
        return data

    @ma.post_load
    def build_adversary(self, data, **kwargs):
        return None if kwargs.get('partial') is True else Adversary(**data)


class Adversary(FirstClassObjectInterface, BaseObject):