Implemented graphviz search using dpkg, it should work for Ubuntu

Author: mmarkov

setup_extra.py

@@ -3,7 +3,7 @@
 """
 Setup helpers for PyGraphviz.
 """
-#    Copyright (C) 2006-2010 by 
+#    Copyright (C) 2006-2015 by
 #    Aric Hagberg <[email protected]>
 #    Dan Schult <[email protected]>
 #    Manos Renieris, http://www.cs.brown.edu/~er/
@@ -14,6 +14,7 @@
 from __future__ import absolute_import
 import subprocess as S
 import sys
+import os
 
 
 def _b2str(buffer):
@@ -23,7 +24,7 @@ def _b2str(buffer):
         if not encoding:
             # can be run without stdout
             if sys.stdout and sys.stdout.encoding:
-                # encoding is nont None only staring Python 3.2
+                # encoding is not None only staring Python 3.2
                 encoding = sys.stdout.encoding
             else:
                 # fall back to default encoding ( normally it should not happen)
@@ -32,23 +33,50 @@ def _b2str(buffer):
             result = buffer.decode(encoding)
         return result
     else:
+        # in Python 2 conversion is implicit
         return buffer
 
 
+def _dpkg_config():
+    # attempt to find graphviz installation with pkg-config
+    # should work with modern versions of graphviz
+    include_dirs=None
+    library_dirs=None
+    
+    try:
+        output = S.check_output(['dpkg', '-S', 'graphviz'])
+        output = _b2str(output)
+        lines = output.split('\n')
+        for line in lines:
+            if not include_dirs and line.endswith('.h'):
+                include_dirs = os.path.dirname(line.split(':')[1].strip())
+                include_dirs = include_dirs.strip() or None
+            if not library_dirs and line.endswith('.so'):
+                library_dirs = os.path.dirname(line.split(':')[1].strip())
+                library_dirs = library_dirs.strip() or None
+            if include_dirs and library_dirs:
+                break
+    except OSError:
+        print("Failed to find dpkg")
+    return include_dirs, library_dirs
+
+
 def _pkg_config():
     # attempt to find graphviz installation with pkg-config
     # should work with modern versions of graphviz
-    library_path=None
     include_path=None
+    library_path=None
     try:
         output = S.check_output(['pkg-config', '--libs-only-L', 'libcgraph'])
         output = _b2str(output)
         if output:
-            library_path=output.strip()[2:]
+            library_path = output.strip()[2:]
+            library_path = library_path.strip() or None
         output = S.check_output(['pkg-config', '--cflags-only-I', 'libcgraph'])
         output = _b2str(output)
         if output:
-            include_path=output.strip()[2:]
+            include_path = output.strip()[2:]
+            include_path = include_path.strip() or None
     except OSError:
         print("Failed to find pkg-config")
     return include_path, library_path
@@ -65,28 +93,35 @@ def _dotneato_config():
         output = _b2str(output)
         if output:
             include_path, library_path = output.split()
-            library_path = library_path.strip()[2:]
-            include_path = include_path.strip()[2:]
+            library_path = library_path.strip()[2:].strip() or None
+            include_path = include_path.strip()[2:].strip() or None
     except OSError:
         print("Failed to find dotneato-config")
         # fall through and test the other syntax
-    if not include_path and not library_path:
+    if not include_path or not library_path:
         try:
             output = S.check_output(['dotneato-config', '--libs', '--cflags'])
             output = _b2str(output)
             if output:
                 include_path, library_path = output.split('\n',1)
-                library_path = library_path.strip()[2:]
-                include_path = include_path.strip()[2:]
+                library_path = library_path or library_path.strip()[2:].strip() or None
+                include_path = include_path or include_path.strip()[2:].strip() or None
         except OSError:
             print("Failed to find dotneato-config")
 
     return include_path, library_path
 
+def _try_configure(include_dirs, library_dirs, try_function):
+    i, l = try_function()
+    i = include_dirs or i
+    l = library_dirs or l
+    return i, l
+
+
 def get_graphviz_dirs():
     """
     First try to read include_dirs from
-    :return: tuple of lists ([include_dirs], [library_dirs], [define_macros])
+    :return: tuple of lists ([include_dirs], [library_dirs])
     """
 
     # If the setup script couldn't find your graphviz installation you can
@@ -116,13 +151,17 @@ def get_graphviz_dirs():
 
     if sys.platform != "win32":
         # Attempt to find Graphviz installation
-        if library_dirs is None and include_dirs is None:
+        if library_dirs is None or include_dirs is None:
+            print("Trying dpkg")
+            include_dirs, library_dirs = _try_configure(include_dirs, library_dirs, _dpkg_config)
+
+        if library_dirs is None or include_dirs is None:
             print("Trying pkg-config")
-            include_dirs, library_dirs = _pkg_config()
+            include_dirs, library_dirs = _try_configure(include_dirs, library_dirs, _pkg_config)
 
-        if library_dirs is None and include_dirs is None:
+        if library_dirs is None or include_dirs is None:
             print("Trying dotneato-config")
-            include_dirs, library_dirs = _dotneato_config()
+            include_dirs, library_dirs = _try_configure(include_dirs, library_dirs, _dotneato_config)
 
         if library_dirs is None or include_dirs is None:
             print()
@@ -140,8 +179,9 @@ def get_graphviz_dirs():
            http://networkx.lanl.gov/pygraphviz/reference/faq.html
 
         If you think your installation is correct you will need to manually
-        change the include_dirs and library_dirs variables in setup.py to
-        point to the correct locations of your graphviz installation.
+        provide path to graphviz include and library. For example:
+
+        pip install pygraphviz --install-option="--include-path=/usr/include/graphviz" --install-option="--library-path=/usr/lib/graphviz/"
 
         The current setting of library_dirs and include_dirs is:""")
             print("library_dirs=%s"%library_dirs)