TIMEOUT = 240
USERNAME = "plomlombot"
NICKNAME = USERNAME
+TWTFILE = ""
class ExceptionForRestart(Exception):
msg = ""
while 1:
new_end = markov(snippet)
- for name in session.uses_in_chan:
+ for name in session.users_in_chan:
if new_end[:len(name)] == name.lower():
new_end = "malkovich" + new_end[len(name):]
break
msg = msg.replace(url_escape, choice(urls), 1)
notice(msg + "malkovich.")
+ def twt():
+ def try_open(mode):
+ try:
+ twtfile = open(session.twtfile, mode)
+ except (PermissionError, FileNotFoundError) as err:
+ notice("CAN'T ACCESS OR CREATE TWT FILE: " + str(err))
+ return None
+ return twtfile
+
+ from datetime import datetime
+ if not os.access(session.twtfile, os.F_OK):
+ twtfile = try_open("w")
+ if None == twtfile:
+ return
+ twtfile.close()
+ twtfile = try_open("a")
+ if None == twtfile:
+ return
+ twtfile.write(datetime.utcnow().isoformat() + "\t" + argument + "\n")
+ twtfile.close()
+ notice("WROTE TWT.")
+
if "addquote" == command:
addquote()
elif "quote" == command:
quote()
elif "markov" == command:
markov()
+ elif "twt" == command:
+ twt()
def handle_url(url, notice, show_url=False):
except (requests.exceptions.TooManyRedirects,
requests.exceptions.ConnectionError,
requests.exceptions.InvalidURL,
+ UnicodeError,
requests.exceptions.InvalidSchema) as error:
notice("TROUBLE FOLLOWING URL: " + str(error))
return
class Session:
- def __init__(self, io, username, nickname, channel):
+ def __init__(self, io, username, nickname, channel, twtfile):
self.io = io
self.nickname = nickname
self.channel = channel
- self.uses_in_chan = []
+ self.users_in_chan = []
+ self.twtfile = twtfile
self.io.send_line("NICK " + self.nickname)
self.io.send_line("USER " + username + " 0 * : ")
self.io.send_line("JOIN " + self.channel)
elif tokens[1] == "353":
names = tokens[5:]
names[0] = names[0][1:]
- self.uses_in_chan += names
+ self.users_in_chan += names
elif tokens[1] == "JOIN":
name = name_from_join_or_part(tokens)
if name != self.nickname:
- self.uses_in_chan += [name]
+ self.users_in_chan += [name]
elif tokens[1] == "PART":
name = name_from_join_or_part(tokens)
- del(self.uses_in_chan[self.uses_in_chan.index(name)])
+ del(self.users_in_chan[self.users_in_chan.index(name)])
def parse_command_line_arguments():
parser = argparse.ArgumentParser()
parser.add_argument("-p, --port", action="store", dest="port", type=int,
default=PORT, help="port to connect to (default : "
+ str(PORT) + ")")
- parser.add_argument("-t, --timeout", action="store", dest="timeout",
+ parser.add_argument("-w, --wait", action="store", dest="timeout",
type=int, default=TIMEOUT,
help="timeout in seconds after which to attempt " +
"reconnect (default: " + str(TIMEOUT) + ")")
parser.add_argument("-n, --nickname", action="store", dest="nickname",
default=NICKNAME, help="nickname to use (default: "
+ NICKNAME + ")")
+ parser.add_argument("-t, --twtfile", action="store", dest="twtfile",
+ default=TWTFILE, help="twtfile to use (default: "
+ + TWTFILE + ")")
parser.add_argument("CHANNEL", action="store", help="channel to join")
opts, unknown = parser.parse_known_args()
return opts
while True:
try:
io = IO(opts.server, opts.port, opts.timeout)
- session = Session(io, opts.username, opts.nickname, opts.CHANNEL)
+ session = Session(io, opts.username, opts.nickname, opts.CHANNEL,
+ opts.twtfile)
session.loop()
except ExceptionForRestart:
io.socket.close()