* some fixes

akme · akme · commit bc5c0667dd89 · 2018-04-11T16:15:53.000+03:00
diff --git a/README.md b/README.md
@@ -5,13 +5,13 @@ A small utility to convert Unix `lsof` output to a graph showing FIFO and UNIX i
 Generate graph:
 
 ````shell
-sudo lsof -n -F | ./lsofgraph | dot -Tjpg > /tmp/a.jpg
+sudo lsof -n -F | python lsofgraph.py | dot -Tjpg > /tmp/a.jpg
 ````
 
 or add `unflatten` to the chain for a better layout:
 
 ````shell
-sudo lsof -n -F | ./lsofgraph | unflatten -l 1 -c 6 | dot -T jpg > /tmp/a.jpg
+sudo lsof -n -F | python lsofgraph.py | unflatten -l 1 -c 6 | dot -T jpg > /tmp/a.jpg
 ````
 
 ![example output](/example.jpg)
diff --git a/lsofgraph.py b/lsofgraph.py
@@ -43,7 +43,6 @@ def parse_lsof():
                     proc = {}
                     file = None
                     cur = None
-
     return procs
 
 
@@ -96,7 +95,7 @@ def find_connections(procs):
                             else:
                                 fs[id] = []
                                 fs[id].append(file)
-                                
+
                 if 't' in file and (file['t'] == 'IPv4' or file['t'] == 'IPv6'):
                     if '->' in file['n']:
                         a, b = file['n'].split("->")
@@ -118,7 +117,6 @@ def find_connections(procs):
                                 else:
                                     fs[id] = []
                                     fs[id].append(file)
-
     return cs
 
 
@@ -128,7 +126,7 @@ def print_graph(procs, conns):
         'unix': "purple",
         'tcp': "red",
         'udp': "orange",
-        'pipe': "orange"
+        'pipe': "blue"
     }
 
     # Generate graph
@@ -175,6 +173,7 @@ def print_graph(procs, conns):
     print("}")
 
 
-procs = parse_lsof()
-conns = find_connections(procs)
-print_graph(procs, conns)
+if __name__ == '__main__':
+    procs = parse_lsof()
+    conns = find_connections(procs)
+    print_graph(procs, conns)
