Merge pull request #39 from sustech-cs304/backend

Optimize db linking and foreign key in backend

Author: RoderickQiu

backend/app/db.py

@@ -2,13 +2,16 @@
 import os
 from sqlalchemy import create_engine
 from sqlalchemy.orm import sessionmaker, declarative_base
+from dotenv import load_dotenv
+
+load_dotenv()
 
 db_port = 5432
 K8S_DB_HOST = "postgres"
 if os.environ.get("ENVNAME") == "k3s":
     DATABASE_URL = f"postgresql://{os.environ.get('POSTGRES_USER')}:{os.environ.get('POSTGRES_PASSWORD')}@{K8S_DB_HOST}:{db_port}/{os.environ.get('POSTGRES_DB')}"
 else:
-    DATABASE_URL = f"postgresql://serendipity:chenmingzhi2004@localhost:{db_port}/cs304"
+    DATABASE_URL = f"postgresql://{os.getenv('POSTGRES_USER')}:{os.getenv('POSTGRES_PASSWORD')}@localhost:{db_port}/{os.getenv('POSTGRES_DB')}"
 
 engine = create_engine(DATABASE_URL, echo=True)
 SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)

backend/app/main.py

@@ -31,6 +31,8 @@
 
 @app.on_event("startup")
 def startup():
+    Base.metadata.reflect(bind=engine)
+    Base.metadata.drop_all(bind=engine)
     Base.metadata.create_all(bind=engine)
 
 

backend/app/models/assignment.py

@@ -1,5 +1,5 @@
 # app/models/assignment.py
-from sqlalchemy import Column, Integer, String, Boolean
+from sqlalchemy import Column, Integer, String, Boolean, ForeignKey
 from sqlalchemy.dialects.postgresql import ARRAY
 from app.db import Base
 
@@ -9,8 +9,8 @@ class Assignment(Base):
 
     assignment_id = Column(String, primary_key=True, index=True)
     name = Column(String, index=True)
-    course_id = Column(String, index=True)
-    teacher_id = Column(String, index=True)
+    course_id = Column(String, ForeignKey("courses.course_id"), index=True)
+    teacher_id = Column(String, ForeignKey("users.user_id"), index=True)
     deadline = Column(String, index=True)
     isOver = Column(Boolean, index=True)
     materials = Column(

backend/app/models/bookmarklist.py

@@ -1,5 +1,5 @@
 # app/models/bookmarklist.py
-from sqlalchemy import Column, Integer, String, Boolean
+from sqlalchemy import Column, Integer, String, Boolean, ForeignKey
 from sqlalchemy.dialects.postgresql import ARRAY
 from app.db import Base
 
@@ -8,7 +8,7 @@ class BookmarkList(Base):
     __tablename__ = "bookmark_lists"
 
     list_id = Column(String, primary_key=True, index=True)
-    material_id = Column(String, index=True)
-    user_id = Column(String, index=True)
+    material_id = Column(String, ForeignKey("materials.material_id"), index=True)
+    user_id = Column(String, ForeignKey("users.user_id"), index=True)
     page = Column(Integer, index=True)
     bookmark_list = Column(ARRAY(Integer), index=True)

backend/app/models/code_snippet.py

@@ -1,13 +1,13 @@
 # app/models/code_snippet.py
-from sqlalchemy import Column, Integer, String, Boolean
+from sqlalchemy import Column, Integer, String, Boolean, ForeignKey
 from app.db import Base
 
 
 class CodeSnippet(Base):
     __tablename__ = "code_snippets"
 
     snippet_id = Column(String, primary_key=True, index=True)
-    material_id = Column(String, nullable=False)
+    material_id = Column(String, ForeignKey("materials.material_id"), index=True, nullable=False)
     lang = Column(String, nullable=False)
     page = Column(Integer, nullable=False)
     content = Column(String, nullable=False)

backend/app/models/comment.py

@@ -1,5 +1,5 @@
 # app/models/comment.py
-from sqlalchemy import Column, Integer, String, Boolean
+from sqlalchemy import Column, Integer, String, Boolean, ForeignKey
 from sqlalchemy.dialects.postgresql import ARRAY
 from app.db import Base
 
@@ -9,7 +9,7 @@ class Comment(Base):
 
     comment_id = Column(String, primary_key=True, index=True)
     content = Column(String, nullable=False)
-    user_id = Column(String, nullable=False)
-    material_id = Column(String, nullable=False)
+    user_id =Column(String, ForeignKey("users.user_id"), index=True, nullable=False)
+    material_id = Column(String, ForeignKey("materials.material_id"), index=True, nullable=False)
     page = Column(Integer, nullable=False)
     ancestor_id = Column(String, nullable=True)

backend/app/models/material.py

@@ -1,5 +1,5 @@
 # app/models/material.py
-from sqlalchemy import Column, Integer, String, Boolean
+from sqlalchemy import Column, Integer, String, Boolean, ForeignKey
 from sqlalchemy.dialects.postgresql import ARRAY
 from app.db import Base
 
@@ -9,7 +9,7 @@ class Material(Base):
 
     material_id = Column(String, primary_key=True, index=True)
     material_name = Column(String, nullable=False)
-    section_id = Column(String, nullable=False)
+    section_id = Column(String, ForeignKey("sections.section_id"), index=True, nullable=False)
     data = Column(String, nullable=False)
     comments = Column(
         ARRAY(String), nullable=True

backend/app/models/note.py

@@ -1,5 +1,5 @@
 # app/models/note.py
-from sqlalchemy import Column, Integer, String, Boolean
+from sqlalchemy import Column, Integer, String, Boolean, ForeignKey
 from sqlalchemy.dialects.postgresql import ARRAY
 from app.db import Base
 
@@ -8,8 +8,8 @@ class Note(Base):
     __tablename__ = "notes"
 
     note_id = Column(String, primary_key=True, index=True)
-    user_id = Column(String, nullable=False)
-    material_id = Column(String, nullable=False)
+    user_id = Column(String, ForeignKey("users.user_id"), nullable=False)
+    material_id = Column(String, ForeignKey("materials.material_id"), nullable=False)
     is_snippet = Column(Boolean, default=False)
     content = Column(String, nullable=False)
     code_snippet = Column(

backend/app/models/schedule.py

@@ -1,5 +1,5 @@
 # app/models/schedule.py
-from sqlalchemy import Column, Integer, String, Boolean
+from sqlalchemy import Column, Integer, String, Boolean, ForeignKey
 from sqlalchemy.dialects.postgresql import ARRAY
 from app.db import Base
 

backend/app/models/section.py

@@ -1,5 +1,5 @@
 # app/models/section.py
-from sqlalchemy import Column, Integer, String, Boolean
+from sqlalchemy import Column, Integer, String, Boolean, ForeignKey
 from sqlalchemy.dialects.postgresql import ARRAY
 from app.db import Base
 
@@ -8,7 +8,7 @@ class Section(Base):
     __tablename__ = "sections"
 
     section_id = Column(String, primary_key=True, index=True)
-    course_id = Column(String, index=True)
+    course_id = Column(String, ForeignKey("courses.course_id"), nullable=False, index=True)
     name = Column(String, index=True)
     materials = Column(
         ARRAY(String), index=True