~vorlon/ubuntu/saucy/gourmet/trunk

from gourmet.plugin import ImporterPlugin

from gourmet.importers.importer import Tester

from gourmet.threadManager import get_thread_manager

from gourmet.importers.interactive_importer import InteractiveImporter

from gourmet import check_encodings

import os.path

 

MAX_PLAINTEXT_LENGTH = 100000

 

class PlainTextImporter (InteractiveImporter):

 

    name = 'Plain Text Importer'

 

    def __init__ (self, filename):

        self.filename = filename

        InteractiveImporter.__init__(self)

        

    def do_run (self):

        if os.path.getsize(self.filename) > MAX_PLAINTEXT_LENGTH*16:

            del data

            ifi.close()

            import gourmet.gtk_extras.dialog_extras as de

            de.show_message(title=_('Big File'),

                            label=_('File %s is too big to import'%self.filename),

                            sublabel=_('Your file exceeds the maximum length of %s characters. You probably didn\'t mean to import it anyway. If you really do want to import this file, use a text editor to split it into smaller files and try importing again.')%MAX_PLAINTEXT_LENGTH,

                            message_type=gtk.MESSAGE_ERROR)

            return

        ifi = file(self.filename,'r')

        data = '\n'.join(check_encodings.get_file(ifi))

        ifi.close()

        self.set_text(data)

        return InteractiveImporter.do_run(self)

 

class PlainTextImporterPlugin (ImporterPlugin):

 

    name = _('Plain Text file')

    patterns = ['*.txt','[^.]*','*']

    mimetypes = ['text/plain']

 

    def test_file (self, filename):

        '''Given a filename, test whether the file is of this type.'''

        return -1 # we are a fallback option

 

    def get_importer (self, filename):

        return PlainTextImporter(filename=filename)