linter fixes

ACMAS · Dec 1, 2023 · 285538e · 285538e
1 parent 5cf3066
commit 285538e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/ACMAS/app/ACMAS_Web/ocr.py b/ACMAS/app/ACMAS_Web/ocr.py
@@ -1,9 +1,10 @@
-import fitz, os  # PyMuPDF
-from .models import UploadedFile, Course
+import fitz  # PyMuPDF
+import os
 from datetime import date
 from django.conf import settings
-
 from django.core.files.storage import FileSystemStorage
+from .models import Course, UploadedFile
+
 
 """
 OCR File Creation Pipeline:
@@ -12,7 +13,7 @@
     - Store text file on filesystem and database
 """
 class OCR:
-    def extract_text_from_pdf(fType, course, fileName, fileUrl):
+    def extract_text_from_pdf(self, fType, course, fileName, fileUrl):
         # Adding file to filesystem
         fs = FileSystemStorage()
         removeExt = os.path.splitext(fileName)[0]

diff --git a/ACMAS/app/ACMAS_Web/upload.py b/ACMAS/app/ACMAS_Web/upload.py
@@ -60,7 +60,8 @@ def uploadFile(self, uni, course, fType, file):
         file_url = fs.url(savedFile)  # Retrieve the file path
         print(f'FILE "{savedFile}" uploaded to "{file_url}"\n')
 
-        OCR.extract_text_from_pdf(fType, course, fileName, file_url)
+        ocrObject = OCR
+        ocrObject.extract_text_from_pdf(fType, course, fileName, file_url)
 
         # Adding file to database
         db_file = UploadedFile(