diff options
Diffstat (limited to 'morphlib/sourcepool.py')
-rw-r--r-- | morphlib/sourcepool.py | 51 |
1 files changed, 51 insertions, 0 deletions
diff --git a/morphlib/sourcepool.py b/morphlib/sourcepool.py new file mode 100644 index 00000000..449d7757 --- /dev/null +++ b/morphlib/sourcepool.py @@ -0,0 +1,51 @@ +# Copyright (C) 2012 Codethink Limited +# +# This program is free software; you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation; version 2 of the License. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License along +# with this program; if not, write to the Free Software Foundation, Inc., +# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. + + +class SourcePool(object): + + '''Manage a collection of Source objects.''' + + def __init__(self): + self._sources = {} + self._order = [] + + def _key(self, repo, sha1, filename): + return (repo, sha1, filename) + + def add(self, source): + '''Add a source to the pool.''' + key = self._key(source.repo, source.sha1, source.filename) + self._sources[key] = source + self._order.append(source) + + def lookup(self, repo, sha1, filename): + '''Find a source in the pool. + + Raise KeyError if it is not found. + + ''' + + key = self._key(repo, sha1, filename) + return self._sources[key] + + def __iter__(self): + '''Iterate over sources in the pool, in the order they were added.''' + for source in self._order: + yield source + + def __len__(self): + return len(self._sources) + |