app/__init__.py

from flask import Flask
from flask_login import LoginManager
from app.database import db, User
import os
import sqlite3
from pathlib import Path

login_manager = LoginManager()
login_manager.login_view = 'main.login'
login_manager.login_message = '로그인이 필요합니다.'
login_manager.login_message_category = 'info'

@login_manager.user_loader
def load_user(user_id):
    return User.query.get(int(user_id))

def create_app():
    # 템플릿 폴더 경로 명시적 설정
    template_folder = os.path.join(os.path.dirname(os.path.abspath(__file__)), '..', 'templates')
    app = Flask(__name__, template_folder=template_folder)
    app.config['SECRET_KEY'] = os.getenv('SECRET_KEY', 'dev-secret-key-change-in-production')
    app.config['SQLALCHEMY_DATABASE_URI'] = os.getenv('DATABASE_URL', 'sqlite:///finance_analysis.db')
    app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
    app.config['MAX_CONTENT_LENGTH'] = 100 * 1024 * 1024  # 100MB 파일 크기 제한
    
    db.init_app(app)
    login_manager.init_app(app)
    
    from app.routes import main_bp
    app.register_blueprint(main_bp)
    
    with app.app_context():
        db.create_all()
        # 데이터베이스 마이그레이션 (nickname 컬럼 추가)
        migrate_database(app)
        # 초기 관리자 계정 생성
        create_admin_user()
    
    return app

def migrate_database(app):
    """데이터베이스 마이그레이션"""
    try:
        # 데이터베이스 URI에서 경로 추출
        db_uri = app.config['SQLALCHEMY_DATABASE_URI']
        if db_uri.startswith('sqlite:///'):
            db_path = db_uri.replace('sqlite:///', '')
            # 상대 경로인 경우 instance 폴더 기준으로 처리
            if not os.path.isabs(db_path):
                db_path = os.path.join(app.instance_path, db_path)
            
            if not os.path.exists(db_path):
                print(f"[마이그레이션] 데이터베이스 파일이 없습니다: {db_path}")
                return
            
            conn = sqlite3.connect(db_path)
            cursor = conn.cursor()
            
            # user 테이블에 nickname 컬럼이 있는지 확인
            cursor.execute("PRAGMA table_info(user)")
            user_columns = [column[1] for column in cursor.fetchall()]
            
            if 'nickname' not in user_columns:
                print("[마이그레이션] user 테이블에 nickname 컬럼 추가 중...")
                cursor.execute("ALTER TABLE user ADD COLUMN nickname VARCHAR(80)")
                conn.commit()
                print("[마이그레이션] user.nickname 컬럼 추가 완료")
            
            # uploaded_file 테이블이 존재하는지 확인
            cursor.execute("SELECT name FROM sqlite_master WHERE type='table' AND name='uploaded_file'")
            if cursor.fetchone():
                # uploaded_file 테이블에 uploaded_by 컬럼이 있는지 확인
                cursor.execute("PRAGMA table_info(uploaded_file)")
                uploaded_file_columns = [column[1] for column in cursor.fetchall()]
                
                if 'uploaded_by' not in uploaded_file_columns:
                    print("[마이그레이션] uploaded_file 테이블에 uploaded_by 컬럼 추가 중...")
                    cursor.execute("ALTER TABLE uploaded_file ADD COLUMN uploaded_by INTEGER")
                    conn.commit()
                    print("[마이그레이션] uploaded_file.uploaded_by 컬럼 추가 완료")
                
                # uploaded_file 테이블에 parent_file_id 컬럼이 있는지 확인
                if 'parent_file_id' not in uploaded_file_columns:
                    print("[마이그레이션] uploaded_file 테이블에 parent_file_id 컬럼 추가 중...")
                    cursor.execute("ALTER TABLE uploaded_file ADD COLUMN parent_file_id INTEGER")
                    conn.commit()
                    print("[마이그레이션] uploaded_file.parent_file_id 컬럼 추가 완료")
            
            conn.close()
            print("[마이그레이션] 데이터베이스 마이그레이션 완료")
    except Exception as e:
        print(f"[마이그레이션] 오류 발생: {e}")
        import traceback
        traceback.print_exc()

def create_admin_user():
    """초기 관리자 계정 생성"""
    admin_username = 'soymedia'
    admin_password = 's0ymedi@1@34'
    
    admin = User.query.filter_by(username=admin_username).first()
    if not admin:
        admin = User(username=admin_username, is_admin=True, is_active=True)
        admin.set_password(admin_password)
        db.session.add(admin)
        db.session.commit()
        print(f'관리자 계정이 생성되었습니다: {admin_username}')