Added the Markov Chain kata.

markov/markov.py

@@ -0,0 +1,48 @@
+#!/usr/bin/env python3
+#
+# Copyright 2023 David Soulayrol.
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the “Software”), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+
+import args
+import json
+import random
+import sys
+
+
+def select_next_word(word, stats):
+    state = stats[word]
+    choices = random.choices(list(state.keys()), weights=list(state.values()))
+
+    return choices[0]
+
+
+if __name__ == '__main__':
+    properties = args.parse('f:filename:s,w:first_word:s,n:words:d', sys.argv[1:])
+    stats = {}
+
+    if len(properties.filename) == 0:
+        print('Missing statistics file.')
+        sys.exit(1)
+
+    with open(properties.filename, 'r') as f:
+        stats = json.loads(f.read())
+
+    n = properties.words if properties.words > 2 else 24
+    word = properties.first_word if len(properties.first_word) \
+        else random.choice(list(stats.keys()))
+
+    if word not in stats.keys():
+        print('Unavailable word "%s".' % word)
+        sys.exit(1)
+
+    print(word, end=' ')
+
+    for i in range(min(n - 1, 1000)):
+        word = select_next_word(word, stats)
+        print(word, end=' ')