def save(self, *args, **kwargs):
"""Perform checks before saving."""
# There can only be one.
if self.parent_org is not None and self.parent_org.parent_org is not None:
raise TooManyNestedOrgs
super(Organization, self).save(*args, **kwargs)
# Create a default cycle for the organization if there isn't one already
from seed.models import Cycle
Cycle.get_or_create_default(self)
from seed.models import Measure
Measure.populate_measures(self.id)
def test_default_cycle(self):
year = date.today().year - 1
cycle_name = str(year) + ' Calendar Year'
self.org = Organization.objects.create()
self.assertEqual(self.org.cycles.count(), 1)
cycle = Cycle.objects.filter(name=cycle_name, organization=self.org)
self.assertEqual(self.org.cycles.count(), 1)
cycle.delete()
cycle = Cycle.objects.filter(name=cycle_name, organization=self.org)
self.assertEqual(self.org.cycles.count(), 0)
cycle = Cycle.get_or_create_default(self.org)
self.assertEqual(self.org.cycles.count(), 1)