pew
/
phenny
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

create bash module

master
Paul Walko 2017-07-29 00:08:31 +00:00
parent 1eb1c361ac
commit ffa9955ce4
1 changed files with 281 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

281
modules/bash.py Normal file
View File

@ -0,0 +1,281 @@
#!/usr/bin/python3
"""
bash.py - bash.org clone. Saves quotes from quote-logger module to a db
author: paulwalko <paul@walko.org>
"""
from threading import Thread
import os, sqlite3, time
queue = []
def setup(self):
fn = self.nick + '-' + self.config.host + '.bash.db'
bash_quotes = 'bash_logs'
self.bash_quotes_path = os.path.join(os.path.expanduser('~/.phenny'), bash_quotes)
self.bash_db = os.path.join(os.path.expanduser('~/.phenny'), fn)
self.bash_conn = sqlite3.connect(self.bash_db)
if not os.path.exists(self.bash_quotes_path):
os.makedirs(self.bash_quotes_path)
c = self.bash_conn.cursor()
c.execute('''create table if not exists quotes (
id integer primary key autoincrement,
channel varchar(255),
nick varchar(255),
quote text,
time timestamp default CURRENT_TIMESTAMP
);''')
c.execute('''create table if not exists stats (
id integer primary key autoincrement,
channel varchar(255),
nick varchar(255),
lines unsigned big int not null default 0,
unique (channel, nick) on conflict replace
);''')
c.close()
"""Start thread to check for new messages"""
Thread(target = insert_into_db_caller, args=(self,)).start()
def bash(phenny, input):
"""'.bash' - queries a quote selection to add"""
usage = "Usage: .bash <nick_1> <# msgs to start at> <nick_2> <# msgs to end at> (Specifies a range of messages)"
input_all = input.group(2)
if not input_all:
phenny.say(usage)
return
input_args = input_all.split()
if len(input_args) < 4:
phenny.say(usage)
return
channel = input.sender
nick1 = input_args[0]
nick1_start_str = input_args[1]
nick2 = input_args[2]
nick2_end_str = input_args[3]
"""Check Input Type"""
if not nick1_start_str.isdigit() or not nick2_end_str.isdigit():
phenny.say("Error: 2nd & 4th argument must be integers")
phenny.say(usage)
return
nick1_start = int(nick1_start_str)
nick2_end = int(nick2_end_str)
"""Connect to db"""
fn = phenny.nick + '-' + phenny.config.host + '.bash.db'
bash_db = os.path.join(os.path.expanduser('~/.phenny'), fn)
bash_conn = sqlite3.connect(bash_db)
c = bash_conn.cursor()
"""Check input for validity"""
c.execute('''SELECT lines FROM stats WHERE (channel=? OR channel='NULL') AND nick=?''', (channel, nick1))
rows = c.fetchall()
nick1_total = 0
for row in rows:
nick1_total = nick1_total + row[0]
c.execute('''SELECT lines FROM stats WHERE (channel=? OR channel='NULL') AND nick=?''', (channel, nick2))
rows = c.fetchall()
nick2_total = 0
for row in rows:
nick2_total = nick2_total + row[0]
if nick1_total < nick1_start:
phenny.say("Error: %s has not done %s actions (EVERYTHING included EXCEPT {bot} replies)" % (nick1, nick1_start, self.phenny.nick))
return
if nick2_total < nick2_end:
phenny.say("Error: %s has not done %s actions (EVERYTHING included EXCEPT {bot} replies)" % (nick2, nick2_end, self.phenny.nick))
return
"""Fetch quote ids"""
c.execute('''SELECT id, channel, nick FROM quotes WHERE (channel=? OR channel='ALL') AND nick=?''', (channel, nick1))
rows = c.fetchall()
nick1_id = -1
for row, i in zip(reversed(rows), range(nick1_start)):
if i == nick1_start - 1:
nick1_id = row[0]
c.execute('''SELECT id, channel, nick FROM quotes WHERE (channel=? OR channel='ALL') AND nick=?''', (channel, nick2))
rows = c.fetchall()
nick2_id = -1
for row, i in zip(reversed(rows), range(nick2_end)):
if i == nick2_end - 1:
nick2_id = row[0]
if nick2_id < nick1_id:
phenny.say("Error, try again. 2nd nick number (Newer) must come after 1st nick number")
phenny.say(usage)
return
"""Fetch quotes within range of ids"""
c.execute('''SELECT quote FROM quotes WHERE (channel=? OR channel='ALL') AND id >= ? AND id <= ?''', (channel, nick1_id, nick2_id))
final_lines = []
for line in c.fetchall():
final_lines.append(line[0])
final_lines = ''.join(final_lines)
"""Write quotes to file"""
files = os.listdir(phenny.bash_quotes_path)
files_num = [0]
new_file = '1.txt'
for f in files:
f = f.strip('.txt')
if f.isdigit():
files_num.append(int(f))
new_file = str(max(files_num) + 1) + '.txt'
new_path = os.path.join(phenny.bash_quotes_path, new_file)
with open(new_path, 'w') as f:
f.write(final_lines)
f.close()
phenny.say('%s: Check bash.walko.org to see your quote' % (input.nick))
def logger(phenny, input):
"""logs EVERYTHING"""
allowed_actions = ['PRIVMSG', 'JOIN', 'QUIT', 'PART', 'NICK', 'KICK', 'MODE']
if input.event not in allowed_actions:
return
message=""
if input.event == 'PRIVMSG':
channel = input.sender
nick = input.nick
if input.group(1)[:8] == '\x01ACTION ':
quote = "* {nick} {msg}".format(nick=nick, msg=input.group(1)[8:-1])
else:
quote = "<{nick}> {msg}".format(nick=nick, msg=input.group(1))
elif input.event == 'JOIN':
channel = input.group(1)
nick = input.nick
quote = "--> {nick} has joined {channel}".format(nick=nick, channel=channel)
elif input.event == 'PART':
channel = input.sender
nick = input.nick
if input.group(1):
message = " ({message})".format(message=input.group(1))
quote = "<-- {nick} has left {channel}{message}".format(nick=nick, channel=channel, message=message)
elif input.event == 'QUIT':
channel = 'ALL'
nick = input.nick
if input.group(1):
message = " ({message})".format(message=input.group(1))
quote = "<-- {nick} has quit{message}".format(nick=nick, message=message)
elif input.event == 'KICK':
channel = input.sender
nick = input.nick
quote = "<-- {nick} has kicked {nick_kick} [{kick_msg}] from {channel}".format(nick=nick, nick_kick=input.args[1], kick_msg=input.group(1), channel=channel)
elif input.event == 'NICK':
channel = 'ALL'
nick = input.nick
quote = "-- {nick} is now known as {nick_new}".format(nick=nick, nick_new=input.group(1))
elif len(input.args) > 0 and input.args[0].startswith('#'):
channel = input.sender
nick = input.nick
args = str(input.args[1:]).replace('(', '').replace(')', '').replace("'", '')
if args.endswith(','):
args = args.replace(',', '')
else:
args = args.replace(',', ' ')
quote = "-- Mode {channel} [{args}] by {nick}".format(channel=input.sender, args=args, nick=input.nick)
else:
return
quote = "{quote}\n".format(quote=quote)
sqlite_data = {
'channel': channel,
'nick': nick,
'quote': quote
}
# format action messages
global queue
queue.append(sqlite_data)
def insert_into_db_caller(phenny):
while True:
global queue
if len(queue) > 0:
insert_into_db(phenny, queue.pop(0))
time.sleep(1)
def insert_into_db(phenny, sqlite_data):
"""inserts message to to temp db"""
if not bash.conn:
bash.conn = sqlite3.connect(phenny.bash_db)
c = bash.conn.cursor()
c.execute('''insert into quotes
(channel, nick, quote, time)
values(
:channel,
:nick,
:quote,
CURRENT_TIMESTAMP
);''', sqlite_data)
c.execute('''insert or replace into stats
(channel, nick, lines)
values(
:channel,
:nick,
coalesce((select lines from stats where channel=:channel and nick=:nick) + 1, 1)
);''', sqlite_data)
c.close()
bash.conn.commit()
c = bash.conn.cursor()
c.execute('''select id from quotes order by id desc limit 1''')
last_id = c.fetchall()[0][0] - 99
c.execute('''select channel, nick from quotes where id < ?''', (last_id,))
rows = c.fetchall()
for row in rows:
channel = row[0]
nick = row[1]
c.execute('''select lines from stats where channel=? and nick=?''', (channel, nick))
lines = 0
lines = c.fetchall()[0][0]
if lines - 1 == 0:
c.execute('''delete from stats where channel=? and nick=?''', (channel, nick))
else:
c.execute('''replace into stats
(channel, nick, lines)
values(
?,
?,
(select lines from stats where channel=? and nick=?) - 1
);''', (channel, nick, channel, nick))
c.execute('''delete from quotes where id < ?''', (last_id,))
c.close()
bash.conn.commit()
bash.conn = None
bash.rule = (['bash'], r'(.*)')
logger.event = '*'
logger.rule = r'(.*)'
if __name__ == '__main__':
print(__doc__.strip())