TSR Messages Application v0.3

# coding:UTF-8

# Author: Chrosson (0.1 onwards)
# 0.1 - Initial Release
# 0.2 - Added compatibility for building on OSX with py2app
# 0.3 - Better unknown charset handling with UnicodeDammit

# This program is free software. It comes without any warranty, to
# the extent permitted by applicable law. You can redistribute it
# and/or modify it under the terms of the Do What The Fuck You Want
# To Public License, Version 2, as published by Sam Hocevar. See
# http://sam.zoy.org/wtfpl/COPYING for more details.

# Known/intentional 'bugs'
# - If the 60secs restriction is lifted from messages, two in the same minute from
#   the same user will be considered the same. Not currently a problem.
# - Throws an error if anything other than 'Inbox' and 'Sent Items' folders are
#   detected to prevent undefined behaviour
# - OSX cannot deal with the updateUI event, so double clicking columns
#   simply doesn't resize the window

# Reading messages
import csv
import os
from BeautifulSoup import UnicodeDammit

# Playing with messages
import itertools
import sys

#############
# FUNCTIONS #
#############

# Retrieve messages from all csv files in current dir
def getmessages():
    if sys.platform == 'darwin':
        messagefiles = ["../../../" + name for name in os.listdir("../../..")
                        if name[-4:] == ".csv"]
    else:
        messagefiles = [name for name in os.listdir(".")
                        if name[-4:] == ".csv"]
    messages = set()
    for name in messagefiles:
        # Try finding the encoding, falling back if unsuccessful
        encoding = UnicodeDammit(open(name).read(),
                                 smartQuotesTo=None).originalEncoding
        if not encoding:
            encoding = 'latin-1'
        with open(name) as csvfile:
            reader = csv.reader(csvfile)
            # Skip header of csv file
            reader.next()
            for [Date,Folder,Title,From,To,Message] in reader:
                messages.add(tuple(
                    map(lambda x: x.decode(encoding),
                        [Date,Title,From,To,Folder,Message])
                    ))
    return sorted(messages, reverse=False)

# Retrieve a message, format and output it for displaying
def msg_display((msg_date, msg_co)):
    [Date,Title,From,To,Folder,Message] = itertools.ifilter(
        lambda x: x[0]==msg_date and (x[2]==msg_co or x[3]==msg_co),
        messages).next()
    return ("Date: " + Date + "        " + "Folder: " + Folder + "\n"
            "Title: " + Title + "\n" +
            "From: " + From + "\n" +
            "To: " + To + "\n" +
            "\n======================\n\n" + Message)


################
# WX GUI SHITE #
################

import wx

