diff --git a/packages/ns-api/files/ns.threatshield b/packages/ns-api/files/ns.threatshield index ca6ee34a5..be8ed3de1 100644 --- a/packages/ns-api/files/ns.threatshield +++ b/packages/ns-api/files/ns.threatshield @@ -133,8 +133,10 @@ def list_dns_feeds(enterprise=False): return ret -def get_confidence(f): +def get_confidence(f, enterprise=False): confidence = -1 + if not enterprise: + return confidence if f.endswith('lvl1') or f.endswith('level1'): confidence = 10 elif f.endswith('lvl2') or f.endswith('level2'): @@ -169,7 +171,7 @@ def list_blocklist(e_uci): if type == 'enterprise' and not has_bl: continue - ret.append({ 'name': f, 'type': type, 'enabled': enabled, 'confidence': get_confidence(f), 'description': feed.get('descr')}) + ret.append({ 'name': f, 'type': type, 'enabled': enabled, 'confidence': get_confidence(f, type == 'enterprise'), 'description': feed.get('descr')}) return { "data": ret } def list_settings(e_uci): @@ -406,7 +408,7 @@ def dns_list_blocklist(e_uci): if type == 'enterprise' and not has_bl: continue - ret.append({ 'name': f, 'type': type, 'enabled': enabled, 'confidence': get_confidence(f), 'description': feed.get('focus')}) + ret.append({ 'name': f, 'type': type, 'enabled': enabled, 'confidence': get_confidence(f, type == 'enterprise'), 'description': feed.get('focus')}) return { "data": ret } def dns_edit_blocklist(e_uci, payload):