pythondslfluent-interface

Fluent interface with Python


I have a Python function "send_message" which takes three arguments:

send_message("i like windmills", to="INBOX", from="OUTBOX")

I am thinking about putting a fluent interface on top of it. Ideally I'd like to write any of the following:

send_message("i like windmills").to("INBOX").from("OUTBOX")

send_message("i like windmills").from("OUTBOX").to("INBOX")

# The `to()` information is mandatory but the `from()` is not (as with real letters), so this one would also be a valid call:
send_message("i like windmills").to("INBOX")

Any ideas how to accomplish this or something similar?

The general approach of having methods of an object returning "self" is understood by me but in my understanding this would lead to something like this:

message = Message("i like windmills")
message.to("INBOX").from("OUTBOX").send()

But this one is not as nice as the previous example and I then would actually prefer the original version with the named arguments.

Any help is appreciated.


Solution

  • It can be accomplished this way, I am unsure if there is a better way because this is my first attempt. Good luck!

    DEFAULT_SENDER = 'my_address'
    #Because the sender object is optional I assume you have a default sender
    
    class Send_message(object):
        def __init__(self, message):
            self.message = message
            self.sender = None
            self.receiver = None
            self.method = None
    
        def to(self, receiver):
            self.receiver = receiver
            self.method = self.send()
            return self
    
        def _from(self, sender):
            self.sender = sender
            self.method = self.send()
            return self
    
        def __call__(self):
            if self.method:
                return self.method()
            return None
    
        def send(self):
            if self.receiver:
                if not self.sender:
                    self.sender = DEFAULT_SENDER
    
                return lambda:actual_message_code(self.message, self.sender, self.receiver)
    
    
    def actual_message_code(message, sender, receiver):
        print "Sent '{}' from: {} to {}.".format(message, sender, receiver)
    
    
    
    Send_message("Hello")._from('TheLazyScripter').to('samba2')()
    Send_message("Hello").to('samba2')._from('TheLazyScripter')()
    Send_message("Hello").to('samba2')()
    
    #Only change in actual calling is the trailing ()
    

    By implementing the __call__ method we can tell the when we are at the end of the call chain. This of course adds the trailing () call. and requires you to change the pointer to your actual messaging method and default sender variable but I feel that this would be the simplest way to accomplish your goals without actually knowing when the chain ends.