path: root/GUI_Components/Compound_Widgets/Library_View.py

# Copyright (C) IBM Corporation 2008 
import pygtk
pygtk.require('2.0')
import gtk
import pango
import cPickle
import time
from threading import Timer
from Processing.IO_Manager import *
from Processing.Article.Article import Article
from Processing.Article.Article_Data import *
from GUI_Components.Compound_Widgets.Base_Widgets.Readonly_Textbox import Readonly_Textbox

theme_xpm = [
"16 16 4 1",
"       c None s None",
".      c black",
"g      c #62F24D",
"G      c black",
"     .......... ",
"     .gggggggg. ",
"   ..........g. ",
"   .gggggggg.g. ",
" ..........g.g. ",
" .gggggggg.g.g. ",
" .gGGGgggg.g.g. ",
" .gggggggg.g.g. ",
" .gGGGGGGg.g.g. ",
" .gggggggg.g.g. ",
" .gGGGGGGg.g.g. ",
" .gggggggg.g... ",
" .gGGGGGGg.g.   ",
" .gggggggg...   ",
" .gGGGGGGg.     ",
" ..........     "]

newtheme_xpm = [
"16 16 4 1",
"       c None s None",
".      c black",
"g      c white",
"G      c black",
"     .......... ",
"     .gggggggg. ",
"   ..........g. ",
"   .gggggggg.g. ",
" ..........g.g. ",
" .gggggggg.g.g. ",
" .gggggggg.g.g. ",
" .gggggggg.g.g. ",
" .gggggggg.g.g. ",
" .gggggggg.g.g. ",
" .gggggggg.g.g. ",
" .gggggggg.g... ",
" .gggggggg.g.   ",
" .gggggggg...   ",
" .gggggggg.     ",
" ..........     "]

wikitheme_xpm = [
"16 16 4 1",
"       c None s None",
".      c black",
"g      c #FC8B65",
"G      c black",
"     .......... ",
"     .gggggggg. ",
"   ..........g. ",
"   .gggggggg.g. ",
" ..........g.g. ",
" .gggggggg.g.g. ",
" .gGGGgggg.g.g. ",
" .gggggggg.g.g. ",
" .gGGGGGGg.g.g. ",
" .gggggggg.g.g. ",
" .gGGGGGGg.g.g. ",
" .gggggggg.g... ",
" .gGGGGGGg.g.   ",
" .gggggggg...   ",
" .gGGGGGGg.     ",
" ..........     "]

mytheme_xpm = [
"16 16 4 1",
"       c None s None",
".      c black",
"g      c #E6E6E6",
"G      c #080808",
"     .......... ",
"     .gggggggg. ",
"   ..........g. ",
"   .gggggggg.g. ",
" ..........g.g. ",
" .gggggggg.g.g. ",
" .gGGGgggg.g.g. ",
" .gggggggg.g.g. ",
" .gGGGGGGg.g.g. ",
" .gggggggg.g.g. ",
" .gGGGGGGg.g.g. ",
" .gggggggg.g... ",
" .gGGGGGGg.g.   ",
" .gggggggg...   ",
" .gGGGGGGg.     ",
" ..........     "]

article_xpm = [
"16 16 4 1",
"       c None s None",
".      c black",
"g      c #62F24D",
"G      c black",
"  ............  ",
"  .gggggggggg.  ",
"  .GGGGgggggg.  ",
"  .gggggggggg.  ",
"  .gGGGGGGGGg.  ",
"  .gggggggggg.  ",
"  .gGGGGGGGGg.  ",
"  .gggggggggg.  ",
"  .gGGGGGGGGg.  ",
"  .gggggggggg.  ",
"  .gGGGGGGGGg.  ",
"  .gggggggggg.  ",
"  .gGGGGGGGGg.  ",
"  .gggggggggg.  ",
"  .gGGGGGGGGg.  ",
"  ............  ",
]

newarticle_xpm = [
"16 16 4 1",
"       c None s None",
".      c black",
"g      c white",
"G      c black",
"  ............  ",
"  .gggggggggg.  ",
"  .gggggggggg.  ",
"  .gggggggggg.  ",
"  .gggggggggg.  ",
"  .gggggggggg.  ",
"  .gggggggggg.  ",
"  .gggggggggg.  ",
"  .gggggggggg.  ",
"  .gggggggggg.  ",
"  .gggggggggg.  ",
"  .gggggggggg.  ",
"  .gggggggggg.  ",
"  .gggggggggg.  ",
"  .gggggggggg.  ",
"  ............  ",
]

wikiarticle_xpm = [
"16 16 4 1",
"       c None s None",
".      c black",
"g      c #FC8B65",
"G      c black",
"  ............  ",
"  .gggggggggg.  ",
"  .GGGGgggggg.  ",
"  .gggggggggg.  ",
"  .gGGGGGGGGg.  ",
"  .gggggggggg.  ",
"  .gGGGGGGGGg.  ",
"  .gggggggggg.  ",
"  .gGGGGGGGGg.  ",
"  .gggggggggg.  ",
"  .gGGGGGGGGg.  ",
"  .gggggggggg.  ",
"  .gGGGGGGGGg.  ",
"  .gggggggggg.  ",
"  .gGGGGGGGGg.  ",
"  ............  ",
]


