Update alias

src/main.py

@@ -8,6 +8,7 @@
 
 import config
 from objects.port import Port
+from objects.alias import Alias
 from firewall.filtering import Filtering
 from translation.nile import Nile
 
@@ -19,6 +20,19 @@ def grouping(entities, filter_entities):
         else:
             entities[k].append(entity['traffic'])
 
+def aggregation(entities, aliasfile):
+    alias = Alias(aliasfile)
+    aliases = alias.read_aliases()
+    for entity in entities:
+        src, dst, action = entity
+        traffic = entities[(src, dst, action)]
+        del entities[(src, dst, action)]
+        if src in aliases:
+            src = aliases[src]
+        if dst in aliases:
+            dst = aliases[dst]
+        entities[(src, dst, action)] = traffic
+
 def get_path(file):
     return '/'.join(file.split('/')[:-1]) + '/'
 
@@ -47,6 +61,8 @@ def get_path(file):
 
         grouping(entities, filter_entities)
 
+    aggregation(entities, aliasfile)
+
     nile = Nile(entities)
     output = nile.translation()
 

src/objects/alias.py

@@ -0,0 +1,18 @@
+#!/usr/bin/env python3
+
+class Alias:
+    def __init__(self, aliasfile):
+        self.aliasfile = aliasfile
+
+    def read_aliases(self):
+        f = open(self.aliasfile)
+
+        aliases = {}
+
+        for line in f:
+            if line.strip() == "" or line.startswith('#'):
+                continue
+            alias, prefix = line.split()
+            aliases[prefix] = alias
+
+        return aliases