Add SourcePool class and make Source store its morphology filename

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)
+