Move TwiML to unique types

twilio · codejudas · Mar 7, 2017 · Mar 2, 2017 · Mar 2, 2017 · Mar 2, 2017
commit 0af4776a961a5b0fef8052082fb1f4027cff172e
diff --git a/twilio/twiml/__init__.py b/twilio/twiml/__init__.py
@@ -0,0 +1,76 @@
+import xml.etree.ElementTree as ET
+
+
+def lower_camel(string):
+    result = "".join([x.title() for x in string.split('_')])
+    if not result:
+        return result
+
+    return result[0].lower() + result[1:]
+
+
+class TwiML(object):
+    """
+    Twilio basic verb object.
+    """
+    MAP = {
+        'from_': 'from'
+    }
+
+    def __init__(self, **kwargs):
+        self.name = self.__class__.__name__
+        self.body = None
+        self.verbs = []
+        self.attrs = {}
+
+        for k, v in kwargs.items():
+            if v is not None:
+                self.attrs[lower_camel(self.MAP.get(k, k))] = v
+
+    def __str__(self):
+        return self.to_xml()
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_value, traceback):
+        return False
+
+    def to_xml(self, xml_declaration=True):
+        """
+        Return the contents of this verb as an XML string
+
+        :param bool xml_declaration: Include the XML declaration. Defaults to
+                                     True
+        """
+        xml = ET.tostring(self.xml()).decode('utf-8')
+
+        if xml_declaration:
+            return '<?xml version="1.0" encoding="UTF-8"?>' + xml
+        else:
+            return xml
+
+    def xml(self):
+        el = ET.Element(self.name)
+
+        keys = self.attrs.keys()
+        keys = sorted(keys)
+        for a in keys:
+            value = self.attrs[a]
+
+            if isinstance(value, bool):
+                el.set(a, str(value).lower())
+            else:
+                el.set(a, str(value))
+
+        if self.body:
+            el.text = self.body
+
+        for verb in self.verbs:
+            el.append(verb.xml())
+
+        return el
+
+    def append(self, verb):
+        self.verbs.append(verb)
+        return verb
diff --git a/twilio/twiml/messaging_response.py b/twilio/twiml/messaging_response.py
@@ -0,0 +1,64 @@
+from twilio.twiml import TwiML
+
+
+class MessagingResponse(TwiML):
+
+    def __init__(self):
+        super(MessagingResponse, self).__init__()
+        self.name = 'Response'
+
+    def message(self,
+                body,
+                to=None,
+                from_=None,
+                method=None,
+                action=None,
+                status_callback=None,
+                **kwargs):
+        return self.append(Message(
+            body=body,
+            to=to,
+            from_=from_,
+            method=method,
+            action=action,
+            status_callback=status_callback,
+            **kwargs
+        ))
+
+    def redirect(self, method=None, url=None, **kwargs):
+        return self.append(Redirect(
+            method=method,
+            url=url,
+            **kwargs
+        ))
+
+
+class Message(TwiML):
+
+    def __init__(self, body=None, **kwargs):
+        super(Message, self).__init__(**kwargs)
+        if body:
+            self.body = body
+
+    def body(self, body):
+        return self.append(Body(body))
+
+    def media(self, url):
+        return self.append(Media(url))
+
+
+class Body(TwiML):
+    def __init__(self, body):
+        super(Body, self).__init__()
+        self.body = body
+
+
+class Media(TwiML):
+    def __init__(self, url):
+        super(Media, self).__init__()
+        self.body = url
+
+
+class Redirect(TwiML):
+    pass
+