» Python Collapsible Title Index » My geek blog

Python Collapsible Title Index

Here is my Trac modification to have collapsible title indexing. I used this script as the base. The following is my new execute method in TitleIndex.py:

def execute(hdf, args, env):
    db = env.get_db_cnx()
    cursor = db.cursor()

prefix = None
    if args:
        prefix = args.replace('\'', '\'\'')

sql = 'SELECT DISTINCT name FROM wiki '
    if prefix:
        sql += 'WHERE name LIKE \'%s%%\' ' % prefix
    sql += 'ORDER BY name'
    cursor.execute(sql)

buf = StringIO()

# find common prefixes
    lastPrefix = ''
    prefixArray = []
    while 1:
        row = cursor.fetchone()
        if row == None:
            break
        if lastPrefix == '':
            lastPrefix = row[0]
        else:
            newPrefix = findcommonstart([lastPrefix, row[0]])
            if len(newPrefix) >= 4:
               lastPrefix = newPrefix
            else:
               prefixArray += [lastPrefix]
               lastPrefix = row[0]

cursor.execute(sql)  # this should be copied instead of doing another DB query
    # print out indecies
    buf.write('<ul id="root">')
    currentPrefix = 0
    row = cursor.fetchone()
    if row == None:
       return buf.getValue()

while currentPrefix < len(prefixArray):
       buf.write('<li><div>%s</div>' % prefixArray[currentPrefix])
       buf.write('</ul><ul>')
       prefix = findcommonstart([prefixArray[currentPrefix],row[0]])
       while len(prefix) >= 4:
          buf.write('<li><a href="%s">' % env.href.wiki(row[0]))
          buf.write(row[0])
          buf.write('</li>\n')
          row = cursor.fetchone()
          if row == None:
             break
          prefix = findcommonstart([prefixArray[currentPrefix],row[0]])
       buf.write('</ul>')
       currentPrefix = currentPrefix + 1
    buf.write('')

return buf.getvalue()

This entry was posted on Tuesday, September 27th, 2005 at 5:00 pm and is filed under Trac. You can follow any responses to this entry through the RSS 2.0 feed. You can leave a response, or trackback from your own site.

You must be logged in to post a comment.

Python Collapsible Title Index

Leave a Reply