Convert the two_state_widget to a class and update it to use the new icon class.

Signed-off-by: Roberto Rosario <roberto.rosario.gonzalez@gmail.com>

mayan/apps/common/icons.py

@@ -16,6 +16,9 @@ icon_current_user_locale_profile_edit = Icon(
     driver_name='fontawesome', symbol='globe'
 )
 icon_documentation = Icon(driver_name='fontawesome', symbol='book')
+icon_fail = Icon(
+    driver_name='fontawesome', symbol='times'
+)
 icon_forum = Icon(
     driver_name='fontawesome', symbol='life-ring'
 )
@@ -31,6 +34,9 @@ icon_menu_user = Icon(
 icon_object_error_list_with_icon = Icon(
     driver_name='fontawesome', symbol='lock'
 )
+icon_ok = Icon(
+    driver_name='fontawesome', symbol='check'
+)
 icon_packages_licenses = Icon(
     driver_name='fontawesome', symbol='certificate'
 )
@@ -43,3 +49,4 @@ icon_support = Icon(
 icon_tools = Icon(
     driver_name='fontawesome', symbol='wrench'
 )
+

mayan/apps/common/widgets.py

@@ -5,6 +5,8 @@ from django.utils.encoding import force_text
 from django.utils.html import format_html
 from django.utils.safestring import mark_safe
 
+from icons import icon_fail as default_icon_fail, icon_ok as default_icon_ok
+
 
 class DisableableSelectWidget(forms.SelectMultiple):
     allow_multiple_selected = True
@@ -76,8 +78,22 @@ class TextAreaDiv(forms.widgets.Widget):
         super(TextAreaDiv, self).__init__(default_attrs)
 
 
-def two_state_template(state, ok_icon='fa fa-check', fail_icon='fa fa-times'):
-    if state:
-        return mark_safe('<i class="text-success {}"></i>'.format(ok_icon))
-    else:
-        return mark_safe('<i class="text-danger {}"></i>'.format(fail_icon))
+class TwoStateWidget(object):
+    def __init__(self, state, icon_ok=None, icon_fail=None):
+        self.state = state
+        self.icon_ok = icon_ok or default_icon_ok
+        self.icon_fail = icon_fail or default_icon_fail
+
+    def render(self):
+        if self.state:
+            return mark_safe(
+                '<div class="text-success">{}</div>'.format(
+                    self.icon_ok.render()
+                )
+            )
+        else:
+            return mark_safe(
+                '<div class="text-danger">{}</div>'.format(
+                    self.icon_fail.render()
+                )
+            )