class tsr_msg_panel(wx.Panel):

    class msg_list_ctrl(wx.ListCtrl):
        def __init__(self, *args, **kwargs):
            kwargs['style'] = wx.LC_REPORT | wx.LC_SINGLE_SEL
            wx.ListCtrl.__init__(self, *args, **kwargs)

            self.InsertColumn(0, 'Date')
            self.InsertColumn(1, 'Title', width=100)
            self.InsertColumn(2, 'Correspondant')
            self.InsertColumn(3, 'Folder')
            # Populate list boxes
            for [Date,Title,From,To,Folder,Message] in messages:
                pos = self.InsertStringItem(0, Date)
                self.SetStringItem(pos, 1, Title)
                if Folder == 'Inbox':
                    self.SetStringItem(pos, 2, From)
                    self.SetStringItem(pos, 3, 'Inbox')
                elif Folder == 'Sent Items':
                    self.SetStringItem(pos, 2, To)
                    self.SetStringItem(pos, 3, 'Sent')
                else:
                    raise Exception
            # Sort out columns etc
            # Autosize doesn't reliably work so add leeway
            self.SetColumnWidth(0, wx.LIST_AUTOSIZE)
            self.SetColumnWidth(0, self.GetColumnWidth(0)+3)
            self.SetColumnWidth(2, wx.LIST_AUTOSIZE)
            self.SetColumnWidth(2, self.GetColumnWidth(2)+3)
            self.SetColumnWidth(3, wx.LIST_AUTOSIZE_USEHEADER)
            self.SetColumnWidth(3, self.GetColumnWidth(3)+3)
            self.column_widths = [self.GetColumnWidth(x)
                                  for x in range(4)]
            size = (
                reduce(lambda x,y: x+self.GetColumnWidth(y),
                       range(4), 0) + 25, 0)
            self.SetMinSize(size)
            self.SetSize(size)

    def setup_controls(self):
        # Give the buttons their own layout for ease
        layout = wx.GridBagSizer()
        # Swap between text and html preview
        self.swap_preview_button = wx.Button(self, label="Full preview")
        self.swap_preview_button.Disable()
        # Help button
        self.help_button = wx.Button(self, label="?", style=wx.BU_EXACTFIT)
        # Creep people out
        self.creepy_label_1 = wx.StaticText(self,label=u"Chrosson is watching you ")
        self.creepy_label_2 = wx.StaticText(self,label=u"ಠ_ಠ")
        self.creepy_label_2.SetFont(wx.Font(10,wx.FONTFAMILY_DEFAULT,
                                          wx.FONTSTYLE_NORMAL,wx.FONTWEIGHT_NORMAL,
                                          False,"Tunga",wx.FONTENCODING_SYSTEM))
        # Lay it all out
        layout.Add(self.help_button,(0,0),(1,1),
                    flag=wx.ALIGN_CENTER_VERTICAL)
        layout.Add(self.creepy_label_1,(0,1),(1,1),
                    flag=wx.LEFT|wx.ALIGN_CENTER_VERTICAL,border=10)
        layout.Add(self.creepy_label_2,(0,2),(1,1),
                    flag=wx.RIGHT|wx.ALIGN_CENTER_VERTICAL,border=10)
        layout.Add(self.swap_preview_button,(0,3),(1,1),
                    flag=wx.EXPAND|wx.ALL|wx.ALIGN_CENTER_VERTICAL)
        layout.AddGrowableCol(3)
        return layout

    class msg_view_panel(wx.Panel):
        # Place to show the messages
        def __init__(self, *args, **kwargs):
            wx.Panel.__init__(self, *args, **kwargs)
            # Place to display messages
            self.tb_msg = wx.TextCtrl(self,-1,
                                      style=wx.TE_MULTILINE|wx.TE_READONLY)
            # Arrange it, make it resize nicely
            view_sizer = wx.GridBagSizer()
            view_sizer.Add(self.tb_msg,(0,0),(1,1),flag=wx.EXPAND)
            view_sizer.AddGrowableCol(0)
            view_sizer.AddGrowableRow(0)
            self.SetSizerAndFit(view_sizer)

    class help_frame(wx.Frame):
        # Give some help to the (l)users
        def __init__(self, *args, **kwargs):
            wx.Frame.__init__(self, *args, **kwargs)

            self.help_text = wx.TextCtrl(self,-1,
                                      style=wx.TE_MULTILINE|wx.TE_READONLY)
            self.help_text.ChangeValue(u"""Welcome to the TSR messages app.
To use, open your PMs on TSR and look near the bottom - you can download selected (or all) messages as a CSV file.
e.g. This URL will get all your PMs in a CSV - http://bit.ly/NFFmi0
Do one or the other and put the file in the same folder as this program.
Close and reopen the program - your messages should now be listed.

You can download as many CSV files as you like, rename them, whatever. As long as they're in the same directory as this program and have .csv at the end of their name, they will be detected and displayed (and duplicates ignored).
This program is only known to work for the particular format of CSV TSR uses at the moment (July 2012), although I'm optimistic about the future.

Foreign characters should work.
If you don't want flickering don't resize columns.

This program will NOT save any images or external files - it's strictly for the text of messages.

That's all.

- Chrosson

And no, I'm not really watching you.""")
            # Lay out the help box
            help_sizer = wx.GridBagSizer()
            help_sizer.Add(self.help_text,(0,0),(1,1),flag=wx.EXPAND)
            help_sizer.AddGrowableCol(0)
            help_sizer.AddGrowableRow(0)
            self.SetSizerAndFit(help_sizer)
            self.SetSize((410,440))
            self.Centre()
            # Show it in init so can forget about it
            self.Show(True)

    def __init__(self, parent, *args, **kwargs):
        wx.Panel.__init__(self, parent, *args, **kwargs)
        # Initialisation of various parts of the GUI
        self.msg_list = tsr_msg_panel.msg_list_ctrl(self)
        self.msg_view = tsr_msg_panel.msg_view_panel(self)
        controls_sizer = self.setup_controls()
        # Set up events for selecting an item,
        # clicking the help button and resizing a column
        # and a final horrible one for double clicking a column
        self.Bind(wx.EVT_LIST_ITEM_SELECTED,
                  self.msg_select,
                  self.msg_list)
        self.Bind(wx.EVT_BUTTON,
                  lambda x: self.help_frame(None, title="Help"),
                  self.help_button)
        self.Bind(wx.EVT_LIST_COL_END_DRAG,
                  self.event_column_resize,
                  self.msg_list)
        if sys.platform == 'win32':
            self.Bind(wx.EVT_UPDATE_UI,
                      self.event_update_UI,
                      self.msg_list)
        # Arrange everything nicely with sane resizing behaviour
        self.top_sizer = wx.GridBagSizer()
        self.top_sizer.Add(self.msg_list, (0,0), (2,1), wx.EXPAND)
        self.top_sizer.Add(controls_sizer, (0,1), (1,1), wx.EXPAND)
        self.top_sizer.Add(self.msg_view, (1,1), (1,1), wx.EXPAND)
        self.top_sizer.AddGrowableCol(1)
        self.top_sizer.AddGrowableRow(1)
        self.top_sizer.SetItemMinSize(self.msg_view, (250,0))
        self.SetSizerAndFit(self.top_sizer)
        # Let the window know it has some resize limits
        self.top_sizer.SetSizeHints(parent)
        parent.SetSize((800,600))
        parent.Centre()

    def event_update_UI(self, event):
        # Horrible event handler just to watch for a double click resize
        new_column_widths = [self.msg_list.GetColumnWidth(x)
                           for x in range(4)]
        if new_column_widths != self.msg_list.column_widths:
            self.msg_list.column_widths = new_column_widths
            self.resize_list_ctrl((sum(new_column_widths) + 23, 0))

    def event_column_resize(self, event):
        # Watch out for when a column has been resized
        if sys.platform == 'win32':
            size = (
                reduce(lambda x,y: x+self.msg_list.GetColumnWidth(y),
                       range(4), 0) + 23 -
                self.msg_list.GetColumnWidth(event.GetColumn()) +
                event.GetItem().GetWidth(), 0)
        else:
            size = (
                reduce(lambda x,y: x+self.msg_list.GetColumnWidth(y),
                       range(4), 0) + 23, 0)
        self.resize_list_ctrl(size)

    def resize_list_ctrl(self, size):
        # Resize the list control
        self.msg_list.SetMinSize(size)
        self.msg_list.SetSize(size)
        # Recompute the layout of the window
        self.TopLevelParent.Layout()
        # Remind the window that it has some new resize limits
        windowsize = self.TopLevelParent.GetSize()
        self.top_sizer.SetSizeHints(self.TopLevelParent)
        self.TopLevelParent.SetSize(windowsize)

    def msg_select(self, event):
        # A message has been selected, let's display it
        row = event.GetIndex()
        msg_key = (self.msg_list.GetItem(event.GetIndex(), 0).Text,
                   self.msg_list.GetItem(event.GetIndex(), 2).Text)

        self.msg_view.tb_msg.SetValue(msg_display(msg_key))

# Dirty global variable
messages = getmessages()

app = wx.App(False)
frame = wx.Frame(None, title="TSR Msg App")
tsr_msg_panel(frame)
frame.Show()
sys.exit(app.MainLoop())