ED/EDMarketConnector
1
0
Fork 0
mirror of https://github.com/EDCD/EDMarketConnector.git synced 2025-06-07 19:03:23 +03:00
Code Issues Projects Releases Wiki Activity

Added util method for URL -> KillSwitchSet

Browse Source
This commit is contained in:
A_D 2020-10-14 21:30:18 +02:00 committed by Athanasius
parent f7890a460b
commit 29adaa413b
1 changed files with 16 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
killswitch.py
View File

@ -1,4 +1,5 @@
"""Fetch kill switches from EDMC Repo.""" """Fetch kill switches from EDMC Repo."""
from ast import parse
from typing import Dict, List, NamedTuple, Optional, Union, cast from typing import Dict, List, NamedTuple, Optional, Union, cast
import requests import requests
@ -49,7 +50,7 @@ class KillSwitchSet:
return f'KillSwitchSet: {str(self.kill_switches)}' return f'KillSwitchSet: {str(self.kill_switches)}'
def __repr__(self) -> str: def __repr__(self) -> str:
"""return __repr__ for KillSwitchSet.""" """Return __repr__ for KillSwitchSet."""
return f'KillSwitchSet(kill_switches={self.kill_switches!r})' return f'KillSwitchSet(kill_switches={self.kill_switches!r})'
@ -117,6 +118,20 @@ def parse_kill_switches(data: KILL_SWITCH_JSON_DICT) -> List[KillSwitch]:
return out return out
def get_kill_switches(target=DEFAULT_KILLSWITCH_URL) -> Optional[KillSwitchSet]:
"""
Get a kill switch set object.
:param target: the URL to fetch the killswitch JSON from, defaults to DEFAULT_KILLSWITCH_URL
:return: the KillSwitchSet for the URL, or None if there was an error
"""
if (data := fetch_kill_switches(target)) is None:
logger.warning('could not get killswitches')
return None
return KillSwitchSet(parse_kill_switches(data))
active: KillSwitchSet = KillSwitchSet([]) active: KillSwitchSet = KillSwitchSet([])