warmpath 0.3.4__tar.gz → 0.3.5__tar.gz

{warmpath-0.3.4 → warmpath-0.3.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: warmpath
-Version: 0.3.4
+Version: 0.3.5
 Summary: Local LinkedIn visible-connections scraper.
 Requires-Python: >=3.10
 Requires-Dist: browser-cookie3>=0.20.1

{warmpath-0.3.4 → warmpath-0.3.5}/README.md

@@ -36,6 +36,9 @@ uvx warmpath human https://www.linkedin.com/in/mitchellh/ --refresh-cache
 uvx warmpath company --help
 ```
 
+`company` prints mutual contacts for second-degree candidates, looking them up
+when LinkedIn's company search does not include their names.
+
 `companies` walks the complete 1st-degree connections list, looks up current
 employers, and deduplicates by company ID. It includes multiple current employers
 and keeps unlinked employer names; `--urls` prints only known LinkedIn company

{warmpath-0.3.4 → warmpath-0.3.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "warmpath"
-version = "0.3.4"
+version = "0.3.5"
 description = "Local LinkedIn visible-connections scraper."
 requires-python = ">=3.10"
 dependencies = [

{warmpath-0.3.4 → warmpath-0.3.5}/tests/test_mutual_connections.py

@@ -1,3 +1,8 @@
+from unittest.mock import Mock
+
+import pytest
+
+from warmpath import cli
 from warmpath.cli import (
     candidate_matches_filter,
     company_path_candidate,
@@ -106,6 +111,98 @@ def test_second_degree_candidate_renders_visible_mutual_connections() -> None:
     assert "unknown introducer" not in rendered
 
 
+@pytest.mark.parametrize("mutual_count", [None, 3])
+@pytest.mark.parametrize("found_mutual", [True, False])
+def test_company_looks_up_missing_mutual_contacts(
+    tmp_path, monkeypatch, mutual_count, found_mutual
+) -> None:
+    monkeypatch.setattr(cli, "resolve_company", Mock(return_value=(
+        {"name": "Acme", "urn_id": "company-id"}, []
+    )))
+    monkeypatch.setattr(cli, "fetch_company_people", Mock(side_effect=lambda *args: (
+        [] if args[3] == 1 else [{
+            "name": "Employee",
+            "urn_id": "employee-id",
+            "distance": "DISTANCE_2",
+            "mutual_count": mutual_count,
+            "mutuals_truncated": True,
+            "_search_source": "search.current_company",
+        }]
+    )))
+    api = Mock()
+    api.search.return_value = [{
+        "entityUrn": "urn:li:fsd_profile:contact-id",
+        "title": {"text": "Ada Lovelace"},
+        "navigationUrl": "https://www.linkedin.com/in/ada-lovelace/",
+        "entityCustomTrackingInfo": {"memberDistance": "DISTANCE_1"},
+    }] if found_mutual else []
+
+    result = cli.find_company_path_candidates(
+        api, "Acme", 2, 5, None, 5, tmp_path, False
+    )
+
+    api.search.assert_called_once_with({
+        "filters": (
+            "List((key:resultType,value:List(PEOPLE)),"
+            "(key:connectionOf,value:List(employee-id)),"
+            "(key:network,value:List(F)))"
+        )
+    }, limit=mutual_count or cli.DEFAULT_MAX_MUTUAL_CONNECTIONS)
+    candidate = result["candidates"][0]
+    rendered = render_company_path_result(result)
+    assert "Employee" in rendered
+    assert candidate["evidence"]["source"] == "search.current_company"
+    if found_mutual:
+        expected = "Mutuals (3): Ada Lovelace, +2 more" if mutual_count else "Mutuals (1): Ada Lovelace"
+        assert expected in rendered
+        assert candidate["path_status"] == "partially_resolved"
+        assert candidate["path"][1]["url"] == "https://www.linkedin.com/in/ada-lovelace/"
+    else:
+        assert "Mutuals" not in rendered
+        assert candidate["path_status"] == "unresolved"
+        assert candidate["mutual_count"] == mutual_count
+        assert candidate["mutuals_truncated"] is True
+
+    # Repeating a company search reuses its mutual-contact lookup.
+    assert cli.find_company_path_candidates(
+        api, "Acme", 2, 5, None, 5, tmp_path, False
+    ) == result
+    api.search.assert_called_once()
+
+
+def test_company_skips_mutual_lookups_for_direct_known_and_unprinted_candidates(
+    tmp_path, monkeypatch
+) -> None:
+    monkeypatch.setattr(cli, "resolve_company", Mock(return_value=(
+        {"name": "Acme", "urn_id": "company-id"}, []
+    )))
+    direct = {"name": "Direct", "urn_id": "direct-id", "jobtitle": "Engineer"}
+    second_degree = [
+        {"name": "Filtered out", "urn_id": "filtered-id", "jobtitle": "Recruiter"},
+        {
+            "name": "Known mutual",
+            "urn_id": "known-id",
+            "jobtitle": "Engineer",
+            "mutual_connections": [{"name": "Ada Lovelace"}],
+        },
+        {"name": "Over limit", "urn_id": "limited-id", "jobtitle": "Engineer"},
+    ]
+    monkeypatch.setattr(cli, "fetch_company_people", Mock(
+        side_effect=lambda *args: [direct] if args[3] == 1 else second_degree
+    ))
+    api = Mock()
+
+    result = cli.find_company_path_candidates(
+        api, "Acme", 2, 2, "Engineer", 5, tmp_path, False
+    )
+
+    assert [candidate["target"]["name"] for candidate in result["candidates"]] == [
+        "Direct", "Known mutual"
+    ]
+    assert "Mutuals (1): Ada Lovelace" in render_company_path_result(result)
+    api.search.assert_not_called()
+
+
 def test_second_degree_candidate_without_mutuals_keeps_unresolved_status() -> None:
     row = {
         "name": "Ruslan Gilemzianov",

{warmpath-0.3.4 → warmpath-0.3.5}/uv.lock

@@ -768,7 +768,7 @@ wheels = [
 
 [[package]]
 name = "warmpath"
-version = "0.3.4"
+version = "0.3.5"
 source = { editable = "." }
 dependencies = [
     { name = "browser-cookie3" },

{warmpath-0.3.4 → warmpath-0.3.5}/warmpath/cli.py

@@ -682,6 +682,9 @@ def company_path_candidate(row: dict[str, Any], degree: int) -> dict[str, Any]:
         "degree": degree,
         "path_status": "unresolved",
         "target": target,
+        "mutual_connections": mutual_connections,
+        "mutual_count": mutual_count,
+        "mutuals_truncated": mutuals_truncated,
         "path": [
             {"role": "me"},
             {"role": "unknown_introducer"},
@@ -1190,6 +1193,23 @@ def find_company_path_candidates(
 
     candidates.sort(key=candidate_score)
     candidates = candidates[:limit]
+    for index, candidate in enumerate(candidates):
+        if candidate["degree"] != 2 or candidate_mutual_names(candidate):
+            continue
+        row = {
+            **candidate["target"],
+            "mutual_count": candidate.get("mutual_count"),
+            "mutuals_truncated": candidate.get("mutuals_truncated", False),
+            "_search_source": candidate["evidence"]["source"],
+        }
+        row = enrich_row_with_mutual_connections(
+            api,
+            row,
+            cache_dir,
+            refresh_cache,
+        )
+        candidates[index] = company_path_candidate(row, 2)
+
     direct_count = sum(1 for candidate in candidates if candidate["degree"] == 1)
     second_count = sum(1 for candidate in candidates if candidate["degree"] == 2)
     resolved_count = sum(