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 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")
