From 80f9bb6d75b2608c7e5b9416d8aef0c987579154 Mon Sep 17 00:00:00 2001
From: 2Shirt <2xShirt@gmail.com>
Date: Mon, 21 Feb 2022 17:45:12 -0700
Subject: [PATCH] Run AV scans using HIGH priority

Addresses issue #13
---
 scripts/wk/repairs/win.py | 26 +++++++++++++++++++++-----
 1 file changed, 21 insertions(+), 5 deletions(-)

diff --git a/scripts/wk/repairs/win.py b/scripts/wk/repairs/win.py
index 787fdea3..2abe8d80 100644
--- a/scripts/wk/repairs/win.py
+++ b/scripts/wk/repairs/win.py
@@ -28,6 +28,7 @@ from wk.exe           import (
   get_procs,
   run_program,
   popen_program,
+  set_proc_priority,
   wait_for_procs,
   )
 from wk.io            import (
@@ -1109,10 +1110,14 @@ def run_emsisoft_cmd_scan():
   if IN_CONEMU:
     cmd.extend(['-new_console:nb', '-new_console:s33V'])
     run_program(cmd, check=False, pipe=False)
-    sleep(1)
+    sleep(5)
+    set_proc_priority('a2cmd.exe', 'HIGH')
     wait_for_procs('a2cmd.exe')
   else:
-    run_program(cmd, check=False)
+    proc = popen_program(cmd, priority=True)
+    sleep(5)
+    set_proc_priority('a2cmd.exe', 'HIGH')
+    proc.wait()
 
 
 def run_hitmanpro():
@@ -1121,7 +1126,11 @@ def run_hitmanpro():
   log_path = log_path.with_suffix('.xml')
   log_path.parent.mkdir(parents=True, exist_ok=True)
   cmd_args = ['/scanonly', f'/log={log_path}']
-  run_tool('HitmanPro', 'HitmanPro', *cmd_args, download=True)
+  proc = run_tool(
+    'HitmanPro', 'HitmanPro', *cmd_args,
+    download=True, popen=True, priority=True,
+    )
+  proc.wait()
 
 
 def run_iobit_uninstaller():
@@ -1154,12 +1163,19 @@ def run_kvrt():
       _f.write(f'"{kvrt_path}" {" ".join(cmd_args)}\n')
     cmd = ('cmd', '/c', tmp_file, '-new_console:nb', '-new_console:s33V')
     run_program(cmd, check=False)
-    sleep(1)
+    sleep(5)
+    set_proc_priority('KVRT', 'HIGH', exact=False)
     wait_for_procs('KVRT.exe')
     return
 
   # Run in background
-  proc = run_tool('KVRT', 'KVRT', *cmd_args, download=True)
+  proc = run_tool(
+    'KVRT', 'KVRT', *cmd_args,
+    download=True, popen=True, priority=True,
+    )
+  sleep(5)
+  set_proc_priority('KVRT', 'HIGH', exact=False)
+  proc.wait()
   log_path.write_text(proc.stdout, encoding='utf-8')