Merge pull request momosecurity#59 from Flynnon/change_small

修改一些格式问题 + 删除策略时提示语优化

risk_models/cache.py

@@ -1,14 +1,13 @@
 # coding=utf8
+import logging
 import os
 import random
-import logging
 from collections import defaultdict
 
 import gevent
 
 from clients.redis_client import get_config_redis_client
 
-
 logger = logging.getLogger(__name__)
 
 

www/core/forms.py

@@ -1,12 +1,11 @@
 # coding=utf8
 from datetime import datetime, timedelta
 
-from django import forms
-from django.utils.translation import ugettext_lazy as _
-
 from crispy_forms.bootstrap import FormActions
 from crispy_forms.helper import FormHelper
-from crispy_forms.layout import Submit, Layout, Div, HTML, Reset, Field
+from crispy_forms.layout import Submit, Layout, Reset, Field
+from django import forms
+from django.utils.translation import ugettext_lazy as _
 
 
 class FormBaseMixin(object):
@@ -75,9 +74,15 @@ def __init__(self, *args, **kwargs):
             self.fields[field].widget.attrs["class"] = "form-control"
 
     def remove_class(self, *field_names):
+        self.change_class(*field_names, class_name=None)
+
+    def change_class(self, *field_names, class_name=None):
         for field_name in field_names:
             if field_name in self.fields:
-                self.fields[field_name].widget.attrs.pop('class')
+                if class_name is None:
+                    self.fields[field_name].widget.attrs.pop('class')
+                else:
+                    self.fields[field_name].widget.attrs['class'] = class_name
 
     def is_edit(self):
         return self.instance.pk is not None