[util] add docstrings
This commit is contained in:
parent
0b0780b213
commit
456a3ec4a4
4 changed files with 111 additions and 5 deletions
|
@ -1,9 +1,15 @@
|
||||||
import copy
|
import copy
|
||||||
|
|
||||||
# algorithm copied from
|
|
||||||
# http://blog.impressiver.com/post/31434674390/deep-merge-multiple-python-dicts
|
|
||||||
# nicely done :)
|
|
||||||
def merge(target, *args):
|
def merge(target, *args):
|
||||||
|
"""Merges arbitrary data - copied from http://blog.impressiver.com/post/31434674390/deep-merge-multiple-python-dicts
|
||||||
|
|
||||||
|
:param target: the data structure to fill
|
||||||
|
:param args: a list of data structures to merge into target
|
||||||
|
|
||||||
|
:return: target, with all data in args merged into it
|
||||||
|
:rtype: whatever type was originally passed in
|
||||||
|
"""
|
||||||
if len(args) > 1:
|
if len(args) > 1:
|
||||||
for item in args:
|
for item in args:
|
||||||
merge(target, item)
|
merge(target, item)
|
||||||
|
|
13
util/cli.py
13
util/cli.py
|
@ -5,6 +5,19 @@ import logging
|
||||||
|
|
||||||
|
|
||||||
def execute(cmd, wait=True, ignore_errors=False, include_stderr=False, env=None):
|
def execute(cmd, wait=True, ignore_errors=False, include_stderr=False, env=None):
|
||||||
|
"""Executes a commandline utility and returns its output
|
||||||
|
|
||||||
|
:param cmd: the command (as string) to execute, returns the program's output
|
||||||
|
:param wait: set to True to wait for command completion, False to return immediately, defaults to True
|
||||||
|
:param ignore_errors: set to True to return a string when an exception is thrown, otherwise might throw, defaults to False
|
||||||
|
:param include_stderr: set to True to include stderr output in the return value, defaults to False
|
||||||
|
:param env: provide a dict here to specify a custom execution environment, defaults to None
|
||||||
|
|
||||||
|
:raises RuntimeError: the command either didn't exist or didn't exit cleanly, and ignore_errors was set to False
|
||||||
|
|
||||||
|
:return: output of cmd, or stderr, if ignore_errors is True and the command failed
|
||||||
|
:rtype: string
|
||||||
|
"""
|
||||||
args = shlex.split(cmd)
|
args = shlex.split(cmd)
|
||||||
logging.debug(cmd)
|
logging.debug(cmd)
|
||||||
try:
|
try:
|
||||||
|
|
|
@ -2,6 +2,14 @@ import re
|
||||||
|
|
||||||
|
|
||||||
def asbool(val):
|
def asbool(val):
|
||||||
|
"""Converts a value into a boolean
|
||||||
|
|
||||||
|
:param val: value to convert; accepts a wide range of
|
||||||
|
possible representations, such as yes, no, true, false, on, off
|
||||||
|
|
||||||
|
:return: True of val maps to true, False otherwise
|
||||||
|
:rtype: boolean
|
||||||
|
"""
|
||||||
if val is None:
|
if val is None:
|
||||||
return False
|
return False
|
||||||
if isinstance(val, bool):
|
if isinstance(val, bool):
|
||||||
|
@ -11,6 +19,17 @@ def asbool(val):
|
||||||
|
|
||||||
|
|
||||||
def asint(val, minimum=None, maximum=None):
|
def asint(val, minimum=None, maximum=None):
|
||||||
|
"""Converts a value into an integer
|
||||||
|
|
||||||
|
:param val: value to convert
|
||||||
|
:param minimum: if specified, this determines the lower
|
||||||
|
boundary for the returned value, defaults to None
|
||||||
|
:param maximum: if specified, this determines the upper
|
||||||
|
boundary for the returned value, defaults to None
|
||||||
|
|
||||||
|
:return: integer representation of value
|
||||||
|
:rtype: integer
|
||||||
|
"""
|
||||||
if val is None:
|
if val is None:
|
||||||
val = 0
|
val = 0
|
||||||
val = int(val)
|
val = int(val)
|
||||||
|
@ -20,6 +39,13 @@ def asint(val, minimum=None, maximum=None):
|
||||||
|
|
||||||
|
|
||||||
def aslist(val):
|
def aslist(val):
|
||||||
|
"""Converts a comma-separated value string into a list
|
||||||
|
|
||||||
|
:param val: value to convert, either a single value or a comma-separated string
|
||||||
|
|
||||||
|
:return: list representation of the value passed in
|
||||||
|
:rtype: list of strings
|
||||||
|
"""
|
||||||
if val is None:
|
if val is None:
|
||||||
return []
|
return []
|
||||||
if isinstance(val, list):
|
if isinstance(val, list):
|
||||||
|
@ -30,11 +56,30 @@ def aslist(val):
|
||||||
__UNITS = {"metric": "C", "kelvin": "K", "imperial": "F", "default": "C"}
|
__UNITS = {"metric": "C", "kelvin": "K", "imperial": "F", "default": "C"}
|
||||||
|
|
||||||
|
|
||||||
def astemperature(value, unit="metric"):
|
def astemperature(val, unit="metric"):
|
||||||
|
"""Returns a temperature representation of the input value
|
||||||
|
|
||||||
|
:param val: value to format, must be convertible into an integer
|
||||||
|
:param unit: unit of the input value, supported units are:
|
||||||
|
metric, kelvin, imperial, defaults to metric
|
||||||
|
|
||||||
|
:return: temperature representation of the input value
|
||||||
|
:rtype: string
|
||||||
|
"""
|
||||||
return "{}°{}".format(int(value), __UNITS.get(unit, __UNITS["default"]))
|
return "{}°{}".format(int(value), __UNITS.get(unit, __UNITS["default"]))
|
||||||
|
|
||||||
|
|
||||||
def byte(val, fmt="{:.2f}"):
|
def byte(val, fmt="{:.2f}"):
|
||||||
|
"""Returns a byte representation of the input value
|
||||||
|
|
||||||
|
:param val: value to format, must be convertible into a float
|
||||||
|
:param fmt: optional output format string, defaults to {:.2f}
|
||||||
|
|
||||||
|
:return: byte representation (e.g. <X> KiB, GiB, etc.) of the input value
|
||||||
|
:rtype: string
|
||||||
|
"""
|
||||||
|
|
||||||
|
val = float(val)
|
||||||
for unit in ["", "Ki", "Mi", "Gi"]:
|
for unit in ["", "Ki", "Mi", "Gi"]:
|
||||||
if val < 1024.0:
|
if val < 1024.0:
|
||||||
return "{}{}B".format(fmt, unit).format(val)
|
return "{}{}B".format(fmt, unit).format(val)
|
||||||
|
@ -46,6 +91,13 @@ __seconds_pattern = re.compile("(([\d\.?]+)h)?(([\d\.]+)m)?([\d\.]+)?s?")
|
||||||
|
|
||||||
|
|
||||||
def seconds(duration):
|
def seconds(duration):
|
||||||
|
"""Returns a time duration (in seconds) representation of the input value
|
||||||
|
|
||||||
|
:param duration: value to format (e.g. 5h30m2s)
|
||||||
|
|
||||||
|
:return: duration in seconds of the input value
|
||||||
|
:rtype: float
|
||||||
|
"""
|
||||||
if isinstance(duration, int) or isinstance(duration, float):
|
if isinstance(duration, int) or isinstance(duration, float):
|
||||||
return float(duration)
|
return float(duration)
|
||||||
|
|
||||||
|
@ -62,6 +114,15 @@ def seconds(duration):
|
||||||
|
|
||||||
|
|
||||||
def duration(duration, compact=False, unit=False):
|
def duration(duration, compact=False, unit=False):
|
||||||
|
"""Returns a time duration string representing the input value
|
||||||
|
|
||||||
|
:param duration: value to format, must be convertible into an into
|
||||||
|
:param compact: whether to show also seconds, defaults to False
|
||||||
|
:param unit: whether to display he unit, defaults to False
|
||||||
|
|
||||||
|
:return: duration representation (e.g. 5:02s) of the input value
|
||||||
|
:rtype: string
|
||||||
|
"""
|
||||||
duration = int(duration)
|
duration = int(duration)
|
||||||
if duration < 0:
|
if duration < 0:
|
||||||
return "n/a"
|
return "n/a"
|
||||||
|
|
|
@ -1,3 +1,13 @@
|
||||||
|
"""Retrieves location information from an external
|
||||||
|
service and caches it for 12h (retries are done every
|
||||||
|
30m in case of problems)
|
||||||
|
|
||||||
|
Right now, it uses (in order of preference):
|
||||||
|
- http://free.ipwhois.io/
|
||||||
|
- http://ipapi.co/
|
||||||
|
"""
|
||||||
|
|
||||||
|
|
||||||
import json
|
import json
|
||||||
import time
|
import time
|
||||||
import urllib.request
|
import urllib.request
|
||||||
|
@ -5,7 +15,6 @@ import urllib.request
|
||||||
__document = None
|
__document = None
|
||||||
__data = {}
|
__data = {}
|
||||||
__next = 0
|
__next = 0
|
||||||
|
|
||||||
__sources = [
|
__sources = [
|
||||||
{
|
{
|
||||||
"url": "http://free.ipwhois.io/json/",
|
"url": "http://free.ipwhois.io/json/",
|
||||||
|
@ -58,19 +67,36 @@ def __get(name, default=None):
|
||||||
|
|
||||||
|
|
||||||
def reset():
|
def reset():
|
||||||
|
"""Resets the location library, ensuring that a new query will be started
|
||||||
|
"""
|
||||||
global __next
|
global __next
|
||||||
__next = 0
|
__next = 0
|
||||||
|
|
||||||
|
|
||||||
def coordinates():
|
def coordinates():
|
||||||
|
"""Returns a latitude, longitude pair
|
||||||
|
|
||||||
|
:return: current latitude and longitude
|
||||||
|
:rtype: pair of strings
|
||||||
|
"""
|
||||||
return __get("latitude"), __get("longitude")
|
return __get("latitude"), __get("longitude")
|
||||||
|
|
||||||
|
|
||||||
def country():
|
def country():
|
||||||
|
"""Returns the current country name
|
||||||
|
|
||||||
|
:return: country name
|
||||||
|
:rtype: string
|
||||||
|
"""
|
||||||
return __get("country")
|
return __get("country")
|
||||||
|
|
||||||
|
|
||||||
def public_ip():
|
def public_ip():
|
||||||
|
"""Returns the current public IP
|
||||||
|
|
||||||
|
:return: public IP
|
||||||
|
:rtype: string
|
||||||
|
"""
|
||||||
return __get("public_ip")
|
return __get("public_ip")
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue