Package pyrobase :: Module iterutil
[hide private]
[frames] | no frames]

Source Code for Module pyrobase.iterutil

 1  # -*- coding: utf-8 -*- 
 2  """ Helpers for iterables and collections. 
 3   
 4      Copyright (c) 2011 The PyroScope Project <pyroscope.project@gmail.com> 
 5  """ 
 6  # This program is free software; you can redistribute it and/or modify 
 7  # it under the terms of the GNU General Public License as published by 
 8  # the Free Software Foundation; either version 2 of the License, or 
 9  # (at your option) any later version. 
10  # 
11  # This program is distributed in the hope that it will be useful, 
12  # but WITHOUT ANY WARRANTY; without even the implied warranty of 
13  # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the 
14  # GNU General Public License for more details. 
15  # 
16  # You should have received a copy of the GNU General Public License along 
17  # with this program; if not, write to the Free Software Foundation, Inc., 
18  # 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. 
19   
20 -def flatten(nested, containers=(list, tuple)):
21 """ Flatten a nested list by yielding its scalar items. 22 """ 23 for item in nested: 24 if hasattr(item, "next") or isinstance(item, containers): 25 for subitem in flatten(item): 26 yield subitem 27 else: 28 yield item
29