Merge pull request #1288 from bcgov/dev-NK-13.0-merge

move zipper changes to dev for 13.0 release
bcgov · Jan 30, 2025 · 562b08a · 562b08a
2 parents 8d60b60 + 790145b
commit 562b08a
Showing 1 changed file with 14 additions and 11 deletions.
diff --git a/computingservices/ZippingServices/services/zipperservice.py b/computingservices/ZippingServices/services/zipperservice.py
@@ -15,6 +15,7 @@
 import json
 import traceback
 import PyPDF2
+from os import path
 
 def processmessage(message):
     try:
@@ -113,16 +114,18 @@ def __zipfilesandupload(_message, s3credentials):
 
                     _docbytes = __getdocumentbytearray(fileobj, s3credentials)
                     _formattedbytes = None
-
-                    try:                           
-                        _formattedbytes = __removesensitivecontent(_docbytes)
-                        if _formattedbytes is not None:
-                            print("_formattedbytes length is {0}".format(len(_formattedbytes)))
-                        else:
-                           print("_formattedbytes is none")                                
-                    except Exception:
-                        print("error happened while removing sensitive content of {0} ".format(filename))
-                        print(traceback.format_exc())
+                    _filename, extension = path.splitext(fileobj["s3uripath"])
+
+                    if extension.lower() == '.pdf':
+                        try:                           
+                            _formattedbytes = __removesensitivecontent(_docbytes)
+                            if _formattedbytes is not None:
+                                print("_formattedbytes length is {0}".format(len(_formattedbytes)))
+                            else:
+                               print("_formattedbytes is none")                                
+                        except Exception:
+                            print("error happened while removing sensitive content of {0} ".format(filename))
+                            print(traceback.format_exc())
                     #added a space to try out code merge on git. 18-Sept-2024    
                     zip.writestr(
                         filename, _docbytes if _formattedbytes is None else _formattedbytes
@@ -172,4 +175,4 @@ def __getzipfilepath(foldername, filename):
         foldername.capitalize() + "/" + filename + ".zip"
         if foldername is not None
         else filename + ".zip"
-    )
+    )