Remove characters the service does not support

This commit is contained in:
rany
2021-12-05 10:05:15 +02:00
parent 4b7f04dc4c
commit f6fa6f606b

View File

@@ -45,12 +45,15 @@ def remove_incompatible_characters(string):
result in an error from the service. result in an error from the service.
Args: Args:
string (byte): The string to be cleaned. string (str or bytes): The string to be cleaned.
Returns: Returns:
byte: The cleaned string. str: The cleaned string.
""" """
cleaned_string = b"" if isinstance(string, bytes):
string = string.decode("utf-8")
cleaned_string = ""
for character in string: for character in string:
character_code = ord(character) character_code = ord(character)
if ( if (
@@ -244,7 +247,8 @@ class Communicate:
+ 50 + 50
) # margin of error ) # margin of error
messages = split_text_by_byte_length( messages = split_text_by_byte_length(
escape(messages), websocket_max_size - overhead_per_message escape(remove_incompatible_characters(messages)),
websocket_max_size - overhead_per_message,
) )
else: else:
if isinstance(messages, str): if isinstance(messages, str):