def handle(self, **options):
verbose = int(options.get('verbosity'))
count = options.get('pool_size')
CaptchaStore.create_pool(count)
verbose and self.stdout.write('Created %d new captchas\n' % count)
options.get('cleanup_expired') and CaptchaStore.remove_expired()
options.get('cleanup_expired') and verbose and self.stdout.write('Expired captchas cleaned up\n')
def handle(self, **options):
verbose = int(options.get("verbosity"))
count = options.get("pool_size")
CaptchaStore.create_pool(count)
verbose and self.stdout.write("Created %d new captchas\n" % count)
options.get("cleanup_expired") and CaptchaStore.remove_expired()
options.get("cleanup_expired") and verbose and self.stdout.write(
"Expired captchas cleaned up\n")
def handle(self, **options):
verbose = int(options.get('verbosity'))
count = options.get('pool_size')
CaptchaStore.create_pool(count)
verbose and self.stdout.write('Created %d new captchas\n' % count)
options.get('cleanup_expired') and CaptchaStore.remove_expired()
options.get('cleanup_expired') and verbose and self.stdout.write(
'Expired captchas cleaned up\n')
def test_get_from_pool(self):
__current_test_get_from_pool_setting = settings.CAPTCHA_GET_FROM_POOL
__current_test_get_from_pool_timeout_setting = settings.CAPTCHA_GET_FROM_POOL_TIMEOUT
__current_test_timeout_setting = settings.CAPTCHA_TIMEOUT
settings.CAPTCHA_GET_FROM_POOL = True
settings.CAPTCHA_GET_FROM_POOL_TIMEOUT = 5
settings.CAPTCHA_TIMEOUT = 90
CaptchaStore.objects.all().delete() # Delete objects created during SetUp
POOL_SIZE = 10
CaptchaStore.create_pool(count=POOL_SIZE)
self.assertEqual(CaptchaStore.objects.count(), POOL_SIZE)
pool = CaptchaStore.objects.values_list('hashkey', flat=True)
random_pick = CaptchaStore.pick()
self.assertIn(random_pick, pool)
# pick() should not create any extra captcha
self.assertEqual(CaptchaStore.objects.count(), POOL_SIZE)
settings.CAPTCHA_GET_FROM_POOL = __current_test_get_from_pool_setting
settings.CAPTCHA_GET_FROM_POOL_TIMEOUT = __current_test_get_from_pool_timeout_setting
settings.CAPTCHA_TIMEOUT = __current_test_timeout_setting
def test_get_from_pool(self):
__current_test_get_from_pool_setting = settings.CAPTCHA_GET_FROM_POOL
__current_test_get_from_pool_timeout_setting = settings.CAPTCHA_GET_FROM_POOL_TIMEOUT
__current_test_timeout_setting = settings.CAPTCHA_TIMEOUT
settings.CAPTCHA_GET_FROM_POOL = True
settings.CAPTCHA_GET_FROM_POOL_TIMEOUT = 5
settings.CAPTCHA_TIMEOUT = 90
CaptchaStore.objects.all().delete() # Delete objects created during SetUp
POOL_SIZE = 10
CaptchaStore.create_pool(count=POOL_SIZE)
self.assertEqual(CaptchaStore.objects.count(), POOL_SIZE)
pool = CaptchaStore.objects.values_list('hashkey', flat=True)
random_pick = CaptchaStore.pick()
self.assertIn(random_pick, pool)
# pick() should not create any extra captcha
self.assertEqual(CaptchaStore.objects.count(), POOL_SIZE)
settings.CAPTCHA_GET_FROM_POOL = __current_test_get_from_pool_setting
settings.CAPTCHA_GET_FROM_POOL_TIMEOUT = __current_test_get_from_pool_timeout_setting
settings.CAPTCHA_TIMEOUT = __current_test_timeout_setting
