Add extended encoding message validation

- validates variables of the "message" type
This commit is contained in:
Peter Šurda 2016-11-15 17:10:33 +01:00
parent e76b10a6ed
commit 7cc0f8fab0
Signed by untrusted user: PeterSurda
GPG Key ID: 0C5F50C0B5F37D87
1 changed files with 10 additions and 2 deletions

View File

@ -1,13 +1,21 @@
from debug import logger
from messagetypes import MsgBase
class Message(MsgBase):
def __init__(self):
def decode(self, data):
self.subject = data["subject"]
self.body = data["body"]
# UTF-8 and variable type validator
if type(data["subject"]) is str:
self.subject = unicode(data["subject"], 'utf-8', 'replace')
self.subject = unicode(str(data["subject"]), 'utf-8', 'replace')
if type(data["body"]) is str:
self.body = unicode(data["body"], 'utf-8', 'replace')
self.body = unicode(str(data["body"]), 'utf-8', 'replace')
def encode(self, data):
super(Message, self).encode()