get_id.py

import sys
import os
from dotenv import load_dotenv

# Add project root to path
sys.path.append(os.path.dirname(os.path.abspath(__file__)))

# Load env
env_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'backend', '.env')
load_dotenv(env_path)

from backend.app.database import SessionLocal
from backend.app.models import Prompt

def get_prompt_id(title: str):
    db = SessionLocal()
    try:
        prompt = db.query(Prompt).filter(Prompt.title == title).first()
        if prompt:
            print(f"Title: {title}, ID: {prompt.id}")
        else:
            print(f"Prompt '{title}' not found.")
    except Exception as e:
        print(f"Error: {e}")
    finally:
        db.close()

if __name__ == "__main__":
    get_prompt_id("互联网黑话专家")