Use architecture-independent wheels and confirm libpq is available at runtime #339

Signed-off-by: Chin Yeung Li <[email protected]>

Author: chinyeungli

etc/scripts/build_nix_docker.py

@@ -227,6 +227,12 @@ def create_defualt_nix(dependencies_list, meta_dict):
                 for component in url_section:
                     if component.get("packagetype") == "bdist_wheel":
                         whl_url = component.get("url")
+                        if (
+                            ("cp313" not in whl_url and "py3" not in whl_url)
+                            or ("manylinux" not in whl_url and "-none-" not in whl_url)
+                            or ("any.whl" not in whl_url and "x86_64" not in whl_url)
+                        ):
+                            continue
                         whl_sha256 = get_sha256_hash(whl_url)
                         nix_content += "    " + name + " = buildCustomPackage {\n"
                         nix_content += '        pname = "' + name + '";\n'
@@ -310,6 +316,16 @@ def create_defualt_nix(dependencies_list, meta_dict):
       pip
     ];
 
+    # Add PostgreSQL to buildInputs to ensure libpq is available at runtime
+    buildInputs = with pkgs; [
+      postgresql
+    ];
+
+    # This wrapper ensures the PostgreSQL libraries are available at runtime
+    makeWrapperArgs = [
+      "--set LD_LIBRARY_PATH ${pkgs.postgresql.lib}/lib"
+    ];
+
     propagatedBuildInputs = with pythonWithOverlay.pkgs; [
 """