Upload New File

app/conflict_resolution.py

+import requests
+import time
+from elasticsearch import Elasticsearch
+import get_intents
+
+#function to resolve conflict in policies
+def conflict_fun(ind, policy_dict, workflow_url, stored_intents_url, elasticsearch_url):
+    es = Elasticsearch(elasticsearch_url)
+    # intent_host_arr is the array containing the host(s) for which the intent(s) apply
+    intent_host_arr = policy_dict['host']
+    stored_intents_arr = get_intents.get_intent_fun(stored_intents_url)
+    conflict_res = False
+    if stored_intents_arr[ind]['host'] in intent_host_arr and \
+            stored_intents_arr[ind]['threat'] == policy_dict['threat'] and \
+            int(policy_dict['priority']) < int(stored_intents_arr[ind]['priority']):
+        for i in range(len(intent_host_arr)):
+            if stored_intents_arr[ind]['host'] == intent_host_arr[i] and \
+                    stored_intents_arr[ind]['threat'] == policy_dict['threat'] and \
+                    int(policy_dict['priority']) < int(stored_intents_arr[ind]['priority']):
+                #st_dict = stored_intents_arr[ind]
+                url_to_delete = stored_intents_url + "/" + str(stored_intents_arr[ind]['id'])
+                requests.delete(url_to_delete)
+                conflict_res = True
+                print('a conflict resolved for host: ', stored_intents_arr[ind]['host'])
+                print('')
+            time.sleep(1)
+
+        if conflict_res == True:
+            stored_intents_arr = get_intents.get_intent_fun(stored_intents_url)
+            #repeat the process for the next intent in the intent store
+            #print('ind inside conflict res = true: ', ind)
+            #print('len(resp1[hits hits): ', len(stored_intents_arr))
+            #print('')
+            if (ind < len(stored_intents_arr)):
+                print('starting new conflict resolution')
+                conflict_fun(ind, policy_dict, workflow_url, stored_intents_url,
+                             elasticsearch_url)
+            else:
+                print('conflicts resolved')
+        else:
+            ind = ind + 1
+            stored_intents_arr = get_intents.get_intent_fun(stored_intents_url)
+            #print('ind inside else of conflict res = true: ', ind)
+            #print('len(resp1[hits hits): ', len(stored_intents_arr))
+            #print('')
+            # repeat the process for the next intent in the intent store
+            if (ind < len(stored_intents_arr)):
+                print('starting new conflict resolution')
+                conflict_fun(ind, policy_dict, workflow_url, stored_intents_url,
+                             elasticsearch_url)
+            else:
+                print('conflicts resolved')
+
+    #if an existing intent doesn't conflict with new intent then move to the next intent in the intent store
+    else:
+        ind = ind + 1
+        stored_intents_arr = get_intents.get_intent_fun(stored_intents_url)
+        #print('ind inside else of every: ', ind)
+        #print('len(resp1[hits hits): ', len(stored_intents_arr))
+        #print('')
+        # repeat the process for the next intent in the intent store
+        if (ind < len(stored_intents_arr)):
+            print('starting new conflict resolution')
+            conflict_fun(ind, policy_dict, workflow_url, stored_intents_url,
+                         elasticsearch_url)
+        else:
+            print('ok')
+
+