from __future__ import absolute_import, unicode_literals

from datetime import datetime, time

from django.conf import settings
from django.db import models
from django.utils.encoding import python_2_unicode_compatible
from django.utils.translation import ugettext_lazy as _

from modelcluster.models import ClusterableModel
from wagtail.wagtailadmin.edit_handlers import FieldPanel


"""
Model for a new segment
"""
@python_2_unicode_compatible
class Segment(ClusterableModel):
    name = models.CharField(max_length=255)

    panels = [
        FieldPanel('name'),
    ]

    def __str__(self):
        return self.name


"""
Base for creating rules to segment users with
"""
@python_2_unicode_compatible
class AbstractRuleBase(models.Model):
    name = models.CharField(max_length=255)
    segment = models.ForeignKey(to=Segment, related_name="segment")

    def test_user(self, request=None):
        return True

    def return_segment_id(self):
        return "".join(self.name.lower().split())

    def __str__(self):
        return self.name


"""
Time rule to segment users with
"""
@python_2_unicode_compatible
class TimeRule(AbstractRuleBase):
    start_time = models.TimeField(_("Starting time"))
    end_time = models.TimeField(_("Ending time"))

    def __init__(self, *args, **kwargs):
        super(TimeRule, self).__init__(*args, **kwargs)

    def test_user(self, request=None):
        current_time = datetime.now().time()
        starting_time = self.start_time
        ending_time = self.end_time

        if starting_time <= current_time <= ending_time:
            return True
        else:
            return False