mbirth/tcl_ota_check
1
0
Fork 0
mirror of https://github.com/mbirth/tcl_ota_check.git synced 2024-11-09 22:06:47 +00:00
Code Issues Releases Wiki Activity

More cleanup.

Browse Source
This commit is contained in:
Markus Birth 2018-02-10 02:38:06 +01:00
parent 485e7b30b8
commit 75aef38a6d
Signed by: mbirth
GPG Key ID: A9928D7A098C3A9A
2 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tcllib/requests/runner.py
View File

@ -1,6 +1,6 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
from . import tcl from .tclrequest import TclRequest
from . import http from . import http
from . import serverselector from . import serverselector
@ -25,7 +25,7 @@ class RequestRunner:
"""Returns a master server.""" """Returns a master server."""
return self.server_selector.get_master_server() return self.server_selector.get_master_server()
def run(self, query: tcl.TclRequest, timeout: int=10) -> bool: def run(self, query: TclRequest, timeout: int=10) -> bool:
"""Runs the actual query.""" """Runs the actual query."""
for _ in range(0, self.max_tries): for _ in range(0, self.max_tries):
url = "{}{}{}".format(self.protocol, self.get_server(), query.uri) url = "{}{}{}".format(self.protocol, self.get_server(), query.uri)

7
tcllib/requests/tclresult.py
View File

@ -1,5 +1,7 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
import xml.dom.minidom
from defusedxml import ElementTree from defusedxml import ElementTree
from . import dumpmgr from . import dumpmgr
@ -14,6 +16,11 @@ class TclResult:
def delete_dump(self): def delete_dump(self):
self.dumper.delete_last_dump() self.dumper.delete_last_dump()
def pretty_xml(self):
"""Return prettified input XML with ``xml.dom.minidom``."""
mdx = xml.dom.minidom.parseString(self.raw_xml)
return mdx.toprettyxml(indent=" ")
class CheckResult(TclResult): class CheckResult(TclResult):
def __init__(self, xml: str): def __init__(self, xml: str):
super().__init__(xml) super().__init__(xml)