fix

actions.py

@@ -1,4 +1,5 @@
-from random import randint
+from random import uniform
+from asyncio import sleep
 
 from tortoise.contrib.postgres.functions import Random
 from telethon.utils import get_input_document
@@ -273,3 +274,18 @@ async def disable_markov(peer_id):
 
 async def list_markov_chats():
     return await MarkovChat.all()
+
+async def markov_say(bot, peer_id, reply_to=None):
+    if not bot.markov.is_ready:
+        return
+
+    text = bot.markov.generate()
+
+    async with bot.action(peer_id, "typing"):
+        amount = 0
+        for _ in range(len(text)):
+            amount += round(uniform(0.05, 0.2), 2)
+
+        await sleep(min(amount, 8))
+
+    await bot.send_message(peer_id, message=text, reply_to=reply_to)

markov.py

@@ -1,7 +1,7 @@
 import os.path
-import json
 import atexit
 
+import ujson
 import markovify
 
 from config import config
@@ -56,7 +56,7 @@ class Markov:
 
         if os.path.isfile(config.MARKOV_CORPUS_PATH):
             with open(config.MARKOV_CORPUS_PATH, "r") as f:
-                self.corpus = json.load(f)
+                self.corpus = ujson.load(f)
 
     def save(self):
         if self.chain:
@@ -64,4 +64,4 @@ class Markov:
                 f.write(self.chain.to_json())
 
         with open(config.MARKOV_CORPUS_PATH, "w") as f:
-            json.dump(self.corpus, f)
+            ujson.dump(self.corpus, f)

openkriemy.py

@@ -1,5 +1,5 @@
-from random import random, uniform
-from asyncio import run, sleep
+from random import random, randomint
+from asyncio import sleep
 from datetime import datetime, timedelta, date, time
 
 from telethon import TelegramClient
@@ -18,6 +18,7 @@ from actions import (
     is_markov_enabled,
     get_markov_option,
     list_markov_chats,
+    markov_say,
 )
 from commands import COMMANDS
 from markov import Markov
@@ -28,25 +29,8 @@ bot = TelegramClient("openkriemy", config.API_ID, config.API_HASH).start(
 markov = Markov()
 
 
-async def markov_say(bot, peer_id, reply_to=None):
-    if not markov.is_ready:
-        return
-
-    text = markov.generate()
-
-    async with bot.action(peer_id, "typing"):
-        amount = 0
-        for _ in range(len(text)):
-            amount += round(uniform(0.05, 0.2), 2)
-
-        await sleep(min(amount, 8))
-
-    await bot.send_message(peer_id, message=text, reply_to=reply_to)
-
-
 # Wait isn't that illegal??
-bot.__markov = markov
-bot.__markov_say = markov_say
+bot.markov = markov
 
 
 @bot.on(NewMessage)
@@ -179,7 +163,7 @@ async def notify_birthdays_loop():
 
 async def markov_say_loop():
     while True:
-        await sleep(random(30, 60 * 5))
+        await sleep(randomint(30, 60 * 5))
 
         for chat in await list_markov_chats():
             if random() > chat.opt_message_prob:

requirements.txt

@@ -5,4 +5,5 @@ aiohttp
 aiofiles
 telethon
 emoji
-markovify
+markovify
+ujson