38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
from flask import render_template_string
|
|
import requests
|
|
|
|
|
|
class Platform:
|
|
def __init__(self, *args, **kwargs):
|
|
for key, value in kwargs.items():
|
|
self.__dict__[key] = value
|
|
|
|
if not hasattr(self, "resource"):
|
|
self.resource = "http://localhost/json"
|
|
|
|
if not hasattr(self, "password"):
|
|
self.password = ""
|
|
|
|
self.id = 1
|
|
self.session = requests.Session()
|
|
self._api_call("auth.login", [self.password])
|
|
self.password = None # Discard password, no longer needed.
|
|
|
|
def _api_call(self, method, params=[]):
|
|
json = {
|
|
"id": self.id,
|
|
"method": method,
|
|
"params": params
|
|
}
|
|
return self.session.post(self.resource, json=json)
|
|
|
|
def process(self):
|
|
r = self._api_call("web.update_ui", ["download_rate", "upload_rate"])
|
|
json = r.json()
|
|
data = {}
|
|
for key, field in json["result"]["stats"].items():
|
|
data[key] = field
|
|
|
|
value_template = render_template_string(self.value_template, **data)
|
|
return value_template
|