48 lines
1.5 KiB
Python
48 lines
1.5 KiB
Python
from requests import get
|
|
|
|
|
|
class Platform:
|
|
def __init__(self, data_source, data_source_args):
|
|
self.data_source = data_source
|
|
self.name = data_source["name"]
|
|
|
|
# parse the user's options from the config entries
|
|
for source_arg in data_source_args:
|
|
if source_arg.get("key") == "resource":
|
|
self.resource = source_arg.get("value")
|
|
|
|
if source_arg.get("key") == "method":
|
|
self.method = source_arg.get("value")
|
|
else:
|
|
self.method = "GET"
|
|
|
|
if source_arg.get("key") == "payload":
|
|
self.payload = source_arg.get("value")
|
|
|
|
if source_arg.get("key") == "authentication":
|
|
self.authentication = source_arg.get("value")
|
|
|
|
if source_arg.get("key") == "username":
|
|
self.username = source_arg.get("value")
|
|
|
|
if source_arg.get("key") == "password":
|
|
self.password = source_arg.get("value")
|
|
|
|
if source_arg.get("key") == "value_template":
|
|
self.value_template = source_arg.get("value")
|
|
else:
|
|
self.value_template = "value"
|
|
|
|
if source_arg.get("key") == "data_template":
|
|
self.data_template = source_arg.get("value")
|
|
else:
|
|
self.value_template = self.name
|
|
|
|
def process(self):
|
|
if self.method.upper() == "GET":
|
|
try:
|
|
value = get(self.resource)
|
|
except:
|
|
pass
|
|
return self.name
|