minor remote fixes

2 files changed, 6 insertions, 1 deletions

diff --git a/pomu/repo/remote/git.py b/pomu/repo/remote/git.py
index 42db11a..2738c1a 100644
--- a/pomu/repo/remote/git.py
+++ b/pomu/repo/remote/git.py
@@ -1,5 +1,6 @@
 """A class for remote git repos"""
 from os import chdir, path
+from shutil import rmtree
 from subprocess import call
 from tempfile import mkdtemp
 
@@ -66,3 +67,6 @@ class RemoteGitRepo(RemoteRepo):
         if k not in dic:
             return Result.Err()
         return parse_object(self.get_object(dic[k]))
+
+    def cleanup(self):
+        rmtree(self.dir)
diff --git a/pomu/repo/remote/remote.py b/pomu/repo/remote/remote.py
index b8815ea..1c2a6df 100644
--- a/pomu/repo/remote/remote.py
+++ b/pomu/repo/remote/remote.py
@@ -15,6 +15,7 @@ class RemoteRepo():
         tp = RemoteRepo.type_for_name(type_)
         if not tp:
             from pomu.repo.remote.git import RemoteGitRepo
+            from pomu.repo.remote.hg import RemoteHgRepo
             from pomu.repo.remote.rsync import RemoteRsyncRepo
             from pomu.repo.remote.svn import RemoteSvnRepo
             try:
@@ -58,7 +59,7 @@ class RemoteRepo():
         return Package(name, '/', None, category, version, filemap=filemap)
 
     def list_cpvs(self):
-        """Gets a list of all pebuilds in the repo"""
+        """Gets a list of all ebuilds in the repo"""
         return filelist_to_cpvs(self.fetch_tree())
 
     def fetch_tree(self):