import logging
import numbers
from pyrpl.modules import Module
from pyrpl.attributes import BoolProperty, FilterProperty, SelectProperty, FloatProperty
from pyrpl.module_attributes import ModuleProperty
from pyrpl.test.test_base import TestPyrpl

logger = logging.getLogger(name=__name__)


class MyFilterProperty(FilterProperty):
    def valid_frequencies(self, module):
        return [2**n for n in range(14)]


class FirstSubModule(Module):
    _setup_attributes = ["b1", "b2"]
    b1 = BoolProperty()
    b2 = BoolProperty()


class SecondSubModule(Module):
    _setup_attributes = ["b1", "b2"]
    b1 = BoolProperty()
    b2 = BoolProperty()


class DummyModule(Module):
    _gui_attributes = ["true_or_false"]
    # it is not necessary to mention the modules here
    # _setup_attributes = ['sub1', 'sub2']
    true_or_false = BoolProperty()
    some_number = FloatProperty(min=-10, max=10, default=1.414)
    some_filter = MyFilterProperty()
    some_options = SelectProperty(options=["foo", "bar"])
    sub1 = ModuleProperty(FirstSubModule)
    sub2 = ModuleProperty(SecondSubModule)


class TestAttributeClass(TestPyrpl):
    # source_config_file = "nosetests_source_dummy_module"

    def test_config_file(self):
        assert "DummyModule" in self.pyrpl.c.pyrpl.modules

    def test_dummy_module(self):
        assert isinstance(self.pyrpl.dummymodule.true_or_false, bool)
        assert isinstance(self.pyrpl.dummymodule.some_number, float)
        assert isinstance(
            self.pyrpl.dummymodule.some_filter, numbers.Number
        )  # should this be a list ?
        assert isinstance(self.pyrpl.dummymodule.some_options, str)  # used to be basestring
        # confirm config file content
        assert self.pyrpl.c.dummymodule.some_number == 3.123, self.pyrpl.c.dummymodule.some_number
        # confirm that loading was done properly (instead of the above default)
        assert self.pyrpl.c.dummymodule.some_number == 3.123, self.pyrpl.c.dummymodule.some_number

    def test_submodule(self):
        self.sub1 = self.pyrpl.dummymodule.sub1
        self.sub2 = self.pyrpl.dummymodule.sub2
        assert self.sub1.b1  # values defined in config file
        assert not self.sub1.b2

        self.sub2.b1 = True
        self.sub2.b2 = False
        assert self.sub2.b1
        assert not self.sub2.b2

        self.pyrpl.dummymodule.save_state("true_false_true_false")

        self.sub1.b1 = False
        self.sub1.b2 = True
        assert not self.sub1.b1
        assert self.sub1.b2

        self.sub2.b1 = False
        self.sub2.b2 = True
        assert not self.sub2.b1
        assert self.sub2.b2

        self.pyrpl.dummymodule.load_state("true_false_true_false")
        assert self.sub1.b1
        assert not self.sub1.b2
        assert self.sub2.b1
        assert not self.sub2.b2