Cleanup: Deleting 03.Code/업로드용/db.py

This commit is contained in:
2026-03-19 14:04:16 +09:00
parent 76f0a4ce57
commit 987fa3f9aa

View File

@@ -1,57 +0,0 @@
# -*- coding: utf-8 -*-
import os
import psycopg2
from psycopg2.extras import RealDictCursor
DATABASE_URL = os.environ.get("DATABASE_URL")
def get_conn():
return psycopg2.connect(DATABASE_URL, cursor_factory=RealDictCursor)
def init_db():
with get_conn() as conn:
with conn.cursor() as cur:
cur.execute("""
CREATE TABLE IF NOT EXISTS files (
file_hash TEXT PRIMARY KEY,
filename TEXT,
uploaded_at TIMESTAMP DEFAULT NOW()
);
""")
cur.execute("""
CREATE TABLE IF NOT EXISTS rag_cache (
file_hash TEXT PRIMARY KEY,
chunks_json TEXT,
faiss_index BYTEA,
vectors BYTEA,
meta_json TEXT,
created_at TIMESTAMP DEFAULT NOW()
);
""")
cur.execute("""
CREATE TABLE IF NOT EXISTS sessions (
session_id TEXT PRIMARY KEY,
file_hashes TEXT[],
doc_type TEXT,
created_at TIMESTAMP DEFAULT NOW()
);
""")
cur.execute("""
CREATE TABLE IF NOT EXISTS outlines (
session_id TEXT PRIMARY KEY,
outline_text TEXT,
created_at TIMESTAMP DEFAULT NOW()
);
""")
cur.execute("""
CREATE TABLE IF NOT EXISTS briefings (
session_id TEXT PRIMARY KEY,
briefing_json TEXT,
html TEXT,
created_at TIMESTAMP DEFAULT NOW()
);
""")
conn.commit()
print("[DB]
珥湲고
", flush=True)