bug 1092 fixing deletion of items referenced by several projects
[osm/NBI.git] / osm_nbi / tests / send_kafka.py
1 #! /usr/bin/python3
2 # -*- coding: utf-8 -*-
3
4 # Licensed under the Apache License, Version 2.0 (the "License");
5 # you may not use this file except in compliance with the License.
6 # You may obtain a copy of the License at
7 #
8 # http://www.apache.org/licenses/LICENSE-2.0
9 #
10 # Unless required by applicable law or agreed to in writing, software
11 # distributed under the License is distributed on an "AS IS" BASIS,
12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
13 # implied.
14 # See the License for the specific language governing permissions and
15 # limitations under the License.
16
17 import sys
18 import requests
19 import yaml
20 from os import getenv
21
22 __author__ = "Alfonso Tierno, alfonso.tiernosepulveda@telefonica.com"
23 __date__ = "$2019-05-31$"
24 __version__ = "0.1"
25 version_date = "May 2019"
26
27
28 def usage():
29 print("Usage: ", sys.argv[0], "topic key message")
30 print(" Sends a kafka message using URL test of NBI")
31 print(" host is defined by env OSMNBI_HOST (localhost by default)")
32 print(" port is defined by env OSMNBI_PORT (9999 by default)")
33 return
34
35
36 if __name__ == "__main__":
37 try:
38 if "--help" in sys.argv:
39 usage()
40 exit(0)
41
42 if len(sys.argv) != 4:
43 print("missing parameters. Type --help for more information", file=sys.stderr)
44 exit(1)
45
46 topic, key, message = sys.argv[1:]
47 host = getenv("OSMNBI_HOST", "localhost")
48 port = getenv("OSMNBI_PORT", "9999")
49 url = "https://{host}:{port}/osm/test/message/{topic}".format(host=host, port=port, topic=topic)
50 print(url)
51 data = {key: message}
52
53 r = requests.post(url, data=yaml.safe_dump(data), verify=False)
54 if r.status_code not in (200, 201, 202, 204):
55 print("Received code={}, content='{}'".format(r.status_code, r.text))
56 exit(1)
57 print("{} -> {}: {}".format(topic, key, message))
58
59 except Exception:
60 raise