feat: jobs 테이블 및 더미 데이터 생성
- PostgreSQL jobs 테이블 스키마 생성 - 더미 프로젝트 데이터 2개 추가 (J24-001, J24-002) - 엔지니어링 업계 구조 반영 (엔드유저-클라이언트-EPC) - 가상환경 경로 이슈 해결 방법 문서화
This commit is contained in:
73
backend/scripts/setup_database.py
Normal file
73
backend/scripts/setup_database.py
Normal file
@@ -0,0 +1,73 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
전체 데이터베이스 설정 및 더미 데이터 생성
|
||||
"""
|
||||
|
||||
import sys
|
||||
import os
|
||||
from sqlalchemy import create_engine, text
|
||||
|
||||
# 프로젝트 루트를 Python path에 추가
|
||||
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
||||
|
||||
def setup_database():
|
||||
"""데이터베이스 전체 설정"""
|
||||
|
||||
try:
|
||||
from app.database import engine
|
||||
print("✅ 기존 데이터베이스 연결 사용")
|
||||
except ImportError:
|
||||
# 개발용 직접 연결
|
||||
DATABASE_URL = "sqlite:///./test.db"
|
||||
engine = create_engine(DATABASE_URL)
|
||||
print("⚠️ 개발용 SQLite 연결")
|
||||
|
||||
try:
|
||||
with engine.connect() as conn:
|
||||
print("🏗️ 1단계: jobs 테이블 생성...")
|
||||
|
||||
# jobs 테이블 생성 SQL 실행
|
||||
with open('scripts/01_create_jobs_table.sql', 'r', encoding='utf-8') as f:
|
||||
sql_commands = f.read().split(';')
|
||||
for command in sql_commands:
|
||||
command = command.strip()
|
||||
if command:
|
||||
conn.execute(text(command))
|
||||
|
||||
print("✅ jobs 테이블 생성 완료")
|
||||
|
||||
print("🔧 2단계: files 테이블 수정...")
|
||||
|
||||
# files 테이블 수정 (선택적)
|
||||
try:
|
||||
with open('scripts/02_modify_files_table.sql', 'r', encoding='utf-8') as f:
|
||||
sql_commands = f.read().split(';')
|
||||
for command in sql_commands:
|
||||
command = command.strip()
|
||||
if command and not command.startswith('--'):
|
||||
conn.execute(text(command))
|
||||
print("✅ files 테이블 수정 완료")
|
||||
except Exception as e:
|
||||
print(f"⚠️ files 테이블 수정 건너뜀: {e}")
|
||||
|
||||
# 커밋
|
||||
conn.commit()
|
||||
|
||||
print("🎯 3단계: 더미 데이터 생성...")
|
||||
# 더미 데이터 스크립트 실행
|
||||
exec(open('scripts/03_insert_dummy_data.py').read())
|
||||
|
||||
print("\n🎉 전체 설정 완료!")
|
||||
print("\n📋 다음 단계:")
|
||||
print(" 1. API 서버 실행")
|
||||
print(" 2. GET /jobs 엔드포인트 테스트")
|
||||
print(" 3. Job 선택 후 BOM 파일 업로드")
|
||||
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
print(f"❌ 데이터베이스 설정 실패: {e}")
|
||||
return False
|
||||
|
||||
if __name__ == "__main__":
|
||||
setup_database()
|
||||
Reference in New Issue
Block a user