summaryrefslogtreecommitdiffstats
path: root/dmaap/dmaapplugin/dr_relationships.py
diff options
context:
space:
mode:
Diffstat (limited to 'dmaap/dmaapplugin/dr_relationships.py')
-rw-r--r--dmaap/dmaapplugin/dr_relationships.py17
1 files changed, 11 insertions, 6 deletions
diff --git a/dmaap/dmaapplugin/dr_relationships.py b/dmaap/dmaapplugin/dr_relationships.py
index dd67394..2f342f3 100644
--- a/dmaap/dmaapplugin/dr_relationships.py
+++ b/dmaap/dmaapplugin/dr_relationships.py
@@ -148,14 +148,17 @@ def add_dr_subscriber(**kwargs):
# Get the parameters for the call
feed_id = ctx.target.instance.runtime_properties["feed_id"]
- location = ctx.source.instance.runtime_properties[target_feed]["location"]
- delivery_url = ctx.source.instance.runtime_properties[target_feed]["delivery_url"]
- username = ctx.source.instance.runtime_properties[target_feed]["username"]
- password = ctx.source.instance.runtime_properties[target_feed]["password"]
+ feed = ctx.source.instance.runtime_properties[target_feed]
+ location = feed["location"]
+ delivery_url = feed["delivery_url"]
+ username = feed["username"]
+ password = feed["password"]
+ decompress = feed["decompress"] if "decompress" in feed else False
+ privileged = feed["privileged"] if "privileged" in feed else False
# Make the request to add the subscriber to the feed
dmc = DMaaPControllerHandle(DMAAP_API_URL, DMAAP_USER, DMAAP_PASS, ctx.logger)
- add_sub = dmc.add_subscriber(feed_id, location, delivery_url,username, password)
+ add_sub = dmc.add_subscriber(feed_id, location, delivery_url,username, password, decompress, privileged)
add_sub.raise_for_status()
subscriber_info = add_sub.json()
subscriber_id = subscriber_info["subId"]
@@ -168,7 +171,9 @@ def add_dr_subscriber(**kwargs):
"location" : location,
"delivery_url" : delivery_url,
"username" : username,
- "password" : password
+ "password" : password,
+ "decompress": decompress,
+ "privilegedSubscriber": privileged
}
ctx.logger.info("on source: {0}".format(ctx.source.instance.runtime_properties[target_feed]))