8. SQL INSERT, UPDATE, DELETE 명령어 실습

import sqlite3

conn = sqlite3.connect("sns.db")
cur  = conn.cursor()

cur.execute("""
    CREATE TABLE IF NOT EXISTS posts (
        id         INTEGER PRIMARY KEY AUTOINCREMENT,
        username   TEXT    NOT NULL,
        content    TEXT    NOT NULL,
        likes      INTEGER DEFAULT 0,
        created_at TEXT    DEFAULT (datetime('now','localtime'))
    )
""")
conn.commit()
print("posts 테이블 생성 완료")

# 게시글 한 개 삽입
cur.execute(
    "INSERT INTO posts (username, content) VALUES (?, ?)",
    ("김철수", "오늘 점심 맛있었다 🍜")
)
conn.commit()
print("게시글 작성 완료, id:", cur.lastrowid)

sample_posts = [
    ("이영희",  "오늘 운동 완료! 💪"),
    ("박지민",  "새 앨범 나왔다!! 🎵"),
    ("최수진",  "비 오는 날 커피 ☕"),
    ("홍길동",  "파이썬 공부 중 🐍"),
    ("강민준",  "주말 영화 추천 받아요 🎬"),
]

cur.executemany(
    "INSERT INTO posts (username, content) VALUES (?, ?)",
    sample_posts
)
conn.commit()
print(f"{cur.rowcount}개 게시글 삽입 완료")

import pandas as pd
df = pd.read_sql("SELECT * FROM posts", conn)
print(df)

# id가 1인 게시글의 likes를 10으로 수정
cur.execute("UPDATE posts SET likes = 10 WHERE id = 1")
conn.commit()
print("좋아요 업데이트 완료")

# 좋아요 0인 게시글을 1로 초기화
cur.execute("UPDATE posts SET likes = 1 WHERE likes = 0")
conn.commit()
print(f"{cur.rowcount}개 게시글 좋아요 초기화")

# id=2 게시글 좋아요 +5 증가
cur.execute("UPDATE posts SET likes = likes + 5 WHERE id = 2")
conn.commit()

# 확인
cur.execute("SELECT id, username, likes FROM posts WHERE id = 2")
print("수정 후:", cur.fetchone())

# id가 3인 게시글 삭제
cur.execute("DELETE FROM posts WHERE id = 3")
conn.commit()
print(f"{cur.rowcount}개 게시글 삭제 완료")

# 좋아요 2 이하인 게시글 삭제
cur.execute("DELETE FROM posts WHERE likes <= 2")
conn.commit()
print(f"{cur.rowcount}개 비인기 게시글 삭제")

# 삭제 전에 먼저 SELECT로 확인
cur.execute("SELECT id, username, content FROM posts WHERE likes <= 2")
targets = cur.fetchall()
print("삭제 예정 게시글:")
for t in targets:
    print(f"  id={t[0]} {t[1]}: {t[2]}")

# 확인 후 삭제
ans = input("삭제하시겠습니까? (y/n): ")
if ans.lower() == 'y':
    cur.execute("DELETE FROM posts WHERE likes <= 2")
    conn.commit()
    print("삭제 완료")
else:
    print("취소")

try:
    # 게시글 작성 + 작성자 포인트 증가를 하나의 트랜잭션으로 처리
    cur.execute("INSERT INTO posts (username, content) VALUES (?, ?)",
                ("테스트유저", "트랜잭션 테스트 게시글"))

    # 의도적 오류 발생 (테스트용)
    # cur.execute("UPDATE 없는테이블 SET x=1")

    conn.commit()  # 모두 성공하면 저장
    print("게시글 작성 성공")

except Exception as e:
    conn.rollback()  # 실패하면 전체 되돌리기
    print(f"오류 발생, 되돌림: {e}")