add server to gen pdf

order-events/app/events/billing/close_window.py

@@ -1,68 +1,93 @@
-import locale
-import os
-from datetime import date
+import json
 
+import requests
 from aws_lambda_powertools import Logger
+from aws_lambda_powertools.shared.json_encoder import Encoder
 from aws_lambda_powertools.utilities.data_classes import (
     EventBridgeEvent,
     event_source,
 )
 from aws_lambda_powertools.utilities.typing import LambdaContext
-from jinja2 import Environment, FileSystemLoader
-from layercake.dateutils import fromisoformat
+from layercake.dateutils import now
 from layercake.dynamodb import DynamoDBPersistenceLayer, KeyPair
-from weasyprint import HTML
 
-from boto3clients import dynamodb_client
-from config import ORDER_TABLE
+from boto3clients import dynamodb_client, s3_client
+from config import BUCKET_NAME, ORDER_TABLE
 
 logger = Logger(__name__)
 order_layer = DynamoDBPersistenceLayer(ORDER_TABLE, dynamodb_client)
-curdir = os.path.dirname(__file__)
-env = Environment(loader=FileSystemLoader(curdir))
-locale.setlocale(locale.LC_ALL, 'pt_BR.UTF-8')
-
-
-def currency(value: float | int) -> str:
-    return locale.currency(value, grouping=True)
-
-
-def datetime_format(dt: date, fmt='%H:%M %d-%m-%y'):
-    if isinstance(dt, str):
-        dt = fromisoformat(dt)  # type: ignore
-
-    return dt.strftime(fmt)
-
-
-env.filters['datetime_format'] = datetime_format
-env.filters['currency'] = currency
 
 
 @event_source(data_class=EventBridgeEvent)
 @logger.inject_lambda_context
 def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
-    new_image = event.detail['new_image']
-    _, start_date, _, end_date, *_ = new_image['sk'].split('#')
+    keys = event.detail['keys']
+    now_ = now()
+    # Key pattern `BILLING#ORG#{org_id}`
+    *_, org_id = keys['id'].split('#')
+    # Key pattern `START#{start_date}#END#{end_date}#SCHEDULE#AUTO_CLOSE`
+    _, start_date, _, end_date, *_ = keys['sk'].split('#')
 
     result = order_layer.collection.query(
         KeyPair(
-            pk=new_image['id'],
+            pk=keys['id'],
             sk=f'START#{start_date}#END#{end_date}#ENROLLMENT',
         ),
     )
 
-    template = env.get_template('tmpl.html')
-    html_rendered = template.render(
-        start_date=start_date,
-        end_date=end_date,
-        items=result['items'],
+    r = requests.post(
+        'https://weasyprint.saladeaula.digital',
+        data=json.dumps(
+            {
+                'template_s3_uri': 's3://saladeaula.digital/billing/template.html',
+                'template_vars': {
+                    'start_date': start_date,
+                    'end_date': end_date,
+                    'items': result['items'],
+                },
+            },
+            cls=Encoder,
+        ),
     )
+    r.raise_for_status()
 
-    HTML(string=html_rendered, base_url='').write_pdf('cert.pdf')
+    object_key = f'billing/{org_id}/{start_date}_{end_date}.pdf'
+    s3_uri = f's3://{BUCKET_NAME}/{object_key}'
 
-    return order_layer.update_item(
-        key=KeyPair(new_image['id'], new_image['sk']),
-        update_expr='SET #status = :status',
-        expr_attr_names={'#status': 'status'},
-        expr_attr_values={':status': 'CLOSED'},
-    )
+    try:
+        s3_client.put_object(
+            Bucket=BUCKET_NAME,
+            Key=object_key,
+            Body=r.content,
+            ContentType='application/pdf',
+        )
+    except Exception as exc:
+        logger.exception(exc)
+        raise
+
+    with order_layer.transact_writer() as transact:
+        transact.update(
+            key=KeyPair(
+                pk=keys['id'],
+                sk=f'START#{start_date}#END#{end_date}',
+            ),
+            update_expr='SET #status = :status, s3_uri = :s3_uri, \
+                        updated_at = :updated_at',
+            expr_attr_names={'#status': 'status'},
+            expr_attr_values={
+                ':status': 'CLOSED',
+                ':s3_uri': s3_uri,
+                ':updated_at': now_,
+            },
+            cond_expr='attribute_exists(sk)',
+        )
+        transact.put(
+            item={
+                'id': keys['id'],
+                'sk': '{sk}#EXECUTED'.format(sk=keys['sk']),
+                'created_at': now_,
+            }
+        )
+
+    logger.info(f'PDF uploaded successfully to {s3_uri}')
+    return True