picsorter/database.py

import sqlite3
from datetime import datetime


class Database:
    def __init__(self):
        self.db_name = 'images.db'
        self.__create_tables()

    def __create_tables(self):
        conn = sqlite3.connect(self.db_name)
        c = conn.cursor()
        c.executescript("""
            CREATE TABLE IF NOT EXISTS images (
                id INTEGER PRIMARY KEY NOT NULL,
                provider TEXT NOT NULL,
                tags TEXT NOT NULL,
                created_at TIMESTAMP,
                UNIQUE(id, provider) ON CONFLICT REPLACE
            )
            """)
        conn.commit()
        conn.close()

    def is_exists(self, provider, _id) -> bool:
        conn = sqlite3.connect(self.db_name)
        c = conn.cursor()
        c.execute("SELECT EXISTS(SELECT 1 FROM images WHERE id=? AND provider=?)", (_id, provider))
        result = c.fetchone()[0]
        conn.close()
        return bool(result)
    
    def add(self, _id, provider, tags):
        conn = sqlite3.connect(self.db_name)
        c = conn.cursor()
        sql = 'INSERT INTO images(id, provider, tags, created_at) VALUES (?,?,?,?)'
        c.execute(sql, (_id, provider, tags, datetime.now()))
        conn.commit()
        conn.close()