Python: jQuery-подобная цепочка функций?

Вот расширение предложения Саймона ListMutator:

class ListMutator(object):

    def __init__(self, seq):
        self.data = seq

    def foo1(self, arg1, arg2):
        self.data = [x + arg1 for x in self.data]
        # This allows chaining:
        return self

    def foo2(self, arg1, arg2):
        self.data = [x*arg1 for x in self.data]
        return self

if __name__ == "__main__":
    lm = ListMutator([1,2,3,4])
    lm.foo1(2, 0).foo2(10, 0)
    print lm.data

    # Or, if you really must:
    print ListMutator([1,2,3,4]).foo1(2, 0).foo2(10, 0).data

Вы могли бы пойти еще лучше и заставить ListMutator вести себя как список, используя коллекции абстрактных базовых классов. На самом деле, вы можете подклассировать сам list, хотя это может ограничить вас от выполнения определенных действий, которые вам могут понадобиться... и я не знаю, каково общее мнение о создании подклассов для встроенных типов, таких как list.

Пока функция возвращает значение, вы можете связать ее. В jQuery метод селектора обычно возвращает сам селектор, что позволяет вам выполнять цепочку. Если вы хотите реализовать цепочку в python, вы можете сделать что-то вроде этого:

class RoboPuppy:

  def bark(self):
    print "Yip!"
    return self

  def growl(self):
    print "Grr!"
    return self

pup = RoboPuppy()
pup.bark().growl().bark()  # Yip! Grr! Yip!

Однако ваша проблема, похоже, заключается в том, что аргументы вашей функции слишком ограничены. Цепочка не является решением этой проблемы. Если вы хотите сжать аргументы вашей функции, просто назначьте аргументы переменным, прежде чем передавать их в функцию, например:

spam = foo(arg1, arg2)
eggs = bar(spam, arg1, arg2)
ham = foobar(eggs, args)

Если говорить о методах объекта, то это тривиально, просто return self от каждого метода. С другой стороны, если вы хотите связать несвязанные функции, мне не имеет смысла связывать их так, как вы хотите. Конечно, это выглядит красиво, однако семантически бессвязно, потому что "." означает доступ к атрибутам объекта, а не «цепочку».

На будущее: взгляните на Moka, минималистский библиотека функционального программирования. Из их примеров:

(List()                    # Create a new instance of moka.List
   .extend(range(1,20))    # Insert the numbers from 1 to 20
   .keep(lambda x: x > 5)  # Keep only the numbers bigger than 5
   .rem(operator.gt, 7)    # Remove the numbers bigger than 7 using partial application
   .rem(eq=6)              # Remove the number 6 using the 'operator shortcut'
   .map(str)               # Call str on each numbers (Creating a list of string)
   .invoke('zfill', 3)     # Call zfill(x, 3) on each string (Filling some 0 on the left)
   .insert(0, 'I am')      # Insert the string 'I am' at the head of the list
   .join(' '))             # Joining every string of the list and separate them with a space.

>>> 'I am 007'

Взгляните на это. Это простой класс-оболочка для цепочки. И он реализовал некоторые функции библиотеки underscore.js. Вы оборачиваете свой список, кортеж или словарь символом подчеркивания и играете с ним, а затем получаете из него значение, добавляя еще одно подчеркивание.

print (_([1,2,3])
       .map(lambda x: x+1)
       .reverse()
       .dict_keys(["a", "b", "c"])
       .invert()
       .items()
       .append(("this is what you got", "chaining"))
       .dict()._)

выход:

{2: 'c', 3: 'b', 4: 'a', 'this is what you got': 'chaining'}