IL2HorusTeam/il2fb-ds-config

# coding: utf-8

import zope.interface

from schematics.models import Model
from schematics.types import BooleanType

from .interfaces import INISerializable, DefaultProvider
from .helpers import field_from_ini, field_to_ini


@zope.interface.implementer(INISerializable)
@zope.interface.implementer(DefaultProvider)
class HUD(Model):
    show_mission_info = BooleanType(
        default=True,
        required=True,
    )
    show_kill_info = BooleanType(
        default=True,
        required=True,
    )
    show_at_bottom = BooleanType(
        default=False,
        required=True,
    )

    @classmethod
    def from_ini(cls, ini):
        return cls({
            'show_mission_info': not field_from_ini(
                cls.show_mission_info, ini,
                'game', 'NoMissionInfoHud',
                False,
            ),
            'show_kill_info': not field_from_ini(
                cls.show_kill_info, ini,
                'game', 'noKillInfoHud',
                False,
            ),
            'show_at_bottom': field_from_ini(
                cls.show_at_bottom, ini,
                'game', 'lowInfoHud',
            ),
        })

    def to_ini(self, ini):
        field_to_ini(not self.show_mission_info, ini, 'game', 'NoMissionInfoHud')
        field_to_ini(not self.show_kill_info, ini, 'game', 'noKillInfoHud')
        field_to_ini(self.show_at_bottom, ini, 'game', 'lowInfoHud')

    @classmethod
    def default(cls):
        return cls({
            field_name: field.default
            for field_name, field in cls.fields.items()
        })