diff options
author | Lars Wirzenius <lars.wirzenius@codethink.co.uk> | 2012-04-10 16:12:06 +0100 |
---|---|---|
committer | Lars Wirzenius <lars.wirzenius@codethink.co.uk> | 2012-04-10 16:28:13 +0100 |
commit | caaa37d42dec002ffe20f2152767f3ce37cb1bf5 (patch) | |
tree | 9bfb0e630f0688aeb31940212da5c32e738cf74f /morphlib/sourcepool.py | |
parent | 7ec6cd23cefe5e116a54c6c0cb2aa0669adcd991 (diff) | |
download | morph-caaa37d42dec002ffe20f2152767f3ce37cb1bf5.tar.gz |
Add SourcePool class and make Source store its morphology filename
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) + |