import datetime

from .. import db

class BaseStorage:
    def __init__(self, app):
        self.app = app

    def add_file(self, file_hash, filename, size, user=None, ip=None):
        '''Adds the file to the database.

        Call from store_file after the file is successfully stored.'''
        with db.session_scope() as sess:
            f = db.File(file_hash, filename, size, datetime.datetime.utcnow(), user.id if user else None, ip)
            sess.add(f)
            sess.commit()
            sess.refresh(f)
        return f

    def store_file(self, uploaded_file, file_hash, filename, user, ip):
        '''Store uploaded_file.'''
        raise NotImplementedError()

    def get_file(self, f):
        '''Return a file object for the specified file.

        Subclasses can also return a flask.Response instance if required.'''
        raise NotImplementedError()

    def delete_file(self, f):
        '''Delete the specified file.'''
        raise NotImplementedError()

    def temp_file(self, f):
        '''Context manager which returns a temporary file for reading.

        This is used internally for eg. thumbnails.'''
        raise NotImplementedError()

    def get_thumbnail(self, f):
        '''Return a file object for the specified file's thumbnail.

        Subclasses can also return a flask.Response instance if required.'''
        raise NotImplementedError()

    def store_thumbnail(self, f, stream):
        '''Store thumbnail for the specified file.'''
        raise NotImplementedError()