Hexor/conf_bot
1
0
Fork 0
mirror of https://github.com/house-of-vanity/conf_bot.git synced 2025-07-06 14:24:08 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Improve markov chain text generator.

Browse Source
This commit is contained in:
AB
2020-02-28 09:53:00 +00:00
parent cded7272d2
commit 40364735e0
2 changed files with 16 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
database.py
View File

@ -165,8 +165,14 @@ class DataBase:
result = self.execute(sql)
return(result)
def get_random_message(self):
sql = "SELECT text FROM xxx_message ORDER BY RANDOM() LIMIT 1"
def get_random_message(self, conf_id=None, count=1):
if not conf_id:
print('get random message from all DB')
sql = "SELECT text FROM xxx_message ORDER BY RANDOM() LIMIT %s" % count
else:
print('get random message from %s ' % conf_id)
sql = """SELECT x.text FROM xxx_message x LEFT JOIN relations r ON r.msg_id == x.id
WHERE r.conf_id = '%s' ORDER BY RANDOM() DESC LIMIT 1""" % conf_id
result = self.execute(sql)
return(result[0][0])

10
worker.py
View File

@ -196,8 +196,14 @@ class MessageWorker:
count = max_sen
except:
count = 5
for i in range(0, count):
rand_messages.append(self.db.get_random_message())
try:
use_all = bool(msg['message']['text'][8:])
except:
use_all = False
if use_all:
rand_messages.append(self.db.get_random_message(count=count))
else:
rand_messages.append(self.db.get_random_message(conf_id, count=count))
rand_text = " ".join(rand_messages)
gen_text = get(rand_text)
try: