1 files changed, 10 insertions, 6 deletions
diff --git a/lorrycontroller/__init__.py b/lorrycontroller/__init__.py
index c00e142..ddc2f74 100644
--- a/lorrycontroller/__init__.py
+++ b/lorrycontroller/__init__.py
@@ -37,14 +37,8 @@ from .removejob import RemoveJob
 from .lsupstreams import LsUpstreams, ForceLsUpstream
 from .pretendtime import PretendTime
 from .maxjobs import GetMaxJobs, SetMaxJobs
-from .gitano import (
-    GitanoCommand,
-    LocalTroveGitanoCommand,
-    GitanoCommandFailure,
-    new_gitano_command)
 from .static import StaticFile
 from .proxy import setup_proxy
-from .gitlab import Gitlab
 from . import gerrit
 from . import gitano
 from . import gitlab
@@ -59,4 +53,14 @@ downstream_types = {
 }
 
 
+upstream_types = {
+    'gitlab': gitlab.GitlabUpstream,
+    'trove':  gitano.TroveUpstream,
+}
+
+
+def get_upstream_host(host_info):
+    return upstream_types[host_info['type']](host_info)
+
+
 __all__ = locals()