update pdf

src/asg/settings.py

@@ -123,4 +123,8 @@ STATIC_URL = '/static/'
 
 STATICFILES_DIRS = (
     os.path.join(BASE_DIR, "static"),
-)
+)
+
+# Media files (uploads)
+MEDIA_URL = '/media/'
+MEDIA_ROOT = os.path.join(BASE_DIR, 'media')

src/demo/views.py

@@ -444,10 +444,12 @@ def upload_refs(request):
                     continue
                 sanitized_filename = f"{sanitized_filename}{file_extension}"
 
-                file_path = os.path.join(get_path('pdf', Global_survey_id), sanitized_filename)
+                # Use a relative path under MEDIA_ROOT for file uploads
+                relative_dir = os.path.join('pdf', Global_survey_id)
+                os.makedirs(os.path.join(settings.MEDIA_ROOT, relative_dir), exist_ok=True)
+                file_path = os.path.join(relative_dir, sanitized_filename)
                 if default_storage.exists(file_path):
                     default_storage.delete(file_path)
-                
                 saved_file_name = default_storage.save(file_path, file)
                 file_size = round(float(file.size) / 1024000, 2)
 
@@ -672,31 +674,28 @@ def download_pdfs(request):
             if not pdf_links:
                 return JsonResponse({"message": "No PDFs to download."}, status=400)
 
-            base_dir = get_path('pdf', 'recommend_pdfs')
-            os.makedirs(base_dir, exist_ok=True)  # 确保文件夹存在
+            # Use MEDIA_ROOT/pdf/recommend_pdfs for saving downloaded PDFs
+            relative_dir = os.path.join('pdf', 'recommend_pdfs')
+            abs_dir = os.path.join(settings.MEDIA_ROOT, relative_dir)
+            os.makedirs(abs_dir, exist_ok=True)
 
             downloaded_files = []
             for i, pdf_url in enumerate(pdf_links):
                 try:
                     response = requests.get(pdf_url, stream=True)
                     if response.status_code == 200:
-                        # 处理文件名，确保合法
                         sanitized_title = clean_filename(pdf_titles[i]) if i < len(pdf_titles) else f"file_{i}"
-                        pdf_filename = os.path.join(base_dir, f"{sanitized_title}.pdf")
-
-                        # 下载 PDF
-                        with open(pdf_filename, "wb") as pdf_file:
+                        relative_path = os.path.join(relative_dir, f"{sanitized_title}.pdf")
+                        abs_path = os.path.join(settings.MEDIA_ROOT, relative_path)
+                        with open(abs_path, "wb") as pdf_file:
                             for chunk in response.iter_content(chunk_size=1024):
                                 pdf_file.write(chunk)
-
-                        downloaded_files.append(pdf_filename)
-                        print(f"Success: {pdf_filename}")
+                        downloaded_files.append(relative_path)
+                        print(f"Success: {abs_path}")
                     else:
                         print(f"Failed to download {pdf_url}")
-
                 except Exception as e:
                     print(f"Error downloading {pdf_url}: {e}")
-
             print("Download finished")
             return JsonResponse({"message": f"Downloaded {len(downloaded_files)} PDFs successfully!", "files": downloaded_files})