Assure that portage repo initialization tests pass

1 files changed, 7 insertions, 3 deletions

diff --git a/pomu/repo/repo.py b/pomu/repo/repo.py
index e8693ce..065c990 100644
--- a/pomu/repo/repo.py
+++ b/pomu/repo/repo.py
@@ -3,15 +3,18 @@ from os import path
 
 import portage
 
-rsets = portage.db[portage.root]['vartree'].settings.repositories
-
 def portage_repos():
+    """Yield the repositories configured for portage"""
+    rsets = portage.db[portage.root]['vartree'].settings.repositories
+
     for repo in rsets.prepos_order:
         yield repo
 
 def portage_repo_path(repo):
     """Get the path of a given portage repository"""
-    if repo in rsets.prepos_order:
+    rsets = portage.db[portage.root]['vartree'].settings.repositories
+
+    if repo in rsets.prepos:
         return rsets.prepos[repo].location
     return None
 
@@ -20,6 +23,7 @@ def pomu_status(repo_path):
     return path.isdir(path.join(repo_path, 'metadata', 'pomu'))
 
 def pomu_active_repo():
+    """Returns a portage repo, for which pomu is enabled"""
     for repo in portage_repos():
         if pomu_status(portage_repo_path(repo)):
             return repo