This commit is contained in:
lucasheld 2022-07-05 22:12:37 +02:00
parent f0a333557d
commit 27cf6bea97
3 changed files with 420 additions and 293 deletions

View file

@ -3,4 +3,5 @@ from .monitor_type import MonitorType
from .notification_providers import NotificationType, notification_provider_options from .notification_providers import NotificationType, notification_provider_options
from .proxy_protocol import ProxyProtocol from .proxy_protocol import ProxyProtocol
from .incident_style import IncidentStyle from .incident_style import IncidentStyle
from .converter import convert_from_socket, convert_to_socket, params_map_monitor, params_map_notification, params_map_proxy
from .api import UptimeKumaApi from .api import UptimeKumaApi

View file

@ -1,3 +1,4 @@
import json
import time import time
import socketio import socketio
@ -7,154 +8,22 @@ from . import MonitorType
from . import NotificationType, notification_provider_options from . import NotificationType, notification_provider_options
from . import ProxyProtocol from . import ProxyProtocol
from . import IncidentStyle from . import IncidentStyle
from . import convert_from_socket, convert_to_socket, params_map_monitor, params_map_notification, params_map_proxy
class UptimeKumaApi(object): def int_to_bool(data: list[dict] | dict, keys):
def __init__(self, url): if type(data) == list:
self.sio = socketio.Client() for d in data:
int_to_bool(d, keys)
else:
for key in keys:
if key in data:
data[key] = True if data[key] == 1 else False
self.event_data: dict[str, any] = {
"monitorList": None,
"notificationList": None,
"proxyList": None,
"statusPageList": None,
"heartbeatList": None,
"importantHeartbeatList": None,
"avgPing": None,
"uptime": None,
"heartbeat": None,
"info": None,
}
self.sio.on("connect", self.event_connect)
self.sio.on("disconnect", self.event_disconnect)
self.sio.on("monitorList", self.event_monitor_list)
self.sio.on("notificationList", self.event_notification_list)
self.sio.on("proxyList", self.event_proxy_list)
self.sio.on("statusPageList", self.event_status_page_list)
self.sio.on("heartbeatList", self.event_heartbeat_list)
self.sio.on("importantHeartbeatList",
self.event_important_heartbeat_list)
self.sio.on("avgPing", self.event_avg_ping)
self.sio.on("uptime", self.event_uptime)
self.sio.on("heartbeat", self.event_heartbeat)
self.sio.on("info", self.event_info)
self.connect(url)
def get_event_data(self, event):
while self.event_data[event] is None:
time.sleep(0.01)
time.sleep(0.01) # wait for multiple messages
return self.event_data[event]
# event handlers
def event_connect(self):
pass
def event_disconnect(self):
pass
def event_monitor_list(self, data):
self.event_data["monitorList"] = data
def event_notification_list(self, data):
self.event_data["notificationList"] = data
def event_proxy_list(self, data):
self.event_data["proxyList"] = data
def event_status_page_list(self, data):
self.event_data["statusPageList"] = data
def event_heartbeat_list(self, id_, data, bool_):
if self.event_data["heartbeatList"] is None:
self.event_data["heartbeatList"] = []
self.event_data["heartbeatList"].append({
"id": id_,
"data": data,
"bool": bool_,
})
def event_important_heartbeat_list(self, id_, data, bool_):
if self.event_data["importantHeartbeatList"] is None:
self.event_data["importantHeartbeatList"] = []
self.event_data["importantHeartbeatList"].append({
"id": id_,
"data": data,
"bool": bool_,
})
def event_avg_ping(self, id_, data):
if self.event_data["avgPing"] is None:
self.event_data["avgPing"] = []
self.event_data["avgPing"].append({
"id": id_,
"data": data,
})
def event_uptime(self, id_, hours_24, days_30):
if self.event_data["uptime"] is None:
self.event_data["uptime"] = []
self.event_data["uptime"].append({
"id": id_,
"hours_24": hours_24,
"days_30": days_30,
})
def event_heartbeat(self, data):
if self.event_data["heartbeat"] is None:
self.event_data["heartbeat"] = []
self.event_data["heartbeat"].append(data)
def event_info(self, data):
self.event_data["info"] = data
# connection
def connect(self, url):
url = url.rstrip("/")
self.sio.connect(f'{url}/socket.io/')
def disconnect(self):
self.sio.disconnect()
# monitors
def get_monitors(self):
return self.get_event_data("monitorList")
def get_monitor(self, id_):
return self.sio.call('getMonitor', id_)
def pause_monitor(self, id_):
return self.sio.call('pauseMonitor', id_)
def resume_monitor(self, id_):
return self.sio.call('resumeMonitor', id_)
def delete_monitor(self, id_):
return self.sio.call('deleteMonitor', id_)
def get_monitor_beats(self, id_, period):
return self.sio.call('getMonitorBeats', (id_, period))
def add_monitor(self, *args, **kwargs):
data = self._build_monitor_data(*args, **kwargs)
return self.sio.call('add', data)
def edit_monitor(self, id_, *args, **kwargs):
data = self._build_monitor_data(*args, **kwargs)
data.update({
"id": id_
})
return self.sio.call('editMonitor', data)
def _build_monitor_data( def _build_monitor_data(
self, type_: MonitorType,
monitor_type: MonitorType, name: str,
friendly_name: str,
heartbeat_interval: int = 60, heartbeat_interval: int = 60,
heartbeat_retry_interval: int = 60, heartbeat_retry_interval: int = 60,
retries: int = 0, retries: int = 0,
@ -211,8 +80,8 @@ class UptimeKumaApi(object):
notification_ids = dict_notification_ids notification_ids = dict_notification_ids
data = { data = {
"type": monitor_type, "type": type_,
"name": friendly_name, "name": name,
"interval": heartbeat_interval, "interval": heartbeat_interval,
"retryInterval": heartbeat_retry_interval, "retryInterval": heartbeat_retry_interval,
"maxretries": retries, "maxretries": retries,
@ -220,12 +89,17 @@ class UptimeKumaApi(object):
"upsideDown": upside_down_mode, "upsideDown": upside_down_mode,
} }
if monitor_type == MonitorType.KEYWORD: if tags:
data.update({
"tags": tags
})
if type_ == MonitorType.KEYWORD:
data.update({ data.update({
"keyword": keyword, "keyword": keyword,
}) })
if monitor_type in [MonitorType.HTTP, MonitorType.KEYWORD]: if type_ in [MonitorType.HTTP, MonitorType.KEYWORD]:
data.update({ data.update({
"url": url, "url": url,
"expiryNotification": certificate_expiry_notification, "expiryNotification": certificate_expiry_notification,
@ -251,23 +125,23 @@ class UptimeKumaApi(object):
"basicauth-workstation": auth_workstation, "basicauth-workstation": auth_workstation,
}) })
if monitor_type in [MonitorType.DNS, MonitorType.PING, MonitorType.STEAM, MonitorType.MQTT]: if type_ in [MonitorType.DNS, MonitorType.PING, MonitorType.STEAM, MonitorType.MQTT]:
data.update({ data.update({
"hostname": hostname, "hostname": hostname,
}) })
if monitor_type in [MonitorType.DNS, MonitorType.STEAM, MonitorType.MQTT]: if type_ in [MonitorType.DNS, MonitorType.STEAM, MonitorType.MQTT]:
data.update({ data.update({
"port": port, "port": port,
}) })
if monitor_type == MonitorType.DNS: if type_ == MonitorType.DNS:
data.update({ data.update({
"dns_resolve_server": dns_resolve_server, "dns_resolve_server": dns_resolve_server,
"dns_resolve_type": dns_resolve_type, "dns_resolve_type": dns_resolve_type,
}) })
if monitor_type == MonitorType.MQTT: if type_ == MonitorType.MQTT:
data.update({ data.update({
"mqttUsername": mqtt_username, "mqttUsername": mqtt_username,
"mqttPassword": mqtt_password, "mqttPassword": mqtt_password,
@ -275,7 +149,7 @@ class UptimeKumaApi(object):
"mqttSuccessMessage": mqtt_success_message, "mqttSuccessMessage": mqtt_success_message,
}) })
if monitor_type == MonitorType.SQLSERVER: if type_ == MonitorType.SQLSERVER:
data.update({ data.update({
"databaseConnectionString": sqlserver_connection_string, "databaseConnectionString": sqlserver_connection_string,
"sqlserverQuery": sqlserver_query, "sqlserverQuery": sqlserver_query,
@ -283,41 +157,8 @@ class UptimeKumaApi(object):
return data return data
# monitor tags
def add_monitor_tag(self, tag_id, monitor_id, value): def _build_notification_data(name: str, type_: NotificationType, default: bool, **kwargs):
return self.sio.call('addMonitorTag', (tag_id, monitor_id, value))
def edit_monitor_tag(self, tag_id, monitor_id, value):
return self.sio.call('editMonitorTag', (tag_id, monitor_id, value))
def delete_monitor_tag(self, tag_id, monitor_id, value):
return self.sio.call('deleteMonitorTag', (tag_id, monitor_id, value))
# notifications
def get_notifications(self):
return self.get_event_data("notificationList")
def test_notification(self, *args, **kwargs):
data = self._build_notification_data(*args, **kwargs)
return self.sio.call('testNotification', data)
def add_notification(self, *args, **kwargs):
data = self._build_notification_data(*args, **kwargs)
return self.sio.call('addNotification', (data, None))
def edit_notification(self, id_: int, *args, **kwargs):
data = self._build_notification_data(*args, **kwargs)
return self.sio.call('addNotification', (data, id_))
def delete_notification(self, id_: int):
return self.sio.call('deleteNotification', id_)
def check_apprise(self):
return self.sio.call('checkApprise')
def _build_notification_data(self, name: str, type_: NotificationType, default: bool, **kwargs):
allowed_options = notification_provider_options[type_] allowed_options = notification_provider_options[type_]
s1 = set(allowed_options) s1 = set(allowed_options)
s2 = set(kwargs.keys()) s2 = set(kwargs.keys())
@ -331,24 +172,8 @@ class UptimeKumaApi(object):
**kwargs **kwargs
} }
# proxy
def get_proxies(self):
return self.get_event_data("proxyList")
def add_proxy(self, *args, **kwargs):
data = self._build_proxy_data(*args, **kwargs)
return self.sio.call('addProxy', (data, None))
def edit_proxy(self, id_: int, *args, **kwargs):
data = self._build_proxy_data(*args, **kwargs)
return self.sio.call('addProxy', (data, id_))
def delete_proxy(self, id_: int):
return self.sio.call('deleteProxy', id_)
def _build_proxy_data( def _build_proxy_data(
self,
protocol: ProxyProtocol, protocol: ProxyProtocol,
host: str, host: str,
port: str, port: str,
@ -371,33 +196,248 @@ class UptimeKumaApi(object):
"applyExisting": apply_existing "applyExisting": apply_existing
} }
class UptimeKumaApi(object):
def __init__(self, url):
self.sio = socketio.Client()
self._event_data: dict[str, any] = {
"monitorList": None,
"notificationList": None,
"proxyList": None,
"statusPageList": None,
"heartbeatList": None,
"importantHeartbeatList": None,
"avgPing": None,
"uptime": None,
"heartbeat": None,
"info": None,
}
self.sio.on("connect", self._event_connect)
self.sio.on("disconnect", self._event_disconnect)
self.sio.on("monitorList", self._event_monitor_list)
self.sio.on("notificationList", self._event_notification_list)
self.sio.on("proxyList", self._event_proxy_list)
self.sio.on("statusPageList", self._event_status_page_list)
self.sio.on("heartbeatList", self._event_heartbeat_list)
self.sio.on("importantHeartbeatList", self._event_important_heartbeat_list)
self.sio.on("avgPing", self._event_avg_ping)
self.sio.on("uptime", self._event_uptime)
self.sio.on("heartbeat", self._event_heartbeat)
self.sio.on("info", self._event_info)
self.connect(url)
def _get_event_data(self, event):
while self._event_data[event] is None:
time.sleep(0.01)
time.sleep(0.01) # wait for multiple messages
return self._event_data[event]
# event handlers
def _event_connect(self):
pass
def _event_disconnect(self):
pass
def _event_monitor_list(self, data):
self._event_data["monitorList"] = data
def _event_notification_list(self, data):
self._event_data["notificationList"] = data
def _event_proxy_list(self, data):
self._event_data["proxyList"] = data
def _event_status_page_list(self, data):
self._event_data["statusPageList"] = data
def _event_heartbeat_list(self, id_, data, bool_):
if self._event_data["heartbeatList"] is None:
self._event_data["heartbeatList"] = []
self._event_data["heartbeatList"].append({
"id": id_,
"data": data,
"bool": bool_,
})
def _event_important_heartbeat_list(self, id_, data, bool_):
if self._event_data["importantHeartbeatList"] is None:
self._event_data["importantHeartbeatList"] = []
self._event_data["importantHeartbeatList"].append({
"id": id_,
"data": data,
"bool": bool_,
})
def _event_avg_ping(self, id_, data):
if self._event_data["avgPing"] is None:
self._event_data["avgPing"] = []
self._event_data["avgPing"].append({
"id": id_,
"data": data,
})
def _event_uptime(self, id_, hours_24, days_30):
if self._event_data["uptime"] is None:
self._event_data["uptime"] = []
self._event_data["uptime"].append({
"id": id_,
"hours_24": hours_24,
"days_30": days_30,
})
def _event_heartbeat(self, data):
if self._event_data["heartbeat"] is None:
self._event_data["heartbeat"] = []
self._event_data["heartbeat"].append(data)
def _event_info(self, data):
self._event_data["info"] = data
# connection
def connect(self, url: str):
url = url.rstrip("/")
self.sio.connect(f'{url}/socket.io/')
def disconnect(self):
self.sio.disconnect()
# monitors
def get_monitors(self):
monitors = list(self._get_event_data("monitorList").values())
int_to_bool(monitors, ["active"])
return monitors
def get_monitor(self, id_: int):
return self.sio.call('getMonitor', id_)
def pause_monitor(self, id_: int):
return self.sio.call('pauseMonitor', id_)
def resume_monitor(self, id_: int):
return self.sio.call('resumeMonitor', id_)
def delete_monitor(self, id_: int):
return self.sio.call('deleteMonitor', id_)
def get_monitor_beats(self, id_: int, period):
return self.sio.call('getMonitorBeats', (id_, period))
def add_monitor(self, *args, **kwargs):
data = _build_monitor_data(*args, **kwargs)
return self.sio.call('add', data)
def edit_monitor(self, id_: int, **kwargs):
monitor = self.get_monitor(id_)["monitor"]
kwargs_sock = convert_to_socket(params_map_monitor, kwargs)
monitor.update(kwargs_sock)
return self.sio.call('editMonitor', monitor)
# monitor tags
def add_monitor_tag(self, tag_id: int, monitor_id: int, value=""):
return self.sio.call('addMonitorTag', (tag_id, monitor_id, value))
# TODO: check!
# def edit_monitor_tag(self, tag_id: int, monitor_id: int, value=""):
# return self.sio.call('editMonitorTag', (tag_id, monitor_id, value))
def delete_monitor_tag(self, tag_id: int, monitor_id: int, value=""):
return self.sio.call('deleteMonitorTag', (tag_id, monitor_id, value))
# notifications
def get_notifications(self):
notifications_raw = self._get_event_data("notificationList")
notifications = []
for notification_raw in notifications_raw:
notification = notification_raw.copy()
config = json.loads(notification["config"])
del notification["config"]
notification.update(config)
notifications.append(notification)
return notifications
def get_notification(self, id_: int):
notifications = self.get_notifications()
for notification in notifications:
if notification["id"] == id_:
return notification
def test_notification(self, *args, **kwargs):
data = _build_notification_data(*args, **kwargs)
return self.sio.call('testNotification', data)
def add_notification(self, *args, **kwargs):
data = _build_notification_data(*args, **kwargs)
return self.sio.call('addNotification', (data, None))
def edit_notification(self, id_: int, **kwargs):
notification = self.get_notification(id_)
kwargs_sock = convert_to_socket(params_map_notification, kwargs)
notification.update(kwargs_sock)
return self.sio.call('addNotification', (notification, id_))
def delete_notification(self, id_: int):
return self.sio.call('deleteNotification', id_)
def check_apprise(self):
return self.sio.call('checkApprise')
# proxy
def get_proxies(self):
proxies = self._get_event_data("proxyList")
proxies = convert_from_socket(params_map_proxy, proxies)
int_to_bool(proxies, ["auth", "active", "default", "apply_existing"])
proxies = convert_to_socket(params_map_proxy, proxies)
return proxies
def get_proxy(self, id_: int):
proxies = self.get_proxies()
for proxy in proxies:
if proxy["id"] == id_:
return proxy
def add_proxy(self, *args, **kwargs):
data = _build_proxy_data(*args, **kwargs)
return self.sio.call('addProxy', (data, None))
def edit_proxy(self, id_: int, **kwargs):
proxy_sock = self.get_proxy(id_)
proxy = convert_from_socket(params_map_proxy, proxy_sock)
for key in ["auth", "active", "default", "apply_existing"]:
if key in proxy:
proxy[key] = True if proxy[key] == 1 else False
kwargs_sock = convert_to_socket(params_map_proxy, kwargs)
proxy_sock = convert_to_socket(params_map_proxy, proxy)
proxy_sock.update(kwargs_sock)
return self.sio.call('addProxy', (proxy_sock, id_))
# return proxy_sock, id_
def delete_proxy(self, id_: int):
return self.sio.call('deleteProxy', id_)
# status page # status page
def get_status_pages(self): def get_status_pages(self):
return self.get_event_data("statusPageList") return list(self._get_event_data("statusPageList").values())
def post_incident(
self,
slug: str,
# incident
title: str,
content: str,
style: IncidentStyle = IncidentStyle.PRIMARY
):
incident = {
"title": title,
"content": content,
"style": style
}
return self.sio.call('postIncident', (slug, incident))
def unpin_incident(self, slug: str):
return self.sio.call('unpinIncident', slug)
def get_status_page(self, slug: str): def get_status_page(self, slug: str):
return self.sio.call('getStatusPage', slug) return self.sio.call('getStatusPage', slug)
def add_status_page(self, title: str, slug: str):
return self.sio.call('addStatusPage', (title, slug))
def delete_status_page(self, slug: str):
return self.sio.call('deleteStatusPage', slug)
def save_status_page( def save_status_page(
self, self,
slug: str, slug: str,
@ -441,37 +481,50 @@ class UptimeKumaApi(object):
} }
return self.sio.call('saveStatusPage', (slug, config, img_data_url, public_group_list)) return self.sio.call('saveStatusPage', (slug, config, img_data_url, public_group_list))
def add_status_page(self, title: str, slug: str): def post_incident(
return self.sio.call('addStatusPage', (title, slug)) self,
slug: str,
def delete_status_page(self, slug: str): # incident
return self.sio.call('deleteStatusPage', slug) title: str,
content: str,
style: IncidentStyle = IncidentStyle.PRIMARY
):
incident = {
"title": title,
"content": content,
"style": style
}
return self.sio.call('postIncident', (slug, incident))
def unpin_incident(self, slug: str):
return self.sio.call('unpinIncident', slug)
# heartbeat # heartbeat
def get_heartbeats(self): def get_heartbeats(self):
return self.get_event_data("heartbeatList") return self._get_event_data("heartbeatList")
def get_important_heartbeats(self): def get_important_heartbeats(self):
return self.get_event_data("importantHeartbeatList") return self._get_event_data("importantHeartbeatList")
def get_heartbeat(self): def get_heartbeat(self):
return self.get_event_data("heartbeat") return self._get_event_data("heartbeat")
# avg ping # avg ping
def avg_ping(self): def avg_ping(self):
return self.get_event_data("avgPing") return self._get_event_data("avgPing")
# uptime # uptime
def uptime(self): def uptime(self):
return self.get_event_data("uptime") return self._get_event_data("uptime")
# info # info
def info(self): def info(self):
return self.get_event_data("info") return self._get_event_data("info")
# clear # clear
@ -489,16 +542,27 @@ class UptimeKumaApi(object):
def get_tags(self): def get_tags(self):
return self.sio.call('getTags') return self.sio.call('getTags')
def edit_tag(self, tag): def get_tag(self, id_: int):
return self.sio.call('editTag', tag) tags = self.get_tags()
for tag in tags:
if tag["id"] == id_:
return tag
def delete_tag(self, id_): # TODO: not working, monitor id required?
# def edit_tag(self, id_: int, name: str, color: str):
# return self.sio.call('editTag', {
# "id": id_,
# "name": name,
# "color": color
# })
def delete_tag(self, id_: int):
return self.sio.call('deleteTag', id_) return self.sio.call('deleteTag', id_)
def add_tag(self, color, name): def add_tag(self, name: str, color: str):
return self.sio.call('addTag', { return self.sio.call('addTag', {
"color": color,
"name": name, "name": name,
"color": color,
"new": True "new": True
}) })
@ -507,16 +571,16 @@ class UptimeKumaApi(object):
def get_settings(self): def get_settings(self):
return self.sio.call('getSettings') return self.sio.call('getSettings')
def set_settings(self, data, password): def set_settings(self, data, password: str):
return self.sio.call('setSettings', (data, password)) return self.sio.call('setSettings', (data, password))
def change_password(self, old_password, new_password): def change_password(self, old_password: str, new_password: str):
return self.sio.call('changePassword', { return self.sio.call('changePassword', {
"currentPassword": old_password, "currentPassword": old_password,
"newPassword": new_password, "newPassword": new_password,
}) })
def upload_backup(self, json_data, import_handle): def upload_backup(self, json_data, import_handle: str):
if import_handle not in ["overwrite", "skip", "keep"]: if import_handle not in ["overwrite", "skip", "keep"]:
raise ValueError() raise ValueError()
return self.sio.call('uploadBackup', (json_data, import_handle)) return self.sio.call('uploadBackup', (json_data, import_handle))
@ -526,28 +590,28 @@ class UptimeKumaApi(object):
def twofa_status(self): def twofa_status(self):
return self.sio.call('twoFAStatus') return self.sio.call('twoFAStatus')
def prepare_2fa(self, password): def prepare_2fa(self, password: str):
return self.sio.call('prepare2FA', password) return self.sio.call('prepare2FA', password)
def save_2fa(self, password): def save_2fa(self, password: str):
return self.sio.call('save2FA', password) return self.sio.call('save2FA', password)
def disable_2fa(self, password): def disable_2fa(self, password: str):
return self.sio.call('disable2FA', password) return self.sio.call('disable2FA', password)
# login # login
def login(self, username, password): def login(self, username: str, password: str):
return self.sio.call('login', { return self.sio.call('login', {
"username": username, "username": username,
"password": password, "password": password,
"token": "" "token": ""
}) })
def login_by_token(self, token): def login_by_token(self, token: str):
return self.sio.call('loginByToken', token) return self.sio.call('loginByToken', token)
def verify_token(self, token, password): def verify_token(self, token: str, password: str):
return self.sio.call('verifyToken', (token, password)) return self.sio.call('verifyToken', (token, password))
def logout(self): def logout(self):
@ -558,7 +622,7 @@ class UptimeKumaApi(object):
def need_setup(self): def need_setup(self):
return self.sio.call('needSetup') return self.sio.call('needSetup')
def setup(self, username, password): def setup(self, username: str, password: str):
return self.sio.call("setup", (username, password)) return self.sio.call("setup", (username, password))
# database # database

View file

@ -0,0 +1,62 @@
# socket -> python
params_map_monitor = {
"type": "type_",
"interval": "heartbeat_interval",
"retryInterval": "heartbeat_retry_interval",
"maxretries": "retries",
"notificationIDList": "notification_ids",
"upsideDown": "upside_down_mode",
"expiryNotification": "certificate_expiry_notification",
"ignoreTls": "ignore_tls_error",
"maxredirects": "max_redirects",
"accepted_statuscodes": "accepted_status_codes",
"proxyId": "proxy_id",
"method": "http_method",
"body": "http_body",
"headers": "http_headers",
"authMethod": "auth_method",
"basicauth-user": "auth_user",
"basicauth-pass": "auth_pass",
"basicauth-domain": "auth_domain",
"basicauth-workstation": "auth_workstation",
"mqttUsername": "mqtt_username",
"mqttPassword": "mqtt_password",
"mqttTopic": "mqtt_topic",
"mqttSuccessMessage": "mqtt_success_message",
"databaseConnectionString": "sqlserver_connection_string",
"sqlserverQuery": "sqlserver_query"
}
params_map_notification = {
"type": "type_",
"isDefault": "default"
}
params_map_proxy = {
"applyExisting": "apply_existing"
}
def _convert_to_from_socket(params_map: dict[str, str], params: list[dict] | dict, to_socket=False):
if type(params) == list:
out = []
params_list = params
for params in params_list:
params_py = _convert_to_from_socket(params_map, params, to_socket)
out.append(params_py)
else:
if to_socket:
params_map = {v: k for k, v in params_map.items()}
out = {}
for key, value in params.items():
key = params_map.get(key, key)
out[key] = value
return out
def convert_from_socket(params_map, params):
return _convert_to_from_socket(params_map, params)
def convert_to_socket(params_map, params):
return _convert_to_from_socket(params_map, params, to_socket=True)