added clipcascade

ClipCascade/stomp_ws/frame.py

@@ -0,0 +1,51 @@
+from urllib.parse import urlparse
+
+Byte = {"LF": "\x0A", "NULL": "\x00"}
+
+
+class Frame:
+
+    def __init__(self, command, headers, body):
+        self.command = command
+        self.headers = headers
+        self.body = "" if body is None else body
+
+    def __str__(self):
+        lines = [self.command]
+        skipContentLength = "content-length" in self.headers
+        if skipContentLength:
+            del self.headers["content-length"]
+
+        for name in self.headers:
+            value = self.headers[name]
+            lines.append("" + name + ":" + value)
+
+        if self.body is not None and not skipContentLength:
+            lines.append("content-length:" + str(len(self.body)))
+
+        lines.append(Byte["LF"] + self.body)
+        return Byte["LF"].join(lines)
+
+    @staticmethod
+    def unmarshall_single(data):
+        lines = data.split(Byte["LF"])
+
+        command = lines[0].strip()
+        headers = {}
+
+        # get all headers
+        i = 1
+        while lines[i] != "":
+            # get key, value from raw header
+            (key, value) = lines[i].split(":", 1)
+            headers[key] = value
+            i += 1
+
+        # set body to None if there is no body
+        body = None if lines[i + 1] == Byte["NULL"] else lines[i + 1][:-1]
+
+        return Frame(command, headers, body)
+
+    @staticmethod
+    def marshall(command, headers, body):
+        return str(Frame(command, headers, body)) + Byte["NULL"]