def set_default_user_plan(sender, instance, created, **kwargs):
if created:
default_plan = Plan.get_default_plan()
if default_plan is not None:
UserPlan.objects.create(user=instance,
plan=default_plan,
active=True,
expire=datetime.utcnow().replace(tzinfo=utc) +
timedelta(days=getattr(settings, 'PLAN_DEFAULT_GRACE_PERIOD', 30)))
def set_default_user_plan(sender, instance, created, **kwargs):
"""
Creates default plan for the new user but also extending an account for default grace period.
"""
if created:
default_plan = Plan.get_default_plan()
if default_plan is not None:
UserPlan.objects.create(user=instance, plan=default_plan, active=False, expire=None)
def set_default_buyer_plan(sender, instance, created, **kwargs):
"""
Creates default plan for the new buyer but also extending an account for default grace period.
"""
if created:
default_plan = Plan.get_default_plan()
if default_plan is not None:
BuyerPlan.objects.create(buyer=instance,
plan=default_plan,
active=False,
expire=None)
def test_get_user_quota_expired_no_default(self):
u = User.objects.get(username='******')
u.userplan.expire = date.today() - timedelta(days=5)
Plan.get_default_plan().delete()
with self.assertRaises(ValidationError):
get_user_quota(u)
