Created
January 3, 2012 00:08
-
-
Save alexmacedo/1552724 to your computer and use it in GitHub Desktop.
Unix pipeline pattern in python
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#! /usr/bin/env python | |
class Pipeline(object): | |
def __init__(self): | |
self.source = None | |
def __iter__(self): | |
return self.generator() | |
def generator(self): | |
while True: | |
value = self.source.next() | |
if self.filter(value): | |
yield self.map(value) | |
def __or__(self, other): | |
other.source = self.generator() | |
return other | |
def filter(self, value): | |
return True | |
def map(self, value): | |
return value | |
class AllNumbers(Pipeline): | |
def generator(self): | |
value = 0 | |
while True: | |
yield value | |
value += 1 | |
class Evens(Pipeline): | |
def filter(self, value): | |
return value % 2 == 0 | |
class MultipleOf(Pipeline): | |
def __init__(self, factor=1): | |
self.factor = factor | |
super(MultipleOf, self).__init__() | |
def filter(self, value): | |
return value % self.factor == 0 | |
class Printer(Pipeline): | |
def map(self, value): | |
print value | |
return value | |
class First(Pipeline): | |
def __init__(self, total=10): | |
self.total = total | |
self.count = 0 | |
super(First, self).__init__() | |
def map(self, value): | |
self.count += 1 | |
if self.count > self.total: | |
raise StopIteration | |
return value | |
def main(): | |
all_numbers = AllNumbers() | |
evens = MultipleOf(2) | |
multiple_of_3 = MultipleOf(3) | |
printer = Printer() | |
first_10 = First(10) | |
pipeline = all_numbers | evens | multiple_of_3 | first_10 | printer | |
for i in pipeline: | |
pass | |
if __name__ == '__main__': | |
main() |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment