import sqlite3

from nntp.tiny.message import Message

class Database():
    __slots__ = 'db',

    def __init__(self, path: str):
        self.db = sqlite3.connect(path)

    def message_add(self, newsgroup_id: int, message: Message):
        sql = """
            insert into
                newsgroup_message
            (newsgroup_id, message_id, created_on, sender, subject, content) values (?, ?, ?, ?, ?, ?)
        """

        self.db.execute(sql, (
            newsgroup_id,
            message.id(),
            message.date().isoformat(),
            message.sender(),
            message.subject(),
            message.content
        ))

    def commit(self):
        self.db.commit()