added better handling for steam guard

DJSchaffner · Dec 16, 2023 · 2e55c2a · 2e55c2a
1 parent 3f3abd1
commit 2e55c2a
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/logic.py b/src/logic.py
@@ -284,6 +284,7 @@ def _depot_downloader(self, options: list):
       responses = [
         "result: OK",
         "Please enter .*: ",
+        "STEAM GUARD!",
         pexpect.EOF
       ]
 
@@ -292,7 +293,7 @@ def _depot_downloader(self, options: list):
       response = p.expect(responses, timeout=timeout)
 
       # Error
-      if response == 2:
+      if response == 3:
         raise ConnectionError("Error logging into account")
 
       # Code required
@@ -317,6 +318,10 @@ def _depot_downloader(self, options: list):
           if p.expect(responses, timeout=timeout) == 1:
             raise ConnectionError("Invalid authentication code")
 
+      # Steam Guard
+      elif response == 2:
+        pass
+
       # Wait for program to finish
       p.expect(pexpect.EOF, timeout=None)
     except pexpect.exceptions.TIMEOUT: