forked from s434596/CatOrNot
43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
from flask import current_app, Markup
|
|
from flask import json
|
|
from werkzeug import url_encode
|
|
JSONEncoder = json.JSONEncoder
|
|
|
|
RECAPTCHA_SCRIPT = u'https://www.google.com/recaptcha/api.js'
|
|
|
|
RECAPTCHA_TEMPLATE = u'''
|
|
<script src='%s' async defer></script>
|
|
<div class="g-recaptcha" %s></div>
|
|
'''
|
|
|
|
__all__ = ["RecaptchaWidget"]
|
|
|
|
|
|
class RecaptchaWidget(object):
|
|
|
|
def recaptcha_html(self, public_key):
|
|
html = current_app.config.get('RECAPTCHA_HTML')
|
|
if html:
|
|
return Markup(html)
|
|
params = current_app.config.get('RECAPTCHA_PARAMETERS')
|
|
script = RECAPTCHA_SCRIPT
|
|
if params:
|
|
script += u'?' + url_encode(params)
|
|
|
|
attrs = current_app.config.get('RECAPTCHA_DATA_ATTRS', {})
|
|
attrs['sitekey'] = public_key
|
|
snippet = u' '.join([u'data-%s="%s"' % (k, attrs[k]) for k in attrs])
|
|
return Markup(RECAPTCHA_TEMPLATE % (script, snippet))
|
|
|
|
def __call__(self, field, error=None, **kwargs):
|
|
"""Returns the recaptcha input HTML."""
|
|
|
|
try:
|
|
public_key = current_app.config['RECAPTCHA_PUBLIC_KEY']
|
|
except KeyError:
|
|
raise RuntimeError("RECAPTCHA_PUBLIC_KEY config not set")
|
|
|
|
return self.recaptcha_html(public_key)
|