def __init__(self, *args, **kwargs): super(ResourceForm, self).__init__(*args, **kwargs) if self.instance.pk and self.instance.published_date: p_date = self.instance.published_date if p_date.month == 7 and p_date.day == 2: self.initial["published_date"] = get_period(period_for_date("yearly", date=p_date)).get_display() else: self.initial["published_date"] = get_period(period_for_date("monthly", date=p_date)).get_display()
def __init__(self, *args, **kwargs): super(ResourceForm, self).__init__(*args, **kwargs) if self.instance.pk and self.instance.published_date: p_date = self.instance.published_date if p_date.month == 7 and p_date.day == 2: self.initial["published_date"] = get_period( period_for_date("yearly", date=p_date)).get_display() else: self.initial["published_date"] = get_period( period_for_date("monthly", date=p_date)).get_display()
def get_queryset(self): queryset = super(TweetListView, self).get_queryset() period = self.request.GET.get("period") if period: queryset = queryset.filter( timestamp__range=get_period(period).get_start_end()) return queryset
def test_yearly_sub_periods_quarterly_all(self): periods = get_period("Y-2015").sub_periods("quarterly") self.assertEquals(len(periods), 4) self.assertEquals(periods[0].raw_value, "Q-2015-1") self.assertEquals(periods[1].raw_value, "Q-2015-2") self.assertEquals(periods[2].raw_value, "Q-2015-3") self.assertEquals(periods[3].raw_value, "Q-2015-4")
def get_queryset(self): queryset = super(TweetListView, self).get_queryset() period = self.request.GET.get("period") if period: queryset = queryset.filter( timestamp__range=get_period(period).get_start_end() ) return queryset
def clean_published_date(self): period_raw = self.cleaned_data["published_date"] if len(period_raw): p = get_period(period_raw) start_date, end_date = p.get_start_end() if p.is_period_type("yearly"): the_date = datetime.date(start_date.year, 7, 2) else: the_date = start_date return str(the_date) else: return ""
def test_yearly_sub_periods_monthly_all(self): periods = get_period("Y-2015").sub_periods("monthly") self.assertEquals(len(periods), 12) self.assertEquals(periods[0].get_display(), "January 2015") self.assertEquals(periods[1].get_display(), "February 2015") self.assertEquals(periods[2].get_display(), "March 2015") self.assertEquals(periods[3].get_display(), "April 2015") self.assertEquals(periods[4].get_display(), "May 2015") self.assertEquals(periods[5].get_display(), "June 2015") self.assertEquals(periods[6].get_display(), "July 2015") self.assertEquals(periods[7].get_display(), "August 2015") self.assertEquals(periods[8].get_display(), "September 2015") self.assertEquals(periods[9].get_display(), "October 2015") self.assertEquals(periods[10].get_display(), "November 2015") self.assertEquals(periods[11].get_display(), "December 2015")
def test_non_equality_false(self): self.assertFalse(self.quarter_1 != get_period("Q-2015-1"))
def test_yearly_sub_periods_weekly_all(self): periods = get_period("Y-2015").sub_periods("weekly") self.assertEquals(len(periods), 53) for i in range(53): self.assertEquals(periods[i].raw_value, "W-2015-{:02d}".format(i + 1))
def test_monthly_sub_periods_weekly_all(self): periods = get_period("M-2015-01").sub_periods("weekly") self.assertEquals(len(periods), 5) for i in range(5): self.assertEquals(periods[i].raw_value, "W-2015-{:02d}".format(i + 1))
def setUp(self): self.quarter_1 = get_period("Q-2015-1") self.quarter_2 = get_period("Q-2015-2") self.year = get_period("Y-2015")
def test_get_period_validation(self): with self.assertRaises(ValidationError): get_period("2013W22")
def test_get_period_str(self): period = get_period("M-2015-01") self.assertEquals(str(period), "M-2015-01")
def test_equality_true(self): self.assertTrue(self.quarter_1 == get_period("Q-2015-1"))
def test_includes_4(self): self.assertTrue(self.quarter_1.includes(get_period("Q-2015-1")))
def test_past(self): period = get_period("M-2013-11") self.assertTrue(period.is_past())
def test_current(self): period = get_period(PERIOD_TYPES["quarterly"].for_date(timezone.now())) self.assertTrue(period.is_current())
def test_get_display(self): self.assertEquals(get_period("M-2015-04").get_display(), "April 2015")