logbot_command_ro... 1 import select
logbot_command_ro... 2 import psycopg2
logbot_command_ro... 3 import psycopg2.extensions
logbot_command_ro... 4 from knobs import config
logbot_command_ro... 5
logbot_command_ro... 6 class postgres_interfacing():
logbot_command_ro... 7
logbot_command_ro... 8 def __init__(self):
logbot_command_ro... 9 self.connection_string = config.postgres['connection_string']
logbot_command_ro... 10 self.postgres_channel = config.postgres['channel']
logbot_command_ro... 11 self.postgres_listen_string = "LISTEN %s;" % self.postgres_channel
logbot_command_ro... 12 self.bot_command_prefix = config.bot['bot_command_prefix']
logbot_command_ro... 13 self.bot_name = config.bot['bot_name']
logbot_command_ro... 14 self.route_all_notifications = config.route_all_notifications
logbot_command_ro... 15 self.ignore_list = config.ignore_list
logbot_command_ro... 16
logbot_command_ro... 17 def send_to_outbox(self, target, payload):
logbot_command_ro... 18 #Insert a target and message into the 'outbox' table on local postgres server
logbot_command_ro... 19 conn = psycopg2.connect(self.connection_string)
logbot_command_ro... 20 cur = conn.cursor()
logbot_command_ro... 21 cur.execute("INSERT INTO outbox(target, message) VALUES(%s,%s)", (target, payload))
logbot_command_ro... 22 conn.commit()
logbot_command_ro... 23 cur.close()
logbot_command_ro... 24 conn.close()
logbot_command_ro... 25
logbot_command_ro... 26 def irc_reply(self, target, source, payload):
logbot_command_ro... 27 if source != self.bot_name:
logbot_command_ro... 28 #make sure bot doesn't talk to itself
logbot_command_ro... 29 if target == self.bot_name:
logbot_command_ro... 30 #is the target this bot
logbot_command_ro... 31 self.send_to_outbox(source, payload)
logbot_command_ro... 32 elif target[:1] == '#':
logbot_command_ro... 33 #is the target a channel
logbot_command_ro... 34 self.send_to_outbox(target, payload)
logbot_command_ro... 35
logbot_command_ro... 36 def fetch_irc_message(self, log_id):
logbot_command_ro... 37 #Fetch the irc message in the 'log' table based off of the log id
logbot_command_ro... 38 conn = psycopg2.connect(self.connection_string)
logbot_command_ro... 39 cur = conn.cursor()
logbot_command_ro... 40 cur.execute("SELECT target, source, message FROM log WHERE id = %s;", (log_id,))
logbot_command_ro... 41 row = cur.fetchone()
logbot_command_ro... 42 target, source, message = row[0], row[1], row[2]
logbot_command_ro... 43 return target, source, message
logbot_command_ro... 44 cur.close()
logbot_command_ro... 45 conn.close()