From e130a18d3c46f311d095819ab901b2fca57fcea3 Mon Sep 17 00:00:00 2001
From: Marek Siarkowicz <mareksiarkowicz@gmail.com>
Date: Mon, 15 Jun 2015 00:13:44 +0200
Subject: Create message class.

---
 message.py | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)
 create mode 100644 message.py

(limited to 'message.py')

diff --git a/message.py b/message.py
new file mode 100644
index 0000000..214b6f9
--- /dev/null
+++ b/message.py
@@ -0,0 +1,30 @@
+class Message:
+    def __init__(self, resource, method, arg=None):
+        self.method = method
+        self.resource = resource
+        self.arg = arg
+
+    def __str__(self):
+        return str(self.request)
+
+    @property
+    def request(self):
+        message = bytearray()
+        message.extend(self.header)
+        message.extend(bytearray(self.request_string))
+        return message
+
+    @property
+    def header(self):
+        header = bytearray([0])
+        header.append(len(self.request_string) + 3)
+        header.append("\x80")
+        return header
+
+    @property
+    def request_string(self):
+        if self.method == 'set':
+            return 'SET {}/{}?arg={}'.format(self.resource, self.method,
+                                             str(self.arg).lower())
+        else:
+            return 'GET {}/{}'.format(self.resource, self.method)
-- 
cgit v1.2.1