first commit
This commit is contained in:
김민성
264
setup.py
Normal file
264
setup.py
Normal file
@@ -0,0 +1,264 @@
|
||||
"""
|
||||
설치 및 설정 가이드 스크립트
|
||||
프로젝트 설치와 초기 설정을 도와주는 스크립트입니다.
|
||||
"""
|
||||
|
||||
import os
|
||||
import sys
|
||||
import subprocess
|
||||
import shutil
|
||||
from pathlib import Path
|
||||
|
||||
def print_header(title):
|
||||
"""헤더 출력"""
|
||||
print("\n" + "=" * 60)
|
||||
print(f" {title}")
|
||||
print("=" * 60)
|
||||
|
||||
def print_step(step_num, title):
|
||||
"""단계 출력"""
|
||||
print(f"\n📋 단계 {step_num}: {title}")
|
||||
print("-" * 40)
|
||||
|
||||
def check_python_version():
|
||||
"""Python 버전 확인"""
|
||||
print_step(1, "Python 버전 확인")
|
||||
|
||||
version = sys.version_info
|
||||
print(f"현재 Python 버전: {version.major}.{version.minor}.{version.micro}")
|
||||
|
||||
if version.major < 3 or (version.major == 3 and version.minor < 9):
|
||||
print("❌ Python 3.9 이상이 필요합니다.")
|
||||
print("https://www.python.org/downloads/ 에서 최신 버전을 다운로드하세요.")
|
||||
return False
|
||||
else:
|
||||
print("✅ Python 버전이 요구사항을 만족합니다.")
|
||||
return True
|
||||
|
||||
def check_pip():
|
||||
"""pip 확인 및 업그레이드"""
|
||||
print_step(2, "pip 확인 및 업그레이드")
|
||||
|
||||
try:
|
||||
# pip 버전 확인
|
||||
result = subprocess.run([sys.executable, "-m", "pip", "--version"],
|
||||
capture_output=True, text=True)
|
||||
print(f"pip 버전: {result.stdout.strip()}")
|
||||
|
||||
# pip 업그레이드
|
||||
print("pip를 최신 버전으로 업그레이드 중...")
|
||||
subprocess.run([sys.executable, "-m", "pip", "install", "--upgrade", "pip"],
|
||||
check=True)
|
||||
print("✅ pip 업그레이드 완료")
|
||||
return True
|
||||
|
||||
except subprocess.CalledProcessError as e:
|
||||
print(f"❌ pip 업그레이드 실패: {e}")
|
||||
return False
|
||||
|
||||
def create_virtual_environment():
|
||||
"""가상 환경 생성 (선택 사항)"""
|
||||
print_step(3, "가상 환경 생성 (권장)")
|
||||
|
||||
venv_path = Path("venv")
|
||||
|
||||
if venv_path.exists():
|
||||
print("⚠️ 가상 환경이 이미 존재합니다.")
|
||||
response = input("기존 가상 환경을 삭제하고 새로 만드시겠습니까? (y/N): ")
|
||||
if response.lower() == 'y':
|
||||
shutil.rmtree(venv_path)
|
||||
else:
|
||||
print("기존 가상 환경을 사용합니다.")
|
||||
return True
|
||||
|
||||
try:
|
||||
print("가상 환경 생성 중...")
|
||||
subprocess.run([sys.executable, "-m", "venv", "venv"], check=True)
|
||||
print("✅ 가상 환경 생성 완료")
|
||||
|
||||
# 활성화 안내
|
||||
if os.name == 'nt': # Windows
|
||||
print("\n가상 환경 활성화 방법:")
|
||||
print(" venv\\Scripts\\activate")
|
||||
else: # macOS/Linux
|
||||
print("\n가상 환경 활성화 방법:")
|
||||
print(" source venv/bin/activate")
|
||||
|
||||
return True
|
||||
|
||||
except subprocess.CalledProcessError as e:
|
||||
print(f"❌ 가상 환경 생성 실패: {e}")
|
||||
return False
|
||||
|
||||
def install_dependencies():
|
||||
"""의존성 설치"""
|
||||
print_step(4, "의존성 설치")
|
||||
|
||||
requirements_file = Path("requirements.txt")
|
||||
|
||||
if not requirements_file.exists():
|
||||
print("❌ requirements.txt 파일을 찾을 수 없습니다.")
|
||||
return False
|
||||
|
||||
try:
|
||||
print("의존성 패키지 설치 중...")
|
||||
print("(이 과정은 몇 분 정도 걸릴 수 있습니다)")
|
||||
|
||||
subprocess.run([sys.executable, "-m", "pip", "install", "-r", "requirements.txt"],
|
||||
check=True)
|
||||
print("✅ 모든 의존성 설치 완료")
|
||||
return True
|
||||
|
||||
except subprocess.CalledProcessError as e:
|
||||
print(f"❌ 의존성 설치 실패: {e}")
|
||||
return False
|
||||
|
||||
def setup_environment_file():
|
||||
"""환경 변수 파일 설정"""
|
||||
print_step(5, "환경 변수 설정")
|
||||
|
||||
env_example = Path(".env.example")
|
||||
env_file = Path(".env")
|
||||
|
||||
if not env_example.exists():
|
||||
print("❌ .env.example 파일을 찾을 수 없습니다.")
|
||||
return False
|
||||
|
||||
if env_file.exists():
|
||||
print("⚠️ .env 파일이 이미 존재합니다.")
|
||||
response = input("기존 .env 파일을 덮어쓰시겠습니까? (y/N): ")
|
||||
if response.lower() != 'y':
|
||||
print("기존 .env 파일을 사용합니다.")
|
||||
return True
|
||||
|
||||
try:
|
||||
# .env.example을 .env로 복사
|
||||
shutil.copy2(env_example, env_file)
|
||||
print("✅ .env 파일 생성 완료")
|
||||
|
||||
print("\n⚠️ 중요: Gemini API 키를 설정해야 합니다!")
|
||||
print("1. Google AI Studio에서 API 키를 발급받으세요:")
|
||||
print(" https://makersuite.google.com/app/apikey")
|
||||
print("2. .env 파일을 열어 GEMINI_API_KEY를 설정하세요:")
|
||||
print(" GEMINI_API_KEY=your_actual_api_key_here")
|
||||
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
print(f"❌ .env 파일 설정 실패: {e}")
|
||||
return False
|
||||
|
||||
def create_directories():
|
||||
"""필요한 디렉토리 생성"""
|
||||
print_step(6, "필요한 디렉토리 생성")
|
||||
|
||||
directories = ["uploads", "results", "assets", "docs"]
|
||||
|
||||
for dir_name in directories:
|
||||
dir_path = Path(dir_name)
|
||||
if not dir_path.exists():
|
||||
dir_path.mkdir(parents=True, exist_ok=True)
|
||||
print(f"✅ {dir_name}/ 디렉토리 생성")
|
||||
else:
|
||||
print(f"✅ {dir_name}/ 디렉토리 이미 존재")
|
||||
|
||||
return True
|
||||
|
||||
def run_tests():
|
||||
"""테스트 실행"""
|
||||
print_step(7, "설치 확인 테스트")
|
||||
|
||||
test_script = Path("test_project.py")
|
||||
|
||||
if not test_script.exists():
|
||||
print("❌ 테스트 스크립트를 찾을 수 없습니다.")
|
||||
return False
|
||||
|
||||
try:
|
||||
print("설치 상태를 확인하는 테스트를 실행합니다...")
|
||||
result = subprocess.run([sys.executable, "test_project.py"],
|
||||
capture_output=True, text=True)
|
||||
|
||||
print(result.stdout)
|
||||
if result.stderr:
|
||||
print("오류:", result.stderr)
|
||||
|
||||
return result.returncode == 0
|
||||
|
||||
except Exception as e:
|
||||
print(f"❌ 테스트 실행 실패: {e}")
|
||||
return False
|
||||
|
||||
def print_final_instructions():
|
||||
"""최종 안내사항 출력"""
|
||||
print_header("설치 완료!")
|
||||
|
||||
print("🎉 PDF 도면 분석기 설치가 완료되었습니다!")
|
||||
print("\n다음 단계:")
|
||||
print("1. .env 파일을 열어 Gemini API 키를 설정하세요")
|
||||
print("2. 가상 환경을 활성화하세요 (사용하는 경우)")
|
||||
print("3. 애플리케이션을 실행하세요:")
|
||||
print(" python main.py")
|
||||
|
||||
print("\n📚 추가 정보:")
|
||||
print("- README.md: 사용법 및 문제 해결")
|
||||
print("- project_plan.md: 프로젝트 계획 및 진행 상황")
|
||||
print("- test_project.py: 테스트 스크립트")
|
||||
|
||||
print("\n🔧 문제가 발생하면:")
|
||||
print("1. python test_project.py 를 실행하여 문제를 확인하세요")
|
||||
print("2. .env 파일의 API 키 설정을 확인하세요")
|
||||
print("3. 가상 환경이 활성화되어 있는지 확인하세요")
|
||||
|
||||
def main():
|
||||
"""메인 설치 함수"""
|
||||
print_header("PDF 도면 분석기 설치 스크립트")
|
||||
print("이 스크립트는 PDF 도면 분석기 설치를 도와드립니다.")
|
||||
print("각 단계를 순차적으로 진행합니다.")
|
||||
|
||||
response = input("\n설치를 시작하시겠습니까? (Y/n): ")
|
||||
if response.lower() == 'n':
|
||||
print("설치가 취소되었습니다.")
|
||||
return
|
||||
|
||||
# 설치 단계들
|
||||
steps = [
|
||||
("Python 버전 확인", check_python_version),
|
||||
("pip 확인 및 업그레이드", check_pip),
|
||||
("가상 환경 생성", create_virtual_environment),
|
||||
("의존성 설치", install_dependencies),
|
||||
("환경 변수 설정", setup_environment_file),
|
||||
("디렉토리 생성", create_directories),
|
||||
("설치 확인 테스트", run_tests)
|
||||
]
|
||||
|
||||
completed = 0
|
||||
|
||||
for step_name, step_func in steps:
|
||||
try:
|
||||
if step_func():
|
||||
completed += 1
|
||||
else:
|
||||
print(f"\n❌ {step_name} 단계에서 문제가 발생했습니다.")
|
||||
response = input("계속 진행하시겠습니까? (y/N): ")
|
||||
if response.lower() != 'y':
|
||||
break
|
||||
except KeyboardInterrupt:
|
||||
print("\n\n설치가 중단되었습니다.")
|
||||
return
|
||||
except Exception as e:
|
||||
print(f"\n❌ {step_name} 단계에서 예상치 못한 오류가 발생했습니다: {e}")
|
||||
response = input("계속 진행하시겠습니까? (y/N): ")
|
||||
if response.lower() != 'y':
|
||||
break
|
||||
|
||||
print(f"\n설치 진행률: {completed}/{len(steps)} 단계 완료")
|
||||
|
||||
if completed == len(steps):
|
||||
print_final_instructions()
|
||||
else:
|
||||
print("\n⚠️ 설치가 완전히 완료되지 않았습니다.")
|
||||
print("문제를 해결한 후 다시 실행하거나 수동으로 남은 단계를 진행하세요.")
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user