titles now in txt file

SELab-2 · May 3, 2024 · c20d320 · c20d320
1 parent d65cb7d
commit c20d320
Show file tree

Hide file tree

Showing 3 changed files with 219 additions and 213 deletions.
diff --git a/backend/seeder/elements.py b/backend/seeder/elements.py
diff --git a/backend/seeder/seeder.py b/backend/seeder/seeder.py
@@ -11,7 +11,6 @@
 from sqlalchemy.exc import SQLAlchemyError
 from sqlalchemy_utils import register_composites
 
-from elements import course_titles
 from project.models.course import Course
 from project.models.course_relation import CourseAdmin, CourseStudent
 from project.models.project import Project
@@ -25,9 +24,19 @@
 
 fake = Faker()
 
+# Get the directory of the current script
+script_dir = os.path.dirname(os.path.realpath(__file__))
+
+# Construct the path to titles.txt relative to the script directory
+titles_path = os.path.join(script_dir, 'titles.txt')
+
+with open(titles_path, 'r', encoding='utf-8') as file:
+    # Read the lines of the file and strip newline characters
+    titles = [line.strip() for line in file]
+
 course_title_provider = DynamicProvider(  # Custom course titles.
     provider_name="course_titles",
-    elements=course_titles,
+    elements=titles,
 )
 fake.add_provider(course_title_provider)