import errno
import os

from django.conf import settings
from django.core.exceptions import ObjectDoesNotExist
from django.core.files.uploadedfile import InMemoryUploadedFile, SimpleUploadedFile
from django.utils.translation import ugettext


from documents.conf.settings import STAGING_DIRECTORY    


def get_all_files():
    try:
        return sorted([os.path.normcase(f) for f in os.listdir(STAGING_DIRECTORY) if os.path.isfile(os.path.join(STAGING_DIRECTORY, f))])
    except OSError, exc:
        raise OSError(ugettext(u'Unable get list of staging files: %s') % exc)


class StagingFile(object):
    """
    Simple class to encapsulate the files in a directory and hide the 
    specifics to the view
    """
    @classmethod
    def get_all(cls):
        staging_files = []
        for id, filename in enumerate(get_all_files()):
            staging_files.append(StagingFile(
                filepath=os.path.join(STAGING_DIRECTORY, filename),
                id=id))
        
        return staging_files

    @classmethod
    def get(cls, id):
        files = get_all_files()
        if id <= len(files):
            return StagingFile(
                filepath=os.path.join(STAGING_DIRECTORY, files[id]),
                id=id)
        raise ObjectDoesNotExist

    def __init__(self, filepath, id):
        self.filepath = filepath
        self.filename = os.path.basename(filepath)
        self._id = id
        
    def __unicode__(self):
        return self.filename

    def __repr__(self):
        return self.__unicode__()

    def __getattr__(self, name):
        if name == 'id':
            return self._id
        else:
            raise AttributeError, name
    
    def upload(self):
        try:
            return SimpleUploadedFile(name=self.filename, content=open(self.filepath).read())
        except Exception, exc:
            raise Exception(ugettext(u'Unable to upload staging file: %s') % exc)
    
    
        #return InMemoryUploadedFile(
        #    file=open(self.filepath, 'r'),
        #    field_name='',
        #    name=self.filename,
        #    content_type='unknown',
        #    size=os.path.getsize(self.filepath),
        #    charset=None,
        #)

    def delete(self):
        try:
            os.unlink(self.filepath)
        except OSError, exc:
            if exc.errno == errno.ENOENT:
                pass
            else: 
                raise OSError(ugettext(u'Unable to delete staging file: %s') % exc)