Fix body and preamble properties

1 files changed, 11 insertions, 4 deletions

diff --git a/viewstate.py b/viewstate.py
index b4c162c..01417a4 100644
--- a/viewstate.py
+++ b/viewstate.py
@@ -9,8 +9,11 @@ CONSTS = {
     '\x68': False
 }
 
-def parse():
-    pass
+def parse(b):
+    print([hex(x) for x in b])
+    if len(b) == 1 and b in CONSTS:
+        return CONSTS[b]
+    return None
 
 
 class ViewState(object):
@@ -21,14 +24,18 @@ class ViewState(object):
         self.decoded = None
 
     @property
-    def body(self):
+    def preamble(self):
         return self.raw[:2]
 
+    @property
+    def body(self):
+        return self.raw[2:]
+
     def is_valid(self):
         format_marker = b'\xff'
         version_marker = b'\x01'
         preamble = format_marker + version_marker
-        return self.body == preamble
+        return self.preamble == preamble
 
     def decode(self):
         if not self.is_valid():