add security modes support:SSL, PLAINTEXT.

Author: stanislav-gabdulgaziev

main.py

@@ -23,14 +23,15 @@
 
 
 # function produce_msgs starts producing messages with Faker
-def produce_msgs(cert_folder = '~/kafka-pizza/',
+def produce_msgs(security_protocol='SSL',
+ cert_folder = '~/kafka-pizza/',
  hostname = 'hostname',
  port = '1234',
  topic_name = 'pizza-orders',
  nr_messages = -1,
  max_waiting_time_in_sec = 5,
  subject = 'pizza'):
- if cert_folder.lower() == 'plaintext':
+ if security_protocol.upper() == 'PLAINTEXT':
  producer = KafkaProducer(
  bootstrap_servers=hostname + ':' + port,
  security_protocol='PLAINTEXT',
@@ -92,20 +93,23 @@ def produce_msgs(cert_folder = '~/kafka-pizza/',
 
 def main():
  parser = argparse.ArgumentParser()
- parser.add_argument('--cert-folder', help='Path to folder containing required Kafka certificates or use plaintext to work with unsecure Kafka', required=True)
+ parser.add_argument('--security-protocol', help='Security protocol for Kafka (PLAINTEXT, SSL)', required=True)
+ parser.add_argument('--cert-folder', help='Path to folder containing required Kafka certificates. Required --security-protocol equal SSL', required=False)
  parser.add_argument('--host', help='Kafka Host (obtained from Aiven console)', required=True)
  parser.add_argument('--port', help='Kafka Port (obtained from Aiven console)', required=True)
  parser.add_argument('--topic-name', help='Topic Name', required=True)
  parser.add_argument('--nr-messages', help='Number of messages to produce (0 for unlimited)', required=True)
  parser.add_argument('--max-waiting-time', help='Max waiting time between messages (0 for none)', required=True)
  parser.add_argument('--subject', help='What type of content to produce (possible choices are [pizza, userbehaviour, stock, realstock, metric] pizza is the default', required=False)
  args = parser.parse_args()
+ p_security_protocol = args.security_protocol
  p_cert_folder =args.cert_folder
  p_hostname =args.host
  p_port =args.port
  p_topic_name=args.topic_name
  p_subject=args.subject
- produce_msgs(cert_folder=p_cert_folder,
+ produce_msgs(security_protocol=p_security_protocol,
+ cert_folder=p_cert_folder,
  hostname=p_hostname,
  port=p_port,
  topic_name=p_topic_name,