csit/resources/scripts/make_topics.py

   1 #!/usr/bin/env python3
   2 #
   3 # ============LICENSE_START====================================================
   4 #  Copyright (C) 2023 Nordix Foundation.
   5 # =============================================================================
   6 # Licensed under the Apache License, Version 2.0 (the "License");
   7 # you may not use this file except in compliance with the License.
   8 # You may obtain a copy of the License at
   9 #
  10 #      http://www.apache.org/licenses/LICENSE-2.0
  11 #
  12 # Unless required by applicable law or agreed to in writing, software
  13 # distributed under the License is distributed on an "AS IS" BASIS,
  14 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15 # See the License for the specific language governing permissions and
  16 # limitations under the License.
  17 #
  18 # SPDX-License-Identifier: Apache-2.0
  19 # ============LICENSE_END======================================================
  20
  21 # Python utility to create a new kafka topic
  22 # Accepts the argument {topic_name}
  23
  24 from confluent_kafka.admin import AdminClient, NewTopic
  25 import sys
  26
  27 def create_topic(bootstrap_servers, topic_name, num_partitions=2, replication_factor=2):
  28     admin_client = AdminClient({'bootstrap.servers': bootstrap_servers})
  29
  30     # Define the topic configuration
  31     topic = NewTopic(topic_name, num_partitions=num_partitions, replication_factor=replication_factor)
  32
  33     # Create the topic
  34     admin_client.create_topics([topic])
  35
  36
  37 if __name__ == '__main__':
  38     topic_name = sys.argv[1]
  39     bootstrap_servers = 'localhost:29092'
  40
  41     create_topic(bootstrap_servers, topic_name)