generators/textsynth.py

@@ -4,6 +4,8 @@ import sqlite3
 from random import expovariate
 import typing
 import aiohttp
+from utils import http_session_factory
+
 
 async def make_sentence(cfg):
 	# set default
@@ -81,8 +83,8 @@ async def make_sentence(cfg):
 	# using aiohttp
 
 	post = None
-	while post is None:
-		async with aiohttp.ClientSession() as session:
+	async with http_session_factory() as session:
+		while post is None:
 			async with session.post(
 				"https://api.textsynth.com/v1/engines/{}/completions".format(cfg["textsynth_engine_id"]),
 				headers={
@@ -103,14 +105,14 @@ async def make_sentence(cfg):
 					continue
 				post: str = data["text"]
 
-		# check wether the post only consists of mentions
-		# split by words
-		words = post.split()
-		# check if all words are mentions
-		if all(word.startswith("@") for word in words):
-			# generate a new sentence
-			post = None
-			continue
+			# check wether the post only consists of mentions
+			# split by words
+			words = post.split()
+			# check if all words are mentions
+			if all(word.startswith("@") for word in words):
+				# generate a new sentence
+				post = None
+				continue
 
 	db.close()