Mikaela/Limnoria
1
0
Fork 0
mirror of https://github.com/Mikaela/Limnoria.git synced 2024-11-27 05:09:23 +01:00
Code Issues Projects Releases Wiki Activity

fixed output when querying for all unread notes

Browse Source
This commit is contained in:
Brett Kelly 2003-04-02 08:30:12 +00:00
parent b21071c382
commit 487c99c909
1 changed files with 27 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
plugins/Notes.py
View File

@ -33,6 +33,7 @@ from baseplugin import *
import time import time
import os.path import os.path
import string
import sqlite import sqlite
@ -96,6 +97,13 @@ class Notes(callbacks.Privmsg):
else: else:
raise KeyError raise KeyError
def isNote(self, noteid):
self.cursor.execute('SELECT * FROM notes WHERE id=%s', noteid)
if self.cursor.rowcount == 0:
return 0
else:
return 1
def makePrivate(self, msg): def makePrivate(self, msg):
args = list(msg.args) args = list(msg.args)
args[0] = msg.nick args[0] = msg.nick
@ -134,10 +142,13 @@ class Notes(callbacks.Privmsg):
self.db.commit() self.db.commit()
irc.reply(msg, conf.replySuccess) irc.reply(msg, conf.replySuccess)
def getnote(self, irc, msg, args): def note(self, irc, msg, args):
"retrieves a single note by unique note id" "retrieves a single note by unique note id"
# BLOODY HELL, THIS ACTUALLY WORKS!!! # BLOODY HELL, THIS ACTUALLY WORKS!!!
noteid = privmsgs.getArgs(args) noteid = privmsgs.getArgs(args)
if not self.isNote(noteid):
irc.error(msg, 'Not a valid note id')
return
sender = ircdb.users.getUserName(msg.prefix) sender = ircdb.users.getUserName(msg.prefix)
senderID = self.getUserID(sender) senderID = self.getUserID(sender)
self.cursor.execute("""SELECT note, to_id, from_id, added_at, public self.cursor.execute("""SELECT note, to_id, from_id, added_at, public
@ -159,7 +170,7 @@ class Notes(callbacks.Privmsg):
WHERE id=%s""", (1, noteid[0])) WHERE id=%s""", (1, noteid[0]))
self.db.commit() self.db.commit()
else: else:
irc.error(msg, 'Error getting note') irc.error(msg, 'You are not the recipient of note %s' % noteid)
def notes(self, irc, msg, args): def notes(self, irc, msg, args):
"takes no arguments, retrieves all unread notes for the requesting user" "takes no arguments, retrieves all unread notes for the requesting user"
@ -169,12 +180,21 @@ class Notes(callbacks.Privmsg):
WHERE to_id=%s WHERE to_id=%s
AND read=0""", senderID) AND read=0""", senderID)
notes = self.cursor.fetchall() notes = self.cursor.fetchall()
self.cursor.execute("""SELECT count(*) FROM notes
WHERE to_id=%s
AND read=0""", senderID)
count = self.cursor.fetchone()[0]
L = [] L = []
for (id, from_id) in notes: for (id, from_id) in notes:
sender = self.getUserName(from_id) sender = self.getUserName(from_id)
L.append(r'#%d from %s;;' % (id, sender)) L.append(r'#%d from %s;; ' % (id, sender))
if count > 5:
L = string.join(L[:5], '')
reply = "you have %s unread notes, 5 shown: %s" % (count, L)
else:
reply = "you have %s unread notes: %s" % (count, L)
debug.printf(L) debug.printf(L)
irc.reply(msg, "lookup was successful") irc.reply(msg, reply)
# def deletenote(self, irc, msg, args): # def deletenote(self, irc, msg, args):
# "removes single note using note id" # "removes single note using note id"
@ -203,5 +223,8 @@ class Notes(callbacks.Privmsg):
# for (id, from_id) in notes: # for (id, from_id) in notes:
# sender = self.getUserName(from_id) # sender = self.getUserName(from_id)
# L.append(r'#%d from %s;;' % (id, sender)) # L.append(r'#%d from %s;;' % (id, sender))
def die(self):
self.db.close()
Class = Notes Class = Notes