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])