Stage

.dockerignore

@@ -0,0 +1,5 @@
+node_modules
+.serverless
+.github
+layers
+scripts

.isort.cfg

@@ -1,2 +1,2 @@
 [settings]
-known_third_party =boto3,fitz,jinja2,pdfkit,pydantic,pynamodb
+known_third_party =boto3,fitz,jinja2,pydantic,pynamodb,weasyprint

Dockerfile

@@ -0,0 +1,17 @@
+FROM public.ecr.aws/lambda/python:3.8
+
+
+ARG FUNCTION_DIR="/var/task"
+RUN mkdir -p ${FUNCTION_DIR}
+
+RUN pip install pipenv
+RUN yum install -y pango
+
+COPY Pipfile ${LAMBDA_TASK_ROOT}
+COPY Pipfile.lock ${LAMBDA_TASK_ROOT}
+
+RUN pipenv install --system --deploy --ignore-pipfile
+
+COPY . ${LAMBDA_TASK_ROOT}
+
+CMD [ "handler.generate_certificate_handler" ]

Pipfile

@@ -9,10 +9,10 @@ boto3 = "==1.22.7"
 pydantic = {version = "<2", extras = ["email"]}
 python-dotenv = "==1.0.0"
 jinja2 = "==3.1.2"
-pdfkit = "1.0.0"
 pillow = "==10.1.0"
 pymupdf = "==1.23.6"
-pynamodb = "==5.5.0"
+pynamodb = "==5.5.1"
+weasyprint = "==52.5"
 
 [dev-packages]
 pre-commit = "==3.3.3"

package.json

@@ -6,6 +6,7 @@
   "license": "MIT",
   "devDependencies": {
     "serverless-iam-roles-per-function": "^3.2.0",
+    "serverless-plugin-log-retention": "^2.0.0",
     "serverless-python-requirements": "^6.0.0"
   }
 }

resources/generate_certificate.yml

@@ -1,9 +1,9 @@
-generate:
+certMaker:
   handler: handler.generate_certificate_handler
-  timeout: 900
   layers:
-    - Ref: WkhtmltopdfLambdaLayer
+    - Ref: WeazyprintLambdaLayer
     - Ref: PythonRequirementsLambdaLayer
+  timeout: 900
   events:
     - sqs:
         arn:

s3/data_store.py

@@ -9,6 +9,7 @@
 
 from s3.exceptions import PdfServiceInternalError
 from s3.s3_constants import PresignedURLMethod
+from utils.logger import logger
 
 
 # pylint: disable=broad-except
@@ -18,7 +19,6 @@ class S3DataStore:
     def __init__(self, bucket_name=os.environ['S3_BUCKET']):
         self.__s3_client = boto3_client('s3')
         self.__bucket_name = bucket_name
-        self.__logger = logging.getLogger()
 
     def upload_file(self, file_name: str, object_name: str = None, verbose: bool = True) -> bool:
         result = True