class Library_View( gtk.HBox ):
    """
    Created by: Jonathan Mace
    
    Library View sets up the view to the library.
    It consists of 3 main widgets:
    1) Treeview widget
        This is populated with the themes and articles in the library.
        They can be renamed, new articles and themes can be added, and
        they can be moved about with drag and drop.
    2) Source-view widget
        This widget displays the current source article.  It is a drag
        destination for drags originating from the treeview.  Upon receiving
        a drag, it will load the appropriate article
    3) Edit-view widget
        This widget displays the current edit article.  It acts the same
        way as the source-view widget.
        
    A few minor things also exist, such as the creation of new blank articles
    and new blank themes if the user goes to editing mode without having anything
    selected.
    
    There is also a method for downloading new articles into the currently selected
    theme, which is activated by a button in the toolbar which is set up in the
    Library_Pane class.
    
    """
        
    # Create the icons
    themeicon = gtk.gdk.pixbuf_new_from_xpm_data(theme_xpm)
    articleicon = gtk.gdk.pixbuf_new_from_xpm_data(article_xpm)
    newarticleicon = gtk.gdk.pixbuf_new_from_xpm_data(newarticle_xpm)
    newthemeicon = gtk.gdk.pixbuf_new_from_xpm_data(newtheme_xpm)
    wikithemeicon = gtk.gdk.pixbuf_new_from_xpm_data(wikitheme_xpm)
    wikiarticleicon = gtk.gdk.pixbuf_new_from_xpm_data(wikiarticle_xpm)
    mythemeicon = gtk.gdk.pixbuf_new_from_xpm_data(mytheme_xpm)
    
    def __init__(self):
        self.ignore = True
            
        gtk.HBox.__init__(self)
        
        # First things first - make it look nice depending on the platform
        running_on = platform.system()
        self.colwidth = 250
        if running_on == "Linux" and "olpc" in platform.platform().lower():
                self.colwidth = 350
            
            
        #self.set_homogeneous(True)
        self.set_spacing(2)
        self.set_border_width(1)
        
        # Create the tree view, pack and show it
        self.treestore, self.treeview = self.__create_tree()
        self.treestorecontainer = gtk.ScrolledWindow()
        self.treestorecontainer.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)
        self.treestorecontainer.add(self.treeview)
        self.pack_start(self.treestorecontainer, False, False, 1)
        self.treestorecontainer.set_size_request(self.colwidth, -1)
        self.treestorecontainer.show_all()
        
        # Create a box for the textviews at the top and the status bar at the bottom
        maindisplay = gtk.VBox()
        maindisplay.set_spacing(2)
        self.pack_start(maindisplay)
        maindisplay.show()
        
        # Create a box for the two text views
        textviewbox = gtk.HBox()
        textviewbox.set_spacing(3)
        textviewbox.set_homogeneous(True)
        maindisplay.pack_start(textviewbox)
        textviewbox.show()
        
        # Create the status bar at the bottom of the page
        
        # Not bothering to show the statusbar
        statuseb = gtk.EventBox()
        statuseb.modify_bg(gtk.STATE_NORMAL, gtk.gdk.color_parse("#EEEEEE"))
        maindisplay.pack_start(statuseb, False, True, 0)
        statuseb.set_size_request(-1, 50)
        statuseb.show()
        
        self.statusbar = gtk.Label()
        statuseb.add(self.statusbar)
        self.statusbar.show()
        
        # Create the sourcepane label and textbox
        self.sourcepanebox = gtk.VBox()
        self.sourcepanebox.set_spacing(2)
        textviewbox.pack_start(self.sourcepanebox)
        self.sourcepanebox.show()
        
        labeleb = gtk.EventBox()
        labeleb.modify_bg(gtk.STATE_NORMAL, gtk.gdk.color_parse("#EEEEEE"))
        self.sourcepanebox.pack_start(labeleb, False, False, 0)
        labeleb.show()
        
        self.sourcepanelabel = gtk.Label()
        self.sourcepanelabel.set_justify(gtk.JUSTIFY_CENTER)
        self.sourcepanelabel.set_markup("<b>Wikipedia Article:</b>\n ")
        labeleb.add(self.sourcepanelabel)
        self.sourcepanelabel.show()
        
        
        sourcetextbox = gtk.ScrolledWindow()
        sourcetextbox.set_policy(gtk.POLICY_NEVER, gtk.POLICY_AUTOMATIC)
        self.sourcepanebox.pack_start(sourcetextbox)
        sourcetextbox.show()
        
        self.sourcetext = Readonly_Textbox(False)
        self.sourcetext.set_editable(False)
        self.sourcetext.set_cursor_visible(False)
        self.sourcetext.connect("button-press-event", self.suppress, None)
        sourcetextbox.add(self.sourcetext)
        self.sourcetext.modify_font(pango.FontDescription('arial 9'))
        self.sourcetext.show()
        self.sourcetext.drag_dest_set(gtk.DEST_DEFAULT_ALL, [("sourcearticle", gtk.TARGET_SAME_APP, 0)], gtk.gdk.ACTION_COPY)
        
        
        # Create the editpane label and textbox
        self.editpanebox = gtk.VBox()
        self.editpanebox.set_spacing(2)
        textviewbox.pack_start(self.editpanebox)
        self.editpanebox.show()
        
        labeleb = gtk.EventBox()
        labeleb.modify_bg(gtk.STATE_NORMAL, gtk.gdk.color_parse("#EEEEEE"))
        self.editpanebox.pack_start(labeleb, False, False, 0)
        labeleb.show()
        
        self.editpanelabel = gtk.Label()
        self.editpanelabel.set_justify(gtk.JUSTIFY_CENTER)
        self.editpanelabel.set_markup("<b>My Article:</b>\n ")
        labeleb.add(self.editpanelabel)
        self.editpanelabel.show()
        
        edittextbox = gtk.ScrolledWindow()
        edittextbox.set_policy(gtk.POLICY_NEVER, gtk.POLICY_AUTOMATIC)
        self.editpanebox.pack_start(edittextbox)
        edittextbox.show()
        
        self.edittext = Readonly_Textbox(False)
        self.edittext.set_editable(False)
        self.edittext.set_cursor_visible(False)
        self.edittext.connect("button-press-event", self.suppress, None)
        edittextbox.add(self.edittext)
        self.edittext.modify_font(pango.FontDescription('arial 9'))
        self.edittext.show()
        self.edittext.drag_dest_set(gtk.DEST_DEFAULT_ALL, [("editarticle", gtk.TARGET_SAME_APP, 0)], gtk.gdk.ACTION_COPY)
        
        
        
    def suppress(self, widget, event, data):
        widget.stop_emission("button-press-event")
        
    def __create_tree(self):
        """
        Written by: Jonathan Mace
        Last Modified: 22/08/2008
        This method sets up the tree structure used to browse topics and articles.
        It populates the treemodel, connects all the events together and then returns the treeview widget.
        """
        
        # Create and populate the treestore.
        # The first element is the text (so, topic title or article title).
        # The second element is the associated icon.
        # The third element is the type (so, theme or article)
        treestore = gtk.TreeStore(str, gtk.gdk.Pixbuf, str, str, bool)
        themes = IO_Manager().get_themes()
        if not "Wikipedia Articles" in themes:
            IO_Manager().add_theme_to_library("Wikipedia Articles")
            themes.append("Wikipedia Articles")
        if not "My Articles" in themes:
            IO_Manager().add_theme_to_library("My Articles")
            themes.append("My Articles")                
            
        for theme in themes:
            if theme == "Wikipedia Articles":
                themeiter = treestore.append(None, ["<b>%s</b>" % (theme, ), self.wikithemeicon, "wikitheme", "aaaaaaaaaaaaaaaaaaa", False])
                articles = IO_Manager().get_pages_in_theme(theme)
                for article in articles:
                    treestore.append(themeiter, [article, self.wikiarticleicon, "wikiarticle", article, False])
            elif theme == "My Articles":
                themeiter = treestore.append(None, ["<b>%s</b>" % (theme, ), self.mythemeicon, "mytheme", "aaaaaaaaaaaaaaaaaab", False])
                articles = IO_Manager().get_pages_in_theme(theme)
                for article in articles:
                    treestore.append(themeiter, [article, self.articleicon, "article", article, True])
                treestore.append(themeiter, ["<i>Create new article</i>", self.newarticleicon, "newarticle", "aaaaaaaaaaaaaaaaaaaa", True])
            else:
                themeiter = treestore.append(None, ["<b>%s</b>" % (theme, ), self.themeicon, "theme", theme, True])
                articles = IO_Manager().get_pages_in_theme(theme)
                for article in articles:
                    treestore.append(themeiter, [article, self.articleicon, "article", article, True])
                treestore.append(themeiter, ["<i>Create new article</i>", self.newarticleicon, "newarticle", "aaaaaaaaaaaaaaaaaaaa", True])
        treestore.append(None, ["<i>Create new theme</i>", self.newthemeicon, "newtheme", "zzzzzzzzzzzzzzzzz", True])
        treestore.set_sort_column_id(3, gtk.SORT_ASCENDING)

        # Create the treeview, set properties.
        treeview = gtk.TreeView(treestore)
        treeview.set_enable_search(False)
        treeview.set_headers_visible(False)
        treeview.set_reorderable(False)
        treeview.enable_model_drag_source(gtk.gdk.BUTTON1_MASK, [('text/plain', 0, 0), ("sourcearticle", gtk.TARGET_SAME_APP, 0), ("editarticle", gtk.TARGET_SAME_APP, 0)], gtk.gdk.ACTION_MOVE | gtk.gdk.ACTION_COPY)
        treeview.enable_model_drag_dest([('text/plain', 0, 0)], gtk.gdk.ACTION_MOVE)
        treeview.connect("drag-begin", self.drag_begin_event, None)
        treeview.connect("drag-drop", self.drag_drop_event, None)
        treeview.connect("drag-data-get", self.drag_data_get_event, None)
        treeview.connect("key-press-event", self.key_pressed, None)
        
        # Create the column to show the data, add to the treeview
        column = gtk.TreeViewColumn()
        treeview.append_column(column)
        
        # Create the cell renderers.  Set properties and add to the column
        iconcell = gtk.CellRendererPixbuf()
        
        cell = gtk.CellRendererText()
        cell.connect("edited", self.name_changed, None)
        
        
        column.pack_start(iconcell, False)
        column.set_attributes(iconcell, pixbuf=1)
        column.set_sort_column_id(0)
        
        column.pack_start(cell, True)
        cell.set_property("editable", True)
        column.set_attributes(cell, markup=0)
        
        path = (0, )
        iter = treestore.get_iter(path)
        path = treestore.get_path(iter)
        treeview.expand_to_path(path)
        
        return treestore, treeview
    
    
    def populate(self):
        # Remove and destroy the current contents of the panel
        self.remove(self.treestorecontainer)
        self.treestorecontainer.destroy()
            
        # Create the new tree view, pack and show it
        self.treestore, self.treeview = self.__create_tree()
        self.treestorecontainer = gtk.ScrolledWindow()
        self.treestorecontainer.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)
        self.treestorecontainer.set_size_request(self.colwidth, -1)
        self.treestorecontainer.add(self.treeview)
        self.pack_start(self.treestorecontainer, False, False, 0)
        self.reorder_child(self.treestorecontainer, 0)
        self.treestorecontainer.show_all()
        
    def add_theme_to_tree(self, theme):
        """ Adds a row to the tree with name 'theme'
        Note, that this does not save anything in the IO Manager
        """
        model = self.treestore
        model.append(None, ["<b>" + theme + "</b>", self.themeicon, "theme", theme, True])
        
    def add_article_to_tree(self, theme, title):
        """ Adds a row to the tree under theme 'theme' with title 'title'
        Note, that this does not save anything in the IO Manager
        A typical usage of this would be:
        
        IO_Manager().save_article(article)
        Library_View.add_article_to_tree(article.article_theme, article.article_title)
        Library_View.highlight_article(article.article_theme, article.article_title)
        """
        theme = "<b>%s</b>"%(theme,)
        model = self.treestore
        iter = model.get_iter_first()
        while iter != None:
            if model.get(iter, 0)[0] == theme:
                if theme == "Wikipedia Articles":
                    model.append(iter, [title, self.wikiarticleicon, "wikiarticle", title, True])
                else:
                    model.append(iter, [title, self.articleicon, "article", title, True])
                return
            else:
                iter = model.iter_next(iter)
        
    def article_exists_in_tree(self, theme, title):
        """ Returns true if the article specified exists in the tree structure.
        This is different to whether it exists as said by IO manager.
        Typical usage would be:
        
        IO_Manager().save_article(article)
        if not Library_View.article_exists_in_tree(article.article_theme, article.article_title):
            Library_View.add_article_to_tree(article.article_theme, article.article_title)
        Library_View.highlight_article(article.article_theme, article.article_title)
        
            
        
        """
        
        theme = "<b>%s</b>"%(theme,)
        iter = self.treestore.get_iter_first()
        while iter != None:
            if self.treestore.get(iter, 0)[0] == theme:
                break
            else:
                iter = self.treestore.iter_next(iter)
        if iter == None:
            return False
        iter = self.treestore.iter_children(iter)
        while iter != None:
            if self.treestore.get(iter, 0)[0] == title:
                return True
            else:
                iter = self.treestore.iter_next(iter)
        return False
        
        
    def highlight_theme(self, title):
        # Highlights the theme specified by title
        title = "<b>%s</b>"%(title,)
        iter = self.treestore.get_iter_first()
        while iter != None:
            if self.treestore.get(iter, 0)[0] == title:
                destpath = self.treestore.get_path(iter)
                self.treeview.set_cursor(destpath, None, True)
                return
            else:
                iter = self.treestore.iter_next(iter)
                
    def highlight_article(self, theme, title):
        # Expands the theme specified by theme and highlights
        # the article within this theme, specified by title.
        theme = "<b>%s</b>"%(theme,)
        iter = self.treestore.get_iter_first()
        while iter != None:
            if self.treestore.get(iter, 0)[0] == theme:
                break
            else:
                iter = self.treestore.iter_next(iter)
        if iter == None:
            return
        iter = self.treestore.iter_children(iter)
        while iter != None:
            if self.treestore.get(iter, 0)[0] == title:
                destpath = self.treestore.get_path(iter)
                self.treeview.expand_to_path(destpath)
                self.treeview.set_cursor(destpath, None, True)
                return
            else:
                iter = self.treestore.iter_next(iter)
                
    def get_current_selection_type(self):
        # Returns one of 'article', 'newarticle', 'theme', 'newtheme' or None
        # depending on which element of the tree is selected.
        model, sourceiter = self.treeview.get_selection().get_selected()
        if sourceiter == None:
            return None
        else:
            type = model.get(sourceiter, 2)
            return type
    
    def get_current_theme(self):
        # Returns the current theme if one is selected,
        # otherwise returns none.
        model, sourceiter = self.treeview.get_selection().get_selected()
        if sourceiter == None:
            return None
        title, type = model.get(sourceiter, 0, 2)
        if type == "theme" or type == "wikitheme" or type == "mytheme":
            return title[3:len(title)-4]
        elif type == "article" or type == "newarticle" or type == "wikiarticle":
            theme = model.get(model.get_iter(model.get_path(sourceiter)[0]), 0)[0]
            theme = theme[3:len(theme)-4]   
            return theme
        else:
            return None
    
    def get_current_article(self):
        # Returns the current article if one is selected,
        # otherwise returns none.
        model, sourceiter = self.treeview.get_selection().get_selected()
        if sourceiter == None:
            return None
        title, type = model.get(sourceiter, 0, 2)
        if type == "article" or type == "wikiarticle":
            return title
        else:
            return None   
        
    def set_status(self, status):
        pass
        
        
    def key_pressed(self, widget, event, data):
        # We want the delete key to do something.
        key = event.keyval
        if key == 65535 or key == 65288 or key == 65307:
            model, sourceiter = self.treeview.get_selection().get_selected()
            if sourceiter == None:
                return
            data = model.get(sourceiter, 0, 1, 2)
            title = data[0]
            type = data[2]
            
            if type == "newarticle" or type == "newtheme" or type == "mytheme" or type == "wikitheme":
                # If the current selection is a "Create new article" or "Create new theme" field then we ignore the keypress
                return
            
            if type == "article" or type == "wikiarticle":
                title = data[0]
                theme = model.get(model.get_iter(model.get_path(sourceiter)[0]), 0)[0]
                theme = theme[3:len(theme)-4]
                
                # Delete the article from the theme
                IO_Manager().remove_page(title, theme)
                
                # If the source/working articles are this one then we remove them
                source = self.sourcetext.get_article()
                working = self.edittext.get_article()
                if source != None and source.article_title == title and source.article_theme == theme:
                    self.sourcepanelabel.set_markup("<b>Wikipedia Article:</b>\n ")
                    self.sourcetext.set_article(Article())
                if working != None and working.article_title == title and working.article_theme == theme:
                    self.editpanelabel.set_markup("<b>My Article:</b>\n ")
                    self.edittext.set_article(Article())
                
                # Remove the article from the tree view
                self.treestore.remove(sourceiter)
                
                # Highlight the theme
                destpath = self.treestore.get_path(sourceiter)
                if destpath == None:
                    self.highlight_theme(theme)
                else:
                    self.treeview.set_cursor(destpath, None, True)
                
            elif type == "theme":
                theme = data[0]
                theme = theme[3:len(theme)-4]
                
                # Delete the theme from the filesystem
                IO_Manager().remove_theme(theme)
                
                # Remove the theme from the tree view
                destpath = self.treestore.get_path(sourceiter)
                self.treestore.remove(sourceiter)
                
                # If the source/working articles are in this theme then we remove them
                source = self.sourcetext.get_article()
                working = self.edittext.get_article()
                if working != None and working.article_theme == theme:
                    self.editpanelabel.set_markup("<b>My Article:</b>\n ")
                    self.edittext.set_article(Article())
                
                # Highlight the next theme
                if destpath != None:
                    self.treeview.set_cursor(destpath, None, True)
                
                self.set_status("Theme %s deleted" % (theme, ))
                
    
    def name_changed(self, renderer, path, newtext, data):
        # Disallowed chars are < > and &
        if newtext == "" or newtext.isspace() or '&' in newtext or '<' in newtext or '>' in newtext:
            return
        
        newtext = newtext
        model, sourceiter = self.treeview.get_selection().get_selected()
        data = model.get(sourceiter, 0, 1, 2)
        title = data[0]
        type = data[2]
        if type == "article":
            """ Rename the selected article """
            
            # Find the source theme and strip formatting tags
            sourcetheme = model.get(model.get_iter(model.get_path(sourceiter)[0]), 0)[0]
            sourcetheme = sourcetheme[3:len(sourcetheme)-4]
            
            # Reset the name in the source/working articles if appropriate
            source = self.sourcetext.get_article()
            edit = self.edittext.get_article()
            if source and source.article_title == title and source.article_theme == sourcetheme:
                print "setting source title"
                source.article_title = newtext
                self.sourcepanelabel.set_markup("<b>Copy from: </b> %s\n "%(newtext, ))
            if edit and edit.article_title == title and edit.article_theme == sourcetheme:
                print "setting edit title"
                edit.article_title = newtext
                self.editpanelabel.set_markup("<span size='medium'><b>Theme: </b> %s   \n<b>Article to edit: </b> %s</span>"%(sourcetheme, newtext))
            
            # If there is already an article with the new name in the theme, then we ignore the name change and highlight the existing article
            if IO_Manager().page_exists(newtext, sourcetheme):
                self.highlight_article(sourcetheme, newtext)
                return
            
            # Rename the page with IO Manager
            IO_Manager().rename_page(sourcetheme, title, newtext)
            
            # Set text on row to new text
            iter = model.get_iter(path)
            model.set(iter, 0, newtext)
            model.sort_column_changed()
            self.set_status("Article %s renamed to %s in theme %s" % (title, newtext, sourcetheme))
        elif type == "theme":
            """ Rename the selected theme """
            
            # Strip formatting tags
            title = title[3:len(title)-4]
            
            # Reset the theme in the source/working articles if appropriate
            source = self.sourcetext.get_article()
            edit = self.edittext.get_article()
            if source and source.article_theme == title:
                print "setting source theme"
                source.article_theme = newtext
            if edit and edit.article_theme == title:
                print "setting edit theme"
                edit.article_theme = newtext   
                self.editpanelabel.set_markup("<span size='medium'><b>Theme: </b> %s   \n<b>Article to edit: </b> %s</span>"%(edit.article_theme, edit.article_title))         
            
            # If there is already a theme with the new name, then we ignore the name change and highlight the existing theme
            if IO_Manager().theme_exists(newtext):
                self.highlight_theme(newtext)
                return
            
            # Rename theme in IO Manager
            IO_Manager().rename_theme(title, newtext)
            
            # Set text on row to the new text (otherwise it would revert back)
            iter = model.get_iter(path)
            model.set(iter, 0, "<b>%s</b>"%(newtext,))
            model.sort_column_changed()
            self.set_status("Theme %s renamed to %s" % (title, newtext))
        elif type == "newarticle":
            
            """ Creating a new article """
            
            if newtext.find("Create new article") != -1:
                # If the contents of the text box haven't changed, then
                # we don't want to do anything.
                return
            
            # Find the source theme and strip formatting tags
            sourceiter = model.get_iter(model.get_path(sourceiter)[0])
            sourcetheme = model.get(sourceiter, 0)[0]
            sourcetheme = sourcetheme[3:len(sourcetheme)-4]
            
            # If there is already a page with the title, we ignore the new article request and highlight the existing article
            if IO_Manager().page_exists(newtext, sourcetheme):
                self.highlight_article(sourcetheme, newtext)
                return
            
            # Create blank article
            data = Article_Data(self, article_title = newtext, article_theme = sourcetheme)
            article = Article(data)
            
            # Save article with IO manager
            IO_Manager().save_article(article)
            
            # Add new row for article
            destiter = model.append(sourceiter, [newtext, self.articleicon, "article", newtext, True])
            
            # Highlight new row
            destpath = model.get_path(destiter)
            self.treeview.set_cursor(destpath, None, True)
            self.set_status("New article '%s' created in theme %s" % (newtext, sourcetheme))
            
        elif type == "newtheme":
            """ Creating a new theme """
            
            if newtext.find("Create new theme") != -1:
                # If the contents of the text box haven't changed, then we don't want to do anything
                return
            
            try:
                # Create new topic in IO manager
                IO_Manager().add_theme_to_library(newtext)
                
                # Add new row for theme
                themeiter = model.append(None, ["<b>%s</b>" % (newtext, ), self.themeicon, "theme", newtext, True])
                model.append(themeiter, ["<i>Create new article</i>", self.newarticleicon, "newarticle", "aaaaaaaaaaaaaaaaaaaa", True])
                
            except theme_exists_error:
                # If the theme already exists, then we proceed to just highlight it.
                self.highlight_theme(newtext)
                return
            
            # Highlight new theme
            destpath = model.get_path(themeiter)
            self.treeview.expand_to_path(destpath)
            self.treeview.set_cursor(destpath, None, True)
            self.set_status("New theme '%s' created" % (newtext, ))
        
    def drag_begin_event(self, widget, context, data):
        # This just aborts the drag if the user attempts to drag a theme.
        model, sourceiter = widget.get_selection().get_selected()
        type = model.get(sourceiter, 2)[0]
        if type != "article" and type != "newarticle" and type != "wikiarticle":
            context.drag_abort(int(time.time()))
    
    def drag_drop_event(self, widget, context, x, y, time, data):
        """
        This method moves the source article to the topic that the user
        has dragged it to.
        """
        # First, retrieve the initial selection
        model, sourceiter = widget.get_selection().get_selected()
        data = model.get(sourceiter, 0, 1, 2, 3, 4)
        title = data[0]
        type = data[2]
        
        # The source theme is the label of the first node along the path to the current selection
        sourcetheme = model.get(model.get_iter(model.get_path(sourceiter)[0]), 0)[0]
            # Strip bold tags
        sourcetheme = sourcetheme[3:len(sourcetheme)-4]
        
        # Check the type, if the user didn't drag an article, we ignore
        if type == "article":
            # Work out the destination
            destrow = widget.get_dest_row_at_pos(x, y)
            destiter = model.get_iter((destrow[0][0], ))
            destpath = (destrow[0][0], 0)            
            # Strip bold tags
            desttheme = model.get(destiter, 0)[0]
            desttheme = desttheme[3:len(desttheme)-4]
            
            # If the theme already has an article with the name, we do nothing, and highlight the article
            # in the destination theme
            if IO_Manager().page_exists(title, desttheme):
                self.highlight_article(desttheme, title)
                return
            

            
            # If the destination is the "Create new theme" bit, then
            # we create a new theme and put the article in the new theme
            if desttheme == "Wikipedia Articles":
                return
            elif desttheme == "Create new theme":
                themes = IO_Manager().get_themes()
                desttheme = "New Theme 1"
                i = 1
                while desttheme in themes:
                    i = i + 1
                    desttheme = "New Theme %d" % (i,)
                IO_Manager().add_theme_to_library(desttheme)
                
                # Add new row for theme
                destiter = model.append(None, ["<b>%s</b>" % (desttheme, ), self.themeicon, "theme", desttheme, True])
                model.append(destiter, ["<i>Create new article</i>", self.newarticleicon, "newarticle", "aaaaaaaaaaaaaaaaaaaa", True])
            
            # Re-jig the rows on the visual side of things
            destiter = self.treestore.insert(destiter, 0, data)
            self.treestore.remove(sourceiter)
            self.treeview.set_model(self.treestore)
    
            
            # Re-jig the files in the filesystem
            IO_Manager().move_page(sourcetheme, desttheme, title)
            
            # Reset the topic in the source/working articles if appropriate
            source = self.sourcetext.get_article()
            edit = self.edittext.get_article()
            if edit != None and edit.article_title == title and edit.article_theme == sourcetheme:
                print "setting edit title"
                edit.article_theme = desttheme
                self.editpanelabel.set_markup("<span size='medium'><b>Theme: </b> %s   \n<b>Article to edit: </b> %s</span>"%(desttheme, title))            
            
            # Highlight the result of the drag
            destpath = model.get_path(destiter)
            self.treeview.expand_to_path(destpath)
            self.treeview.set_cursor(destpath, None, True)
            self.set_status("Article %s moved from %s to %s" % (title, sourcetheme, desttheme))
        elif type == "wikiarticle":
            # Work out the destination
            destrow = widget.get_dest_row_at_pos(x, y)
            destiter = model.get_iter((destrow[0][0], ))
            destpath = (destrow[0][0], 0)            
            # Strip bold tags
            desttheme = model.get(destiter, 0)[0]
            desttheme = desttheme[3:len(desttheme)-4]
            # Strip the "from en.wikipedia.org" or equivalent name
            i = title.find(" (from ")
            newtitle = title[0:i]
            
            i = 0
            temptitle = newtitle
            while IO_Manager().page_exists(temptitle, desttheme):
                temptitle = "%s %d" % (newtitle, i)
                
            newtitle = temptitle
            
            # If the theme already has an article with the name, we do nothing, and highlight the article
            # in the destination theme
            if IO_Manager().page_exists(newtitle, desttheme):
                self.highlight_article(desttheme, newtitle)
                return
            

            
            # If the destination is the "Create new theme" bit, then
            # we create a new theme and put the article in the new theme
            if desttheme == "Wikipedia Articles":
                return
            elif desttheme == "Create new theme":
                themes = IO_Manager().get_themes()
                desttheme = "New Theme 1"
                i = 1
                while desttheme in themes:
                    i = i + 1
                    desttheme = "New Theme %d" % (i,)
                IO_Manager().add_theme_to_library(desttheme)
                
                # Add new row for theme
                destiter = model.append(None, ["<b>%s</b>" % (desttheme, ), self.themeicon, "theme", desttheme, True])
                model.append(destiter, ["<i>Create new article</i>", self.newarticleicon, "newarticle", "aaaaaaaaaaaaaaaaaaaa", True])
            
            # Re-jig the rows on the visual side of things
            newdata = [newtitle, self.articleicon, "article", newtitle, True]
            destiter = self.treestore.insert(destiter, 0, newdata)
            self.treeview.set_model(self.treestore)
    
            # Re-jig the files in the filesystem
            IO_Manager().copy_page(title, sourcetheme, desttheme)
            IO_Manager().rename_page(desttheme, title, newtitle)
            
            # Reset the topic in the source/working articles if appropriate
            source = self.sourcetext.get_article()
            edit = self.edittext.get_article()            
            
            # Highlight the result of the drag
            destpath = model.get_path(destiter)
            self.treeview.expand_to_path(destpath)
            self.treeview.set_cursor(destpath, None, True)
            
    def drag_data_get_event(self, widget, context, selection_data, info, timestamp, data):
        model, sourceiter = self.treeview.get_selection().get_selected()
        type = model.get(sourceiter, 2)[0]
        target = selection_data.target
        if target == "sourcearticle":
            if type == "wikiarticle":
                atom = gtk.gdk.atom_intern("article")
                title = self.get_current_article()
                theme = self.get_current_theme()
                string = cPickle.dumps([title, theme])
                selection_data.set(atom, 8, string)
                self.sourcepanelabel.set_markup("<span size='medium'><b>Wikipedia Article: </b> \n%s</span> "%(title, ))
        elif target == "editarticle":        
            if type == "newarticle":
                # If the type is "newarticle" then we create a new article
                # and highlight it.
                
                # Create and save the new article
                theme = self.get_current_theme()
                i = 0
                title = "New Article"
                while IO_Manager().page_exists(title, theme):
                    i = i + 1
                    title = "New Article %d" % (i, )
                data = Article_Data(article_title = title, article_theme = theme)
                article = Article(data)
                IO_Manager().save_article(article)
                
                # Add the row to the tree model
                sourceiter = model.get_iter(model.get_path(sourceiter)[0])
                destiter = model.append(sourceiter, [title, self.articleicon, "article", title, True])
                
                # Highlight new row
                destpath = model.get_path(destiter)
                self.treeview.set_cursor(destpath, None, True)  
                
                # Set the selection data to the new article
                atom = gtk.gdk.atom_intern("article")
                string = cPickle.dumps([title, theme])
                selection_data.set(atom, 8, string)
                self.editpanelabel.set_markup("<span size='medium'><b>Theme: </b> %s   \n<b>Article to edit: </b> %s</span>"%(theme, title))
            elif type == "article":
                # If type is article, set the selection data to the theme and articlename
                atom = gtk.gdk.atom_intern("article")
                title = self.get_current_article()
                theme = self.get_current_theme()
                string = cPickle.dumps([title, theme])
                selection_data.set(atom, 8, string)
                self.editpanelabel.set_markup("<span size='medium'><b>Theme: </b> %s   \n<b>Article to edit: </b> %s</span>"%(theme, title))
            elif type == "wikiarticle":
                title = self.get_current_article()
                theme = self.get_current_theme()
                i = title.find(" (from ")
                newtitle = title[0:i]
                
                i = 0
                temptitle = newtitle
                while IO_Manager().page_exists(temptitle, "My Articles"):
                    i = i + 1
                    temptitle = "%s %d" % (newtitle, i)
                    
                newtitle = temptitle
                article = Article()
                article.article_title = newtitle
                article.article_theme = "My Articles"
                IO_Manager().save_article(article)
                
                
                sourceiter = model.get_iter((1,))
                destiter = model.append(sourceiter, [newtitle, self.articleicon, "article", newtitle, True])
                
                destpath = self.treestore.get_path(destiter)
                self.treeview.expand_to_path(destpath)
                self.treeview.set_cursor(destpath, None, True)
                
                self.sourcetext.set_article(IO_Manager().load_article(title, "Wikipedia Articles"))
                self.sourcepanelabel.set_markup("<span size='medium'><b>Wikipedia Article: </b> \n%s</span> "%(title, ))
                
                
                atom = gtk.gdk.atom_intern("article")
                string = cPickle.dumps([newtitle, "My Articles"])
                selection_data.set(atom, 8, string)
                self.editpanelabel.set_markup("<span size='medium'><b>Theme: </b> %s   \n<b>Article to edit: </b> %s</span>"%("My Articles", newtitle))
                
        else:
            # No data so nothing happens
            pass
        

    def commence_retrieval(self, widget, entry, statuslabel, wikimenu, wikidictionary):
        title = entry.get_text()
        wiki = wikidictionary[wikimenu.get_active_text()]
        theme = "Wikipedia Articles"
        if title == "" or title == None:
            return
        if IO_Manager().page_exists("%s (from %s)"%(title, wiki), theme):
            statuslabel.set_label("%s already exists" % (title, ))
            t = Timer(10, self.clear_label, [statuslabel])
            t.start()
        else:
            t = Timer(0, self.download_and_add, [title, theme, wiki, statuslabel])
            t.start()
        #self.download(title, theme)

                    
    def download_and_add(self, title, theme, wiki, statuslabel):                        
        #exceptions handled by IO_Manager
        try:
            statuslabel.set_label("Downloading %s..."%(title,))
            IO_Manager().download_wiki_article(title=title, theme=theme, wiki = wiki, statuslabel=statuslabel)
        except PageNotFoundError, e:
            statuslabel.set_label("%s could not be found."%(title, ))
            return
        except:
            statuslabel.set_label("Error downloading %s. Check your connection."%(title, ))
            return
        
        model = self.treeview.get_model()
        
        wikipath = (0, )
        wikiiter = model.get_iter(wikipath)
        destiter = model.append(wikiiter, ["%s (from %s)"%(title, wiki), self.wikiarticleicon, "wikiarticle", "%s (from %s)"%(title, wiki), False])

        destpath = model.get_path(destiter)
        self.treeview.expand_to_path(destpath)     
        
        statuslabel.set_markup("%s successfully downloaded." % (title, ))
        t = Timer(10, self.clear_label, [statuslabel])
        t.start()
    
    def clear_label(self, label):
        label.set_markup("")
                            
            
    def set_source(self, article):
        # Set the article in the source pane to 'article'
        title = article.article_title
        buf = article.getBuffer()
        text = buf.get_slice(buf.get_start_iter(), buf.get_end_iter())
        if article.article_title != None and text != "" and article.article_theme == "Wikipedia Articles":
            self.sourcetext.set_article(article)
            self.sourcepanelabel.set_markup("<b>Wikipedia Article: </b> \n%s "%(title, ))
        else:
            self.sourcetext.set_article(Article())
        self.sourcetext.set_editable(False)
        self.sourcetext.set_cursor_visible(False)
            
    def get_source(self):
        """ 
        This returns the currently selected source article.
        If the selected source article is from a different theme to the currently
        selected edit article, then the article theme is changed.  This only
        a temporary change and since it is a source article it is read only
        so it will never be saved.
        New themes / articles are created and loaded in the event that
        nothing is currently selected.
        """
        currentarticle = self.sourcetext.get_article()
        if not currentarticle or not currentarticle.article_title or currentarticle.article_title == "":
            article = Article()
        else:
            article = currentarticle
        article.article_theme == "Wikipedia Articles"
        return article
        
        
    def set_working(self, article):
        # Set the working article to 'article'
        theme = article.article_theme
        title = article.article_title
        buf = article.getBuffer()
        text = buf.get_slice(buf.get_start_iter(), buf.get_end_iter())
        if text:
            print "article has text"
            if not theme:
                theme = "My Articles"
            if not title:
                i = 0
                title = "New Article"
                while IO_Manager().page_exists(title, theme):
                    i = i + 1
                    title = "New Article %s" % (i, )
        else:
            if title and theme and "New Article" in title and not IO_Manager().page_exists(title, theme):
                title == None
        if theme and title and theme != "Wikipedia Articles":
            if not self.article_exists_in_tree(theme, title):
                self.add_article_to_tree(theme, title)
            IO_Manager().save_article(article)
            self.highlight_article(theme, title)
            self.edittext.set_article(article)
            self.editpanelabel.set_markup("<span size='medium'><b>Theme: </b> %s   \n<b>Article to edit: </b> %s</span>"%(theme, title))
        
        self.sourcetext.set_editable(False)
        self.sourcetext.set_cursor_visible(False)
        
    def get_working(self):
        # Return the current working article, or create a new article in the current theme
        # if there is no current working article.
        currentarticle = self.edittext.get_article()
        if currentarticle != None:
            article = currentarticle
        else:
            article = Article()
        title = article.article_title
        theme = article.article_theme
        
        if not theme or theme == "":
            theme = "My Articles"
            article.article_theme = theme
        
        if not title or title == "":
            sourcearticle = self.sourcetext.get_article()
            if sourcearticle and sourcearticle.article_title:
                sourcetitle = sourcearticle.article_title
            else:
                sourcetitle = ""
            j = sourcetitle.find(" (from ")
            if j != -1:
                title = sourcetitle[0:j]
            else:
                title = "New Article"                
            articles = IO_Manager().get_pages_in_theme(theme)
            i = 0
            while title in articles:
                i = i + 1
                title = "New Article %d" % (i, )
            print title
            article.article_title = title
                    
        return article