diff --git a/Dockers/puppeteer-api/main.py b/Dockers/puppeteer-api/main.py index ec91677..498d30c 100644 --- a/Dockers/puppeteer-api/main.py +++ b/Dockers/puppeteer-api/main.py @@ -22,7 +22,7 @@ CUSTOM_USER_AGENT = 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.3 # Initialize SQLite database def init_db(): - conn = sqlite3.connect('cache.db') + conn = sqlite3.connect('/db/cache.db') cursor = conn.cursor() cursor.execute(''' CREATE TABLE IF NOT EXISTS cache ( @@ -38,7 +38,7 @@ def init_db(): # Get cached data if it exists and is not older than 36 hours def get_cached_data(url, route): - conn = sqlite3.connect('cache.db') + conn = sqlite3.connect('/db/cache.db') cursor = conn.cursor() cache_expiry = int(time.time()) - (36 * 60 * 60) # 36 hours in seconds cursor.execute( @@ -55,7 +55,7 @@ def get_cached_data(url, route): # Save data to cache def save_to_cache(url, route, data): - conn = sqlite3.connect('cache.db') + conn = sqlite3.connect('/db/cache.db') cursor = conn.cursor() timestamp = int(time.time()) @@ -338,7 +338,7 @@ async def clear_cache(x_api_key: Optional[str] = Header(None)): if not x_api_key or x_api_key != API_KEY: raise HTTPException(status_code=401, detail="Invalid API key") - conn = sqlite3.connect('cache.db') + conn = sqlite3.connect('/db/cache.db') cursor = conn.cursor() cursor.execute("DELETE FROM cache") conn.commit() @@ -353,7 +353,7 @@ async def cache_stats(x_api_key: Optional[str] = Header(None)): if not x_api_key or x_api_key != API_KEY: raise HTTPException(status_code=401, detail="Invalid API key") - conn = sqlite3.connect('cache.db') + conn = sqlite3.connect('/db/cache.db') cursor = conn.cursor() # Get total entries