driverclient 0.2.28__tar.gz → 0.2.29__tar.gz

{driverclient-0.2.28 → driverclient-0.2.29}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: driverclient
-Version: 0.2.28
+Version: 0.2.29
 Summary: Driver Server client with an embeddable connector for config injection at runtime.
 Project-URL: Homepage, https://example.com/driverclient
 Author-email: Raja Sanaullah <sanaullah@99technologies.com>

{driverclient-0.2.28 → driverclient-0.2.29}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "driverclient"
-version = "0.2.28"
+version = "0.2.29"
 description = "Driver Server client with an embeddable connector for config injection at runtime."
 readme = "README.md"
 requires-python = ">=3.10"

{driverclient-0.2.28 → driverclient-0.2.29}/src/driverclient/__init__.py

@@ -26,7 +26,7 @@ from driverclient.main import run as _run
 
 __all__ = ["ClientEvent", "DriverClient"]
 
-__version__ = "0.2.28"
+__version__ = "0.2.29"
 
 
 class DriverClient:

{driverclient-0.2.28 → driverclient-0.2.29}/src/driverclient/ops/capture.py

@@ -97,9 +97,6 @@ def _machine_meta() -> dict:
         "machine_id": machine_id,
     }
 
-
-_WINDOWS_INF = Path("C:\\Windows\\INF")
-
 _PNPUTIL_KEYS: dict[str, str] = {
     "published name": "published_name",
     "original name": "original_name",
@@ -349,12 +346,13 @@ def wu_update(force: bool = False) -> CaptureResult:
         **skip_reasons,
     )
 
+    baseline = _snapshot_published_names(cfg)
     trigger = _trigger_wu()
     if trigger is None:
         return CaptureResult(still_missing=len(wu_hwids))
 
     wu_start = datetime.now(timezone.utc)
-    total = _poll_and_submit_incremental(wu_start, cfg, trigger)
+    total = _poll_and_submit_incremental(wu_start, cfg, trigger, baseline)
     total.wu_triggered = True
     total.still_missing = max(0, len(wu_hwids) - total.submitted)
     total.reboot_required = _wu_reboot_pending()
@@ -383,12 +381,13 @@ def wu_full() -> CaptureResult:
         "windows_update", "[wu-full] Triggering full Windows Update scan (no HWID filtering)…"
     )
 
+    baseline = _snapshot_published_names(cfg)
     trigger = _trigger_wu()
     if trigger is None:
         return CaptureResult()
 
     wu_start = datetime.now(timezone.utc)
-    total = _poll_and_submit_incremental(wu_start, cfg, trigger)
+    total = _poll_and_submit_incremental(wu_start, cfg, trigger, baseline)
     total.wu_triggered = True
     _post_wu_session(cfg, wu_start, [], total.submitted)
     events.done(
@@ -528,17 +527,25 @@ def _parse_enum_drivers(output: str) -> list[dict]:
     return packages
 
 
-def _enumerate_packages_since(cutoff: datetime, cfg: dict) -> list[dict]:
-    """Return packages whose C:\\Windows\\INF mtime >= cutoff."""
-    result = []
-    for pkg in _enumerate_all_packages(cfg):
-        inf_path = _WINDOWS_INF / pkg["published_name"]
-        if not inf_path.exists():
-            continue
-        mtime = datetime.fromtimestamp(inf_path.stat().st_mtime, tz=timezone.utc)
-        if mtime >= cutoff:
-            result.append(pkg)
-    return result
+def _snapshot_published_names(cfg: dict) -> set[str]:
+    """Published names (oemNN.inf) already in the driver store — the baseline a
+    WU run's new-package detection diffs against. Must be taken BEFORE
+    triggering WU so nothing it installs is mistaken for pre-existing."""
+    return {p["published_name"] for p in _enumerate_all_packages(cfg) if p.get("published_name")}
+
+
+def _enumerate_new_packages(baseline: set[str], cfg: dict) -> list[dict]:
+    """Return packages present now that weren't in the pre-WU baseline.
+
+    Published names are assigned once by Windows and never reused while a
+    package stays in the store, so "not in baseline" is a robust new-package
+    signal. The previous approach compared each INF's mtime under
+    C:\\Windows\\INF against the WU start time — but that mtime can predate the
+    run (e.g. Windows already staged the file) or otherwise not bump the way
+    expected, silently dropping a genuinely new package (e.g. a discrete AMD
+    display driver landing via WU alongside an already-captured Intel one)
+    from ever being detected as new, so it was never submitted."""
+    return [p for p in _enumerate_all_packages(cfg) if p.get("published_name") not in baseline]
 
 
 def _published_names_for_hwids(
@@ -878,7 +885,7 @@ def _drain_wu_log(trigger: "_WuTrigger | None", pos: int) -> int:
 
 
 def _poll_and_submit_incremental(
-    wu_start: datetime, cfg: dict, trigger: "_WuTrigger | None"
+    wu_start: datetime, cfg: dict, trigger: "_WuTrigger | None", baseline: set[str]
 ) -> CaptureResult:
     """
     Wait on Windows Update, submitting each new DriverStore package as it lands.
@@ -913,7 +920,7 @@ def _poll_and_submit_incremental(
 
     while datetime.now(timezone.utc) < deadline:
         log_pos = _drain_wu_log(trigger, log_pos)  # surface WU's live phase
-        all_new = _enumerate_packages_since(wu_start, cfg)
+        all_new = _enumerate_new_packages(baseline, cfg)
         new_batch = [p for p in all_new if p.get("published_name", "") not in seen]
 
         if new_batch:
@@ -1214,4 +1221,4 @@ def _upload_worker(upload_q: queue.Queue, counts: dict, lock: threading.Lock, cf
 # Extension selection was removed: capture now packs every file
 # pnputil /export-driver produces (see _do_export), so the dump_extensions /
 # dump_extensions_overrides config keys are no longer consulted — filtering by
-# extension is what caused incomplete packages (broken link D).
+# extension is what caused incomplete packages (broken link D).