diff options
author | Marek Siarkowicz <mareksiarkowicz@gmail.com> | 2015-06-15 00:13:44 +0200 |
---|---|---|
committer | Marek Siarkowicz <mareksiarkowicz@gmail.com> | 2015-06-15 00:13:44 +0200 |
commit | e130a18d3c46f311d095819ab901b2fca57fcea3 (patch) | |
tree | fce9b549ace62d6085a65c584642f466f0ca17d9 /message.py | |
parent | 46123a0a78149a8832e9201f33de3a8dfa520c4b (diff) | |
download | pyParrotZikTCP-e130a18d3c46f311d095819ab901b2fca57fcea3.tar.xz pyParrotZikTCP-e130a18d3c46f311d095819ab901b2fca57fcea3.zip |
Create message class.
Diffstat (limited to '')
-rw-r--r-- | message.py | 30 |
1 files changed, 30 insertions, 0 deletions
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) |