Support Aurora Master Proxy

tellapart/aurproxy/source/__init__.py

@@ -16,6 +16,7 @@
 from .source import ProxySource
 from .sources.api import ApiSource
 from .sources.aurora import AuroraProxySource
+from .sources.aurora import AuroraMasterProxySource
 from .sources.mesos import MesosMasterProxySource
 from .sources.static import StaticProxySource
 from .sources.serverset import ServerSetSource

tellapart/aurproxy/source/sources/aurora.py

@@ -17,6 +17,7 @@
 from .serverset import ServerSetSource
 
 _DEFAULT_ANNOUNCER_SERVERSET_PATH = '/aurora/'
+_DEFAULT_AURORA_SCHEDULER_PATH = '/aurora/scheduler'
 
 
 def get_service_discovery_path(job, announcer_serverset_path):
@@ -42,3 +43,13 @@ def __init__(self,
         announcer_serverset_path
     )
     super(AuroraProxySource, self).__init__(serverset_path, zk_servers, **kw)
+
+
+class AuroraMasterProxySource(ServerSetSource):
+
+  def __init__(self,
+               zk_servers,
+               aurora_master_path=_DEFAULT_AURORA_SCHEDULER_PATH,
+               **kw):
+
+    super(AuroraMasterProxySource, self).__init__(aurora_master_path, zk_servers, **kw)