ExternalVendorCode/onefuzz
1
0
Fork 0
mirror of https://github.com/microsoft/onefuzz.git synced 2025-06-16 03:48:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use .get instead of lookup (#2165)

Browse Source
This commit is contained in:
George Pollard
2022-07-15 11:55:30 +12:00
committed by GitHub
parent fe8b75045f
commit 3347e7e67b
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/api-service/__app__/onefuzzlib/workers/nodes.py
View File

@ -133,7 +133,7 @@ class Node(BASE_NODE, ORMMixin):
@classmethod
def mark_outdated_nodes(cls) -> None:
# if outdated agents are allowed, do not attempt to update
if os.environ["ONEFUZZ_ALLOW_OUTDATED_AGENT"] == "true":
if os.environ.get("ONEFUZZ_ALLOW_OUTDATED_AGENT") == "true":
return
# ony update 500 nodes at a time to mitigate timeout issues
@ -268,7 +268,10 @@ class Node(BASE_NODE, ORMMixin):
from .pools import Pool
from .scalesets import Scaleset
if self.is_outdated() and os.environ["ONEFUZZ_ALLOW_OUTDATED_AGENT"] != "true":
if (
self.is_outdated()
and os.environ.get("ONEFUZZ_ALLOW_OUTDATED_AGENT") != "true"
):
logging.info(
"can_process_new_work agent and service versions differ, "
"stopping node. "