@@ -30,10 +30,10 @@ def upload_file(self, file_name: str, object_name: str = None, verbose: bool = T
         try:
             self.__s3_client.upload_file(file_name, self.__bucket_name, object_name)
             if verbose:
-                self.__logger.info('Stored file in S3: %s/%s', self.__bucket_name, object_name)
+                logger.info('Stored file in S3: %s/%s', self.__bucket_name, object_name)
         except Exception as e:
             message = f'Failed to upload file ({file_name}) to S3, Reason: {type(e).__name__} - {str(e)}'
-            self.__logger.error(message)
+            logger.error(message)
             raise PdfServiceInternalError(status_code=HTTPStatus.INTERNAL_SERVER_ERROR, message=message) from e
 
         return result
@@ -44,10 +44,10 @@ def download_file(self, object_name: str, file_name: str, verbose: bool = True)
         try:
             self.__s3_client.download_file(self.__bucket_name, object_name, file_name)
             if verbose:
-                self.__logger.info('Downloaded file in S3: %s/%s', self.__bucket_name, object_name)
+                logger.info('Downloaded file in S3: %s/%s', self.__bucket_name, object_name)
         except Exception as e:
             message = f'Failed to download file ({object_name}) from S3, Reason: {type(e).__name__} - {str(e)}'
-            self.__logger.error(message)
+            logger.error(message)
             raise PdfServiceInternalError(status_code=HTTPStatus.INTERNAL_SERVER_ERROR, message=message) from e
 
         return result
@@ -69,10 +69,10 @@ def zip_files(self, bucket_keys: List[str], output_zip_key: str, verbose: bool =
             self.__s3_client.put_object(Body=zip_buffer.getvalue(), Bucket=self.__bucket_name, Key=output_zip_key)
 
             if verbose:
-                self.__logger.info('Stored file in S3: %s', f'{self.__bucket_name}/{output_zip_key}')
+                logger.info('Stored file in S3: %s', f'{self.__bucket_name}/{output_zip_key}')
         except Exception as e:
             message = f'Failed to zip files, Reason: {type(e).__name__} - {str(e)}'
-            self.__logger.error(message)
+            logger.error(message)
             raise PdfServiceInternalError(status_code=HTTPStatus.INTERNAL_SERVER_ERROR, message=message) from e
 
     def generate_presigned_url(
@@ -90,14 +90,14 @@ def generate_presigned_url(
             url = self.__s3_client.generate_presigned_url(
                 ClientMethod=method.value, Params=params, ExpiresIn=expires_in
             )
-            self.__logger.info('Pre-signed URL generated for file: %s', object_name)
+            logger.info('Pre-signed URL generated for file: %s', object_name)
         except Exception as e:
             message = (
                 f'Failed to generate pre-signed URl for file: ({object_name}), '
                 f'Reason: '
                 f'{type(e).__name__} - {str(e)}'
             )
-            self.__logger.error(message)
+            logger.error(message)
             raise PdfServiceInternalError(status_code=HTTPStatus.INTERNAL_SERVER_ERROR, message=message) from e
 
         return url
@@ -109,7 +109,7 @@ def delete_file(self, object_name: str):
         try:
             self.__s3_client.delete_object(Bucket=self.__bucket_name, Key=object_name)
         except Exception as e:
-            self.__logger.error('Failed to delete file (%s), Reason: %s - %s', object_name, type(e).__name__, str(e))
+            logger.error('Failed to delete file (%s), Reason: %s - %s', object_name, type(e).__name__, str(e))
             return False
 
         return True
@@ -118,7 +118,7 @@ def get_file(self, object_key):
         try:
             s3_response = self.__s3_client.get_object(Bucket=self.__bucket_name, Key=object_key)
         except Exception as e:
-            self.__logger.error('Failed to get file (%s), Reason: %s - %s', object_key, type(e).__name__, str(e))
+            logger.error('Failed to get file (%s), Reason: %s - %s', object_key, type(e).__name__, str(e))
             return None
 
         return s3_response['Body'].read()
@@ -135,5 +135,5 @@ def copy_object(self, src_bucket, src_key, target_key):
                 CopySource={'Bucket': src_bucket, 'Key': src_key}, Bucket=self.__bucket_name, Key=target_key
             )
         except Exception as e:
-            self.__logger.error('Failed to get copy (%s), Reason: %s - %s', src_key, type(e).__name__, str(e))
+            logger.error('Failed to get copy (%s), Reason: %s - %s', src_key, type(e).__name__, str(e))
             return

serverless.yml

@@ -2,7 +2,7 @@ service: sparcs-certificate-service
 
 custom:
   projectName: sparcs-events
-  serviceName: certificate
+  serviceName: certificate-generator
   stage: ${opt:stage, self:provider.stage}
   entities: ${self:custom.stage}-${self:custom.projectName}-entities
   bucket: ${self:custom.stage}-${self:custom.projectName}-file-bucket
@@ -19,6 +19,7 @@ custom:
       compatibleRuntimes:
         - python3.8
     slim: true
+  logRetentionInDays: 30
 
 provider:
   name: aws
@@ -46,10 +47,11 @@ functions:
 plugins:
   - serverless-python-requirements
   - serverless-iam-roles-per-function
+  - serverless-plugin-log-retention
 
 layers:
-  wkhtmltopdf:
-    name: ${self:custom.projectName}-${self:custom.stage}-wkhtmltopdf
-    description: wkhtmltopdf Linux Package Layer
+  weazyprint:
+    name: ${self:custom.projectName}-${self:custom.stage}-weazyprint
+    description: weazyprint Linux Package Layer
     package:
-      artifact: layers/wkhtmltox-0.12.6-4.amazonlinux2_lambda.zip
+      artifact: layers/weasyprint-layer-python3.8.zip

template/certificate_template.html

@@ -4,48 +4,43 @@
     <meta charset="UTF-8">
     <title>{{name}} Certificate</title>
     <style>
-        * {
-            margin: 0;
-            padding: 0;
-            box-sizing: border-box;
-            overflow: hidden;
-        }
-
-        body {
+         html, body {
             width: 100%;
             height: 100%;
+            margin: 0;
+            padding: 0;
             position: relative;
         }
 
         img {
             width: 100%;
             height: auto;
-            object-fit: cover; /* Ensure the image covers the full area without distortion */
+            object-fit: cover;
+            position: absolute;
+            left: 0;
+            top: 0;
+            z-index: -1;
         }
 
-        span {
+        .centered {
             text-align: center;
             font-family: Verdana, sans-serif;
             font-size: 48px;
+            font-weight: bold;
+            position: absolute;
+            top: 50%;
+            left: 50%;
+            transform: translate(-50%, -50%);
         }
 
-        div {
-            display: -webkit-box; /* wkhtmltopdf uses this one */
-            display: flex;
-            -webkit-box-pack: center; /* wkhtmltopdf uses this one */
-            justify-content: center;
-            align-items: center;
-            -webkit-box-align: center; /* wkhtmltopdf uses this one */
-            width: 100%;
-            height: 100%;
-            position: absolute;
-            top: 0;
+        .centered span {
+            white-space: nowrap; /* Prevents the text from wrapping */
         }
     </style>
 </head>
 <body>
     <img src="{{template_img}}">
-    <div>
+    <div class="centered">
         <span>{{name}}</span>
     </div>
 </body>

usecase/certificate_usecase.py

@@ -4,7 +4,8 @@
 
 import fitz
 import jinja2
-import pdfkit
+import weasyprint
+from weasyprint import CSS
 
 from model.registrations.registration import RegistrationIn
 from repository.events_repository import EventsRepository
@@ -65,22 +66,8 @@ def generate_certficates(self, event_id: str):
                     # Convert HTML to PDF-------------------------------------------------------------------------------------------------
                     certificate_name_pdf = f'{certificate_name}.pdf'
                     certificate_path = os.path.join(tmpdir, certificate_name_pdf)
-                    options = {
-                        'page-size': 'A4',
-                        'orientation': 'Landscape',
-                        'margin-top': '0mm',
-                        'margin-right': '0mm',
-                        'margin-bottom': '0mm',
-                        'margin-left': '0mm',
-                        'encoding': "UTF-8",
-                        "print-media-type": True,
-                        'disable-smart-shrinking': True,
-                        "enable-local-file-access": True,
-                    }
-                    PATH_WKHTMLTOPDF = '/opt/bin/wkhtmltopdf'
-                    PDFKIT_CONFIG = pdfkit.configuration(wkhtmltopdf=PATH_WKHTMLTOPDF)
-                    pdfkit.from_string(
-                        input=html_out, output_path=certificate_path, options=options, configuration=PDFKIT_CONFIG
+                    weasyprint.HTML(output_path).write_pdf(
+                        certificate_path, stylesheets=[CSS(string='@page { size: A4 landscape; margin: 0;}')]
                     )
 
                     # Get only the first page of the PDF----------------------------------------------------------------------------------