From fffee660e57e1f8be19ae6ffa979c48cfc6cef9e Mon Sep 17 00:00:00 2001
From: Aldo Cortesi <aldo@nullcube.com>
Date: Sat, 2 May 2015 21:42:09 +1200
Subject: language: create and use Integer and PreValue generics

---
 libpathod/language/websockets.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

(limited to 'libpathod/language/websockets.py')

diff --git a/libpathod/language/websockets.py b/libpathod/language/websockets.py
index 7ec4a2b1..b666b2fe 100644
--- a/libpathod/language/websockets.py
+++ b/libpathod/language/websockets.py
@@ -17,15 +17,23 @@ class WF(base.CaselessLiteral):
     TOK = "wf"
 
 
+class Body(base.PreValue):
+    preamble = "b"
+
+
 class WebsocketFrame(base._Message):
     comps = (
-        base.Body,
+        Body,
         base.PauseAt,
         base.DisconnectAt,
         base.InjectAt
     )
     logattrs = ["body"]
 
+    @property
+    def body(self):
+        return self.tok(Body)
+
     @classmethod
     def expr(klass):
         parts = [i.expr() for i in klass.comps]
-- 
cgit v1.2.3