python-patterns/patterns/behavioral/chaining_method.py at master · botdigit/python-patterns · GitHub

40 lines (28 loc) · 751 Bytes

#!/usr/bin/env python

# -*- coding: utf-8 -*-

from __future__ import print_function

class Person(object):

def __init__(self, name, action):

self.name = name

self.action = action

def do_action(self):

print(self.name, self.action.name, end=' ')

return self.action

class Action(object):

def __init__(self, name):

self.name = name

def amount(self, val):

print(val, end=' ')

return self

def stop(self):

print('then stop')

def main():

"""

>>> move = Action('move')

>>> person = Person('Jack', move)

>>> person.do_action().amount('5m').stop()

Jack move 5m then stop

"""

if __name__ == '__main__':

import doctest

doctest.testmod()

Expand file tree