# coding=utf-8
import sqlite3
from settings import Settings
import os


class History(object):
    def __init__(self, name):
        self._name = name
        os.chdir(Settings.get_default_path())
        db = sqlite3.connect(name + '.hstr')
        cursor = db.cursor()
        cursor.execute('CREATE TABLE IF NOT EXISTS friends('
                       '    tox_id TEXT PRIMARY KEY'
                       ')')
        db.close()

    def add_friend_to_db(self, tox_id):
        try:
            os.chdir(Settings.get_default_path())
            db = sqlite3.connect(self._name + '.hstr')
            cursor = db.cursor()
            cursor.execute('INSERT INTO friends VALUES (?);', (tox_id, ))
            cursor.execute('CREATE TABLE id' + tox_id + '('
                           '    id INTEGER PRIMARY KEY,'
                           '    message TEXT,'
                           '    owner INTEGER,'
                           '    unix_time INTEGER'
                           '    message_type INTEGER'
                           ')')
            db.commit()
        except Exception as e:
            db.rollback()
            raise e
        finally:
            db.close()

    def delete_friend_from_db(self, tox_id):
        try:
            os.chdir(Settings.get_default_path())
            db = sqlite3.connect(self._name + '.hstr')
            cursor = db.cursor()
            cursor.execute('DELETE FROM friends WHERE tox_id=?;', (tox_id, ))
            cursor.execute('DROP TABLE id' + tox_id + ';')
            db.commit()
        except Exception as e:
            db.rollback()
            raise e
        finally:
            db.close()

    def save_messages_to_db(self, tox_id, messages_iter):
        try:
            os.chdir(Settings.get_default_path())
            db = sqlite3.connect(self._name + '.hstr')
            cursor = db.cursor()
            cursor.executemany('INSERT INTO id' + tox_id + '(message, owner, unix_time) '
                               'VALUES (?, ?, ?);', messages_iter)
            db.commit()
        except Exception as e:
            db.rollback()
            raise e
        finally:
            db.close()

    def messages_getter(self, tox_id):
        return MessageGetter(self._name, tox_id)


class MessageGetter(object):
    def __init__(self, name, tox_id):
        os.chdir(Settings.get_default_path())
        self._db = sqlite3.connect(name + '.hstr')
        self._cursor = self._db.cursor()
        self._cursor.execute('SELECT message, owner, unix_time FROM id' + tox_id +
                             ' ORDER BY unix_time DESC;')

    def get_one(self):
        return self._cursor.fetchone()

    def get_all(self):
        return self._cursor.fetchall()

    def get(self, count):
        return self._cursor.fetchmany(count)

    def __del__(self):
        self._db.close()


if __name__ == '__main__':
    h = History('test')
    getter = h.messages_getter('42')
    print getter.get_all()