Add dirs

setup.py

@@ -14,19 +14,33 @@
     "exprtk-custom-types",
 ]
 
-def list_files(start_path):
-    for root, _, files in os.walk(start_path):
-        level = root.replace(start_path, "").count(os.sep)
-        indent = " " * 4 * (level)
-        print("{}{}/".format(indent, os.path.basename(root)))
-        sub_indent = " " * 4 * (level + 1)
-        for f in files:
-            print("{}{}".format(sub_indent, f))
+library_dirs = []
 
 
-if os.path.exists(mpfr_include := "/opt/homebrew/Cellar/mpfr"):
-    list_files(mpfr_include)
-    include_dirs.append(mpfr_include)
+def add_to_include(target: str, header: str):
+    if not os.path.exists(target):
+        return
+
+    for root, _, files in os.walk(target):
+        if any(header in file for file in files):
+            include_dirs.append(root)
+            break
+
+
+def add_to_library(target: str, header: str):
+    if not os.path.exists(target):
+        return
+
+    for root, _, files in os.walk(target):
+        if any(header in file for file in files):
+            library_dirs.append(root)
+            break
+
+add_to_include("/opt/homebrew/Cellar/mpfr", "mpfr.h")
+add_to_include("/opt/homebrew/Cellar/gmp", "gmp.h")
+
+add_to_library("/opt/homebrew/Cellar/mpfr", "libmpfr")
+add_to_library("/opt/homebrew/Cellar/gmp", "libgmp")
 
 # noinspection PyTypeChecker
 setup(
@@ -42,6 +56,7 @@ def list_files(start_path):
             "_pyvpmr",
             ["src/VPMR.cpp"],
             include_dirs=include_dirs,
+            library_dirs=library_dirs,
             libraries=["mpfr", "gmp", "tbb"],
             define_macros=[("PYVPMR", 1)],
             extra_compile_args=["-fno-aligned-allocation"]