def check_whitelist(cleaned_data):
cidr = cleaned_data.get('cidr')
source = cleaned_data.get('source')
skip_whitelist = cleaned_data.get('skip_whitelist')
if cidr and not skip_whitelist:
item = is_whitelisted(cidr)
if item:
raise forms.ValidationError("whitelisted: %s: %s" % (item.who, item.why))
if is_prefixlen_too_small(cidr):
raise forms.ValidationError("Prefix length in %s is too small" % cidr)
item = is_source_blacklisted(source)
if item:
raise forms.ValidationError("Source %s is blacklisted: %s: %s" % (source, item.who, item.why))
return cleaned_data
def check_whitelist(cleaned_data):
cidr = cleaned_data.get('cidr')
source = cleaned_data.get('source')
skip_whitelist = cleaned_data.get('skip_whitelist')
if cidr and not skip_whitelist:
item = is_whitelisted(cidr)
if item:
raise forms.ValidationError("whitelisted: %s: %s" % (item.who, item.why))
if is_prefixlen_too_small(cidr):
raise forms.ValidationError("Prefix length in %s is too small" % cidr)
item = is_source_blacklisted(source)
if item:
raise forms.ValidationError("Source %s is blacklisted: %s: %s" % (source, item.who, item.why))
return cleaned_data
def validate(self, attrs):
if attrs.get('duration') and attrs.get('unblock_at'):
raise serializers.ValidationError("Specify only one of duration and unblock_at")
cidr = attrs.get('cidr')
source = attrs.get('source')
skip_whitelist = attrs.get('skip_whitelist')
if cidr and not skip_whitelist:
item = is_whitelisted(cidr)
if item:
raise serializers.ValidationError("whitelisted: %s: %s" % (item.who, item.why))
if is_prefixlen_too_small(cidr):
raise serializers.ValidationError("Prefix length in %s is too small" % cidr)
item = is_source_blacklisted(source)
if item:
raise serializers.ValidationError("Source %s is blacklisted: %s: %s" % (source, item.who, item.why))
return attrs
def validate(self, attrs):
if attrs.get('duration') and attrs.get('unblock_at'):
raise serializers.ValidationError(
"Specify only one of duration and unblock_at")
cidr = attrs.get('cidr')
source = attrs.get('source')
skip_whitelist = attrs.get('skip_whitelist')
if cidr and not skip_whitelist:
item = is_whitelisted(cidr)
if item:
raise serializers.ValidationError("whitelisted: %s: %s" %
(item.who, item.why))
if is_prefixlen_too_small(cidr):
raise serializers.ValidationError(
"Prefix length in %s is too small" % cidr)
item = is_source_blacklisted(source)
if item:
raise serializers.ValidationError(
"Source %s is blacklisted: %s: %s" %
(source, item.who, item.why))
return attrs
def test_source_blacklisted(self):
self.assertEqual(bool(is_source_blacklisted("test")), False)
SourceBlacklistEntry(who=self.user, why='test', source='test').save()
self.assertEqual(bool(is_source_blacklisted("test")), True)
def test_source_blacklisted(self):
self.assertEqual(bool(is_source_blacklisted("test")), False)
SourceBlacklistEntry(who=self.user, why='test', source='test').save()
self.assertEqual(bool(is_source_blacklisted("test")), True)
