1026 18:19

Author: Junseo1026

main.py

@@ -1,6 +1,8 @@
 # src/main.py
 import os
 from dotenv import load_dotenv
+# 환경 변수를 최대한 빨리 로드하여 GPU 설정(CUDA_VISIBLE_DEVICES)이 라우터 임포트 전에 적용되도록 함
+load_dotenv()
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 from routers.auth import router as auth_router
@@ -11,8 +13,7 @@
 import logging
 import uvicorn  
 
-# 1) 환경변수 로드
-load_dotenv()
+# 1) 환경변수 로드 (상단에서 선 로드됨)
 
 # 2) 로깅 설정
 logging.basicConfig(level=logging.INFO)

models/base.py

@@ -1,2 +1,3 @@
 from sqlalchemy.ext.declarative import declarative_base
-Base = declarative_base()
+
+Base = declarative_base()

models/file.py

@@ -1,4 +1,4 @@
-# Backend/models/file.py
+from sqlalchemy.orm import relationship
 from sqlalchemy import Column, Integer, String, ForeignKey, TIMESTAMP, text
 from .base import Base
 
@@ -8,7 +8,13 @@ class File(Base):
     id            = Column(Integer, primary_key=True, autoincrement=True)
     user_id       = Column(Integer, ForeignKey('user.u_id', ondelete='CASCADE'), nullable=False)
     folder_id     = Column(Integer, ForeignKey('folder.id', ondelete='SET NULL'), nullable=True)
-    original_name = Column(String(255), nullable=False)   # 유저가 업로드한 원본 파일 이름
-    saved_path    = Column(String(512), nullable=False)  # 서버에 저장된(실제) 경로
-    content_type  = Column(String(100), nullable=False)  # MIME 타입
+    note_id       = Column(Integer, ForeignKey('note.id', ondelete='CASCADE'), nullable=True)
+    original_name = Column(String(255), nullable=False)
+    saved_path    = Column(String(512), nullable=False)
+    content_type  = Column(String(100), nullable=False)
     created_at    = Column(TIMESTAMP, nullable=False, server_default=text('CURRENT_TIMESTAMP'))
+
+    # ✅ 관계
+    user   = relationship("User", back_populates="files")
+    folder = relationship("Folder", back_populates="files")
+    note   = relationship("Note", back_populates="files")

models/folder.py

@@ -1,4 +1,5 @@
 from sqlalchemy import Column, Integer, String, ForeignKey, TIMESTAMP, text
+from sqlalchemy.orm import relationship
 from .base import Base
 
 class Folder(Base):
@@ -12,3 +13,9 @@ class Folder(Base):
     updated_at = Column(TIMESTAMP, nullable=False,
                         server_default=text('CURRENT_TIMESTAMP'),
                         onupdate=text('CURRENT_TIMESTAMP'))
+
+    # ✅ 관계
+    user     = relationship("User", back_populates="folders")
+    parent   = relationship("Folder", remote_side=[id], backref="children")
+    notes    = relationship("Note", back_populates="folder", cascade="all, delete")
+    files    = relationship("File", back_populates="folder", cascade="all, delete")

models/note.py

@@ -1,4 +1,5 @@
 from sqlalchemy import Column, Integer, String, Text, Boolean, ForeignKey, TIMESTAMP, text
+from sqlalchemy.orm import relationship
 from .base import Base
 
 class Note(Base):
@@ -15,3 +16,8 @@ class Note(Base):
     updated_at    = Column(TIMESTAMP, nullable=False,
                            server_default=text('CURRENT_TIMESTAMP'),
                            onupdate=text('CURRENT_TIMESTAMP'))
+
+    # ✅ 관계
+    user   = relationship("User", back_populates="notes")
+    folder = relationship("Folder", back_populates="notes")
+    files  = relationship("File", back_populates="note", cascade="all, delete")

models/user.py

@@ -1,4 +1,5 @@
 from sqlalchemy import Column, Integer, String, Enum, TIMESTAMP, text
+from sqlalchemy.orm import relationship
 from .base import Base
 
 class User(Base):
@@ -9,11 +10,16 @@ class User(Base):
     email      = Column(String(150), nullable=False, unique=True)
     password   = Column(String(255), nullable=False)
     provider   = Column(
-                   Enum('local','google','kakao','naver', name='provider_enum'),
-                   nullable=False,
-                   server_default=text("'local'")
-                 )
+        Enum('local','google','kakao','naver', name='provider_enum'),
+        nullable=False,
+        server_default=text("'local'")
+    )
     created_at = Column(TIMESTAMP, nullable=False, server_default=text('CURRENT_TIMESTAMP'))
     updated_at = Column(TIMESTAMP, nullable=False,
                         server_default=text('CURRENT_TIMESTAMP'),
                         onupdate=text('CURRENT_TIMESTAMP'))
+
+    # ✅ 관계
+    folders = relationship("Folder", back_populates="user", cascade="all, delete")
+    notes   = relationship("Note", back_populates="user", cascade="all, delete")
+    files   = relationship("File", back_populates="user", cascade="all, delete")