1 files changed, 53 insertions, 0 deletions
diff --git a/diplomacy/daide/utils.py b/diplomacy/daide/utils.py
new file mode 100644
index 0000000..b6ef60d
--- /dev/null
+++ b/diplomacy/daide/utils.py
@@ -0,0 +1,53 @@
+# ==============================================================================
+# Copyright (C) 2019 - Philip Paquette
+#
+#  This program is free software: you can redistribute it and/or modify it under
+#  the terms of the GNU Affero General Public License as published by the Free
+#  Software Foundation, either version 3 of the License, or (at your option) any
+#  later version.
+#
+#  This program is distributed in the hope that it will be useful, but WITHOUT
+#  ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
+#  FOR A PARTICULAR PURPOSE.  See the GNU Affero General Public License for more
+#  details.
+#
+#  You should have received a copy of the GNU Affero General Public License along
+#  with this program.  If not, see <https://www.gnu.org/licenses/>.
+# ==============================================================================
+""" Settings - Contains a list of utils to help handle DAIDE communication """
+from diplomacy.daide.tokens import is_integer_token, Token
+
+def str_to_bytes(daide_str):
+    """ Converts a str into its bytes representation
+        :param daide_str: A DAIDE string with tokens separated by spaces
+        :return: The bytes representation of the string
+
+        Note: Integers starts with a '#' character
+    """
+    buffer = []
+    str_split = daide_str.split(' ') if daide_str else []
+    for word in str_split:
+        if word == '':
+            buffer.append(bytes(Token(from_str=' ')))
+        elif word[0] == '#':
+            buffer.append(bytes(Token(from_int=int(word[1:]))))
+        else:
+            buffer.append(bytes(Token(from_str=word)))
+    return b''.join(buffer)
+
+def bytes_to_str(daide_bytes):
+    """ Converts a bytes into its str representation
+        :param daide_bytes: A DAIDE bytes with tokens separated by spaces
+        :return: The bytes representation of the string
+
+        Note: Integers starts with a '#' character
+    """
+    buffer = []
+    length = len(daide_bytes) if daide_bytes else 0
+    for i in range(0, length, 2):
+        token = Token(from_bytes=(daide_bytes[i], daide_bytes[i + 1]))
+        if is_integer_token(token):
+            buffer.append('#' + str(token))
+        else:
+            buffer.append(str(token))
+    return ' '.join(buffer)