diff --git a/src/logic.py b/src/logic.py index c0e826e..abea4b8 100644 --- 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: