Merge fdd9b26 into 8eee1df

baseframe/forms/validators.py

@@ -26,6 +26,7 @@
 from lxml import html
 from pyisemail import is_email
 import dns.resolver
+import emoji
 import requests
 
 from coaster.utils import deobfuscate_email, make_name
@@ -299,6 +300,19 @@ class NotEqualTo(_Comparison):
     def compare(self, value, other):
         return value != other
 
+class IsEmoji(object):
+    """
+    Validate field to contain a single emoji.
+
+    :param message:
+        Error message to raise in case of a validation error.
+    """
+    def __init__(self, message=None):
+        self.message = message or _(u"This is not a valid emoji.")
+
+    def __call__(self, form, field):
+        if field.data not in emoji.UNICODE_EMOJI:
+            raise ValidationError(self.message)
 
 class IsPublicEmailDomain(object):
     """

setup.py

@@ -23,6 +23,7 @@
     'pytz',
     'pyIsEmail',
     'dnspython',
+    'emoji',
     'WTForms>=2.2',
     'Flask>=1.0',
     'Flask-Assets',

tests/fixtures.py

@@ -57,6 +57,9 @@ class UrlFormTest(forms.Form):
         filters=[forms.filters.strip()],
     )
 
+class EmojiFormTest(forms.Form):
+    emoji = forms.StringField(__("Emoji"), validators=[forms.validators.IsEmoji()])
+
 
 class AllUrlsFormTest(forms.Form):
     content_with_urls = forms.TextAreaField(

tests/test_validators.py

@@ -12,6 +12,7 @@
 
 from .fixtures import (
     AllUrlsFormTest,
+    EmojiFormTest,
     PublicEmailDomainFormTest,
     TestCaseBaseframe,
     UrlFormTest,
@@ -23,6 +24,7 @@ def setUp(self):
         super(TestValidators, self).setUp()
         with self.app.test_request_context('/'):
             self.form = UrlFormTest(meta={'csrf': False})
+            self.emoji_form = EmojiFormTest(meta={'csrf': False})
             self.all_urls_form = AllUrlsFormTest(meta={'csrf': False})
             self.webmail_form = PublicEmailDomainFormTest(meta={'csrf': False})
         urllib3.disable_warnings()
@@ -42,6 +44,18 @@ def test_invalid_url(self):
             url = 'https://hasgeek'
             self.form.process(url=url)
             assert not self.form.validate()
+
+    def test_valid_emoji(self):
+        with self.app.test_request_context('/'):
+            dat = u'👍'
+            self.emoji_form.process(emoji=dat)
+            assert self.emoji_form.validate() is True
+
+    def test_invalid_emoji(self):
+        with self.app.test_request_context('/'):
+            dat = 'eviltext'
+            self.emoji_form.process(emoji=dat)
+            assert self.emoji_form.validate() is False
 
     def test_public_email_domain(self):
         with self.app.test_request_context('/'):