mirror of
https://gitlab.com/vylion/velascobot.git
synced 2025-04-19 13:36:36 +02:00
d075624263
📝 Changed the whole script files hierarchy: - velasco.py starts up the telegram bot - speaker.py has all the bot behavior methods - A Parrot is what stores a Markov object - An Scribe stores a single chat's data - A Speaker has an Scriptorium, with all active Scribes - A Speaker has a single Parrot, the one associated with the last Scribe that had to send a message - An Archivist is in charge of loading the Scriptorium at startup, as well as storing Scribes and Parrots in files 🐞 Fixed a bug that stopped new Parrots from being saved, prevented because their non-existing file could not be loaded into the Speaker's Parrot
106 lines
3.7 KiB
Python
106 lines
3.7 KiB
Python
#!/usr/bin/env python3
|
|
|
|
def parse(l):
|
|
s = l.split('=', 1)
|
|
if len(s) < 2:
|
|
return ""
|
|
else:
|
|
return s[1]
|
|
|
|
class Chatlog(object):
|
|
def __init__(self, cid, ctype, title, count=0, freq=None, answer=0.5, restricted=False, silenced=False):
|
|
self.id = str(cid)
|
|
self.type = ctype
|
|
self.title = title
|
|
if freq is None:
|
|
if "group" in ctype:
|
|
freq = 10
|
|
#elif ctype is "private":
|
|
else:
|
|
freq = 2
|
|
self.count = count
|
|
self.freq = freq
|
|
self.answer = answer
|
|
self.restricted = restricted
|
|
self.silenced = silenced
|
|
|
|
def add_msg(self, message):
|
|
self.gen.add_text(message)
|
|
self.count += 1
|
|
|
|
def set_freq(self, freq):
|
|
if freq < 1:
|
|
raise ValueError('Tried to set freq a value less than 1.')
|
|
else:
|
|
self.freq = freq
|
|
return self.freq
|
|
|
|
def set_answer(self, afreq):
|
|
if afreq > 1:
|
|
raise ValueError('Tried to set answer probability higher than 1.')
|
|
elif afreq < 0:
|
|
raise ValueError('Tried to set answer probability lower than 0.')
|
|
else:
|
|
self.answer = afreq
|
|
return self.answer
|
|
|
|
def dumps(self):
|
|
lines = ["LOG=v4"]
|
|
lines.append("CHAT_ID=" + self.id)
|
|
lines.append("CHAT_TYPE=" + self.type)
|
|
lines.append("CHAT_NAME=" + self.title)
|
|
lines.append("WORD_COUNT=" + str(self.count))
|
|
lines.append("MESSAGE_FREQ=" + str(self.freq))
|
|
lines.append("ANSWER_FREQ=" + str(self.answer))
|
|
lines.append("RESTRICTED=" + str(self.restricted))
|
|
lines.append("SILENCED=" + str(self.silenced))
|
|
#lines.append("WORD_DICT=")
|
|
return '\n'.join(lines)
|
|
|
|
def loads(text):
|
|
lines = text.splitlines()
|
|
return Chatlog.loadl(lines)
|
|
|
|
def loadl(lines):
|
|
version = parse(lines[0]).strip()
|
|
version = version if len(version.strip()) > 1 else (lines[4] if len(lines) > 4 else "LOG_ZERO")
|
|
if version == "v4":
|
|
return Chatlog(cid=parse(lines[1]),
|
|
ctype=parse(lines[2]),
|
|
title=parse(lines[3]),
|
|
count=int(parse(lines[4])),
|
|
freq=int(parse(lines[5])),
|
|
answer=float(parse(lines[6])),
|
|
restricted=(parse(lines[7]) == 'True'),
|
|
silenced=(parse(lines[8]) == 'True')
|
|
)
|
|
elif version == "v3":
|
|
return Chatlog(cid=parse(lines[1]),
|
|
ctype=parse(lines[2]),
|
|
title=parse(lines[3]),
|
|
count=int(parse(lines[7])),
|
|
freq=int(parse(lines[4])),
|
|
answer=float(parse(lines[5])),
|
|
restricted=(parse(lines[6]) == 'True')
|
|
)
|
|
elif version == "v2":
|
|
return Chatlog(cid=parse(lines[1]),
|
|
ctype=parse(lines[2]),
|
|
title=parse(lines[3]),
|
|
count=int(parse(lines[6])),
|
|
freq=int(parse(lines[4])),
|
|
answer=float(parse(lines[5]))
|
|
)
|
|
elif version == "dict:":
|
|
return Chatlog(cid=lines[0],
|
|
ctype=lines[1],
|
|
title=lines[2],
|
|
count=int(lines[5]),
|
|
freq=int(lines[3])
|
|
)
|
|
else:
|
|
return Chatlog(cid=lines[0],
|
|
ctype=lines[1],
|
|
title=lines[2],
|
|
freq=int(lines[3])
|
|
)
|