michael/scrypted
1
0
Fork 0
mirror of https://github.com/koush/scrypted.git synced 2026-03-20 16:40:24 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Revert "server: do not serialize python nan in rpc protocol."

Browse Source 
This reverts commit e9d73c6faa.
This commit is contained in:
Koushik Dutta
2024-11-10 12:29:26 -08:00
parent 8954de3c93
commit bc95a15f89
1 changed files with 2 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
server/python/rpc_reader.py
View File

@@ -12,23 +12,7 @@ import multiprocessing.connection
import rpc
import concurrent.futures
import json
import math
def convert_nan_inf_to_none(obj):
if isinstance(obj, dict):
return {key: convert_nan_inf_to_none(value) for key, value in obj.items()}
elif isinstance(obj, list):
return [convert_nan_inf_to_none(item) for item in obj]
elif isinstance(obj, float):
if math.isnan(obj):
return None
if math.isinf(obj):
return None
return obj
def json_dumps_clean(obj, **kwargs):
cleaned_obj = convert_nan_inf_to_none(obj)
return json.dumps(cleaned_obj, **kwargs)
class BufferSerializer(rpc.RpcSerializer):
def serialize(self, value, serializationContext):
@@ -110,7 +94,7 @@ class RpcFileTransport(RpcTransport):
reject(e)
def writeJSON(self, j, reject):
return self.writeMessage(0, bytes(json_dumps_clean(j), 'utf8'), reject)
return self.writeMessage(0, bytes(json.dumps(j), 'utf8'), reject)
def writeBuffer(self, buffer, reject):
return self.writeMessage(1, buffer, reject)
@@ -144,7 +128,7 @@ class RpcStreamTransport(RpcTransport):
reject(e)
def writeJSON(self, j, reject):
return self.writeMessage(0, bytes(json_dumps_clean(j), 'utf8'), reject)
return self.writeMessage(0, bytes(json.dumps(j), 'utf8'), reject)
def writeBuffer(self, buffer, reject):
return self.writeMessage(1, buffer, reject)