sort out args for play

Author: rerpha

src/saluki/main.py

@@ -93,17 +93,15 @@ def main() -> None:
     # saluki dig - push data from dump generated by saluki bury to topic 
     # saluki dig mybroker:9092/topicname -p 0 outputfile
 
-    # saluki sniff - broker metadata ie. topic watermarks and num_messages. 
-    # saluki sniff mybroker:9092 
-
     play_parser = sub_parsers.add_parser(
         _PLAY,
         help="replay mode - replay data into another topic",
-        parents=[topic_parser],
+        parents=[],
     )
-    play_parser.add_argument("-o", "--offset", help="offsets to replay between (inclusive)", type=int, nargs=2)
-    play_parser.add_argument("-t", "--timestamp", help="timestamps to replay between", type=str, nargs=2)
-
+    play_parser.add_argument("topics", type=str, nargs=2, help="SRC topic DEST topic")
+    g = play_parser.add_mutually_exclusive_group(required=True)
+    g.add_argument("-o", "--offsets", help="offsets to replay between (inclusive)", type=int, nargs=2)
+    g.add_argument("-t", "--timestamps", help="timestamps to replay between", type=str, nargs=2)
 
     if len(sys.argv) == 1:
         parser.print_help()
@@ -127,8 +125,18 @@ def main() -> None:
             args.go_forwards,
         )
     elif args.command == _PLAY:
-        pass
-        #play(src_broker, src_topic, dest_broker, dest_topic, args.offset, args.timestamp)
+        src_broker, src_topic = parse_kafka_uri(args.topics[0])
+        dest_broker, dest_topic = parse_kafka_uri(args.topics[1])
+
+        print(f"SOURCE BROKER: {src_broker}, SOURCE TOPIC: {src_topic}, DEST BROKER: {dest_broker}, DEST TOPIC: {dest_topic} ")
+        if args.offsets is not None:
+            print(f"Replaying {src_broker}/{src_topic} between offsets {args.offsets[0]} and {args.offsets[1]} to {dest_broker}/{dest_topic}")
+        elif args.timestamps is not None:
+            print(f"Replaying {src_broker}/{src_topic} between timestamps {args.timestamps[0]} and {args.timestamps[1]} to {dest_broker}/{dest_topic}")
+
+        if input("OK? (y/n)").lower() == 'y':
+            #play(src_broker, src_topic, dest_broker, dest_topic, args.offset, args.timestamp)
+            print("replayed")
     elif args.command == _SNIFF:
         sniff(args.broker)
     elif args.command == _BURY: