forked from ckh08045/AutoPercenty
25 lines
861 B
Python
25 lines
861 B
Python
import sqlite3
|
|
from config import PRODUCTS_DB_PATH
|
|
import logging
|
|
|
|
# 로거 인스턴스 가져오기
|
|
logger = logging.getLogger('default_logger')
|
|
|
|
conn = sqlite3.connect(PRODUCTS_DB_PATH)
|
|
cursor = conn.cursor()
|
|
|
|
def setup_database():
|
|
"""데이터베이스 테이블을 설정합니다."""
|
|
cursor.execute('CREATE TABLE IF NOT EXISTS processed_products (product_id TEXT PRIMARY KEY)')
|
|
conn.commit()
|
|
|
|
def is_product_processed(product_id):
|
|
"""상품이 이미 처리되었는지 확인합니다."""
|
|
cursor.execute('SELECT product_id FROM processed_products WHERE product_id = ?', (product_id,))
|
|
return cursor.fetchone() is not None
|
|
|
|
def mark_product_processed(product_id):
|
|
"""상품을 처리된 것으로 표시합니다."""
|
|
cursor.execute('INSERT INTO processed_products (product_id) VALUES (?)', (product_id,))
|
|
conn.commit()
|