velocity-python 0.0.6__py3-none-any.whl → 0.0.7__py3-none-any.whl

velocity/aws/__init__.py

@@ -1,12 +1,12 @@
 import os
 import requests
 
-from velocity.aws.handlers import LambdaHandler
-from velocity.aws.handlers import SqsHandler
-
 DEBUG = (os.environ.get('ENV') != 'production') \
      or (os.environ.get('DEBUG') == 'Y')
 
+from velocity.aws.handlers import LambdaHandler
+from velocity.aws.handlers import SqsHandler
+
 # This is helpful for running HTTPS clients on lambda.
 if os.path.exists('/opt/python/ca-certificates.crt'):
     os.environ["REQUESTS_CA_BUNDLE"] = '/opt/python/ca-certificates.crt'

velocity/aws/handlers/lambda_handler.py

@@ -1,123 +1,154 @@
 from velocity.misc.format import to_json
 import json
-import pprint
+import sys
+import os
 import traceback
+from velocity.aws import DEBUG
+from support.app import helpers, AlertError, enqueue
+from response import Response
 
 class LambdaHandler:
     def __init__(self, event, context):
         self.event = event
         self.context = context
         self.serve_action_default = True
+        self.skip_action = False
 
-        requestContext = event.get('requestContext', {})
-        identity = requestContext.get('identity', {})
-        headers = event.get('headers', {})
+        requestContext = event.get("requestContext") or {}
+        identity = requestContext.get("identity") or {}
+        headers = event.get("headers") or {}
+        auth = identity.get("cognitoAuthenticationProvider")
         self.session = {
-            'authentication_provider':
-            identity.get('cognitoAuthenticationProvider'),
-            'authentication_type':
-            identity.get('cognitoAuthenticationType'),
-            'cognito_user':
-            identity.get('user'),
-            'is_desktop':
-            headers.get('CloudFront-Is-Desktop-Viewer') == 'true',
-            'is_mobile':
-            headers.get('CloudFront-Is-Mobile-Viewer') == 'true',
-            'is_smart_tv':
-            headers.get('CloudFront-Is-SmartTV-Viewer') == 'true',
-            'is_tablet':
-            headers.get('CloudFront-Is-Tablet-Viewer') == 'true',
-            'origin':
-            headers.get('origin'),
-            'path':
-            event.get('path'),
-            'referer':
-            headers.get('Referer'),
-            'source_ip':
-            identity.get('sourceIp'),
-            'user_agent':
-            identity.get('userAgent'),
+            "authentication_provider": identity.get("cognitoAuthenticationProvider"),
+            "authentication_type": identity.get("cognitoAuthenticationType"),
+            "cognito_user": identity.get("user"),
+            "is_desktop": headers.get("CloudFront-Is-Desktop-Viewer") == "true",
+            "is_mobile": headers.get("CloudFront-Is-Mobile-Viewer") == "true",
+            "is_smart_tv": headers.get("CloudFront-Is-SmartTV-Viewer") == "true",
+            "is_tablet": headers.get("CloudFront-Is-Tablet-Viewer") == "true",
+            "origin": headers.get("origin"),
+            "path": event.get("path"),
+            "referer": headers.get("Referer"),
+            "source_ip": identity.get("sourceIp"),
+            "user_agent": identity.get("userAgent"),
+            "sub": auth.split(":")[-1] if auth else None,
         }
-        if self.session.get('is_mobile'):
-            self.session['device_type'] = 'mobile'
-        elif self.session.get('is_desktop'):
-            self.session['device_type'] = 'desktop'
-        elif self.session.get('is_tablet'):
-            self.session['device_type'] = 'tablet'
-        elif self.session.get('is_smart_tv'):
-            self.session['device_type'] = 'smart_tv'
+        if self.session.get("is_mobile"):
+            self.session["device_type"] = "mobile"
+        elif self.session.get("is_desktop"):
+            self.session["device_type"] = "desktop"
+        elif self.session.get("is_tablet"):
+            self.session["device_type"] = "tablet"
+        elif self.session.get("is_smart_tv"):
+            self.session["device_type"] = "smart_tv"
         else:
-            self.session['device_type'] = 'unknown'
+            self.session["device_type"] = "unknown"
 
-    def serve(self, tx):
-        response = {
-            'statusCode': 200,
-            'body': '{}',
-            'headers': {
-                'Content-Type': 'application/json',
-                "Access-Control-Allow-Origin": "*",
-            },
+    def log(self, tx, message, function=None):
+        if not function:
+            function = "<Unknown>"
+            idx = 0
+            while True:
+                try:
+                    temp = sys._getframe(idx).f_code.co_name
+                except ValueError as e:
+                    break
+                if temp in ["x", "log", "_transaction"]:
+                    idx += 1
+                    continue
+                function = temp
+                break
+
+        data = {
+            "app_name": os.environ["ProjectName"],
+            "source_ip": self.session["source_ip"],
+            "referer": self.session["referer"],
+            "user_agent": self.session["user_agent"],
+            "device_type": self.session["device_type"],
+            "function": function,
+            "message": message,
         }
+        if "email_address" in self.session:
+            data["sys_modified_by"] = self.session["email_address"]
+        tx.table("sys_log").insert(data)
+
+    def serve(self, tx):
+        response = Response()
+        body = self.event.get("body")
+        postdata = {}
+        if isinstance(body, str) and len(body) > 0:
+            try:
+                postdata = json.loads(body)
+            except:
+                postdata = {"raw_body": body}
+        elif isinstance(body, dict):
+            postdata = body
+        elif isinstance(body, list) and len(body) > 0:
+            try:
+                new = "\n".join(body)
+                postdata = json.loads(new)
+            except:
+                postdata = {"raw_body": body}
+
+        req_params = self.event.get("queryStringParameters") or {}
         try:
-            postdata = {}
-            if self.event.get('body'):
-                postdata = json.loads(self.event.get('body'))
-            req_params = self.event.get('queryStringParameters') or {}
-            if hasattr(self, 'beforeAction'):
-                self.beforeAction(args=req_params,
-                                  postdata=postdata,
-                                  response=response)
+            if hasattr(self, "beforeAction"):
+                self.beforeAction(args=req_params, postdata=postdata, response=response)
             actions = []
-            action = postdata.get('action', req_params.get('action'))
+            action = postdata.get("action", req_params.get("action"))
             if action:
                 actions.append(
-                    f"on action {action.replace('-', ' ').replace('_', ' ')}".
-                    title().replace(' ', ''))
+                    f"on action {action.replace('-', ' ').replace('_', ' ')}".title().replace(
+                        " ", ""
+                    )
+                )
             if self.serve_action_default:
-                actions.append('OnActionDefault')
+                actions.append("OnActionDefault")
             for action in actions:
+                if self.skip_action:
+                    break
                 if hasattr(self, action):
-                    getattr(self, action)(args=req_params,
-                                          postdata=postdata,
-                                          response=response)
+                    result = getattr(self, action)(
+                        args=req_params, postdata=postdata, response=response
+                    )
+                    if result and result != response:
+                        response.set_body(result)
                     break
-            if hasattr(self, 'afterAction'):
-                self.afterAction(args=req_params,
-                                 postdata=postdata,
-                                 response=response)
-
+            if hasattr(self, "afterAction"):
+                self.afterAction(args=req_params, postdata=postdata, response=response)
+        except AlertError as e:
+            response.alert(e.get_payload())
         except Exception as e:
-            response = {
-                'statusCode':
-                500,
-                'body':
-                to_json({
-                    'type': 'Unhandled Exception',
-                    'error_message': str(e),
-                    'user_message': 'Oops! An unhandled error occurred.',
-                    'traceback': traceback.format_exc() if DEBUG else None
-                }),
-                'headers': {
-                    'Content-Type': 'application/json',
-                    'Access-Control-Allow-Origin': '*'
-                }
-            }
-            if hasattr(self, 'onError'):
-                self.onError(args=req_params,
-                             postdata=postdata,
-                             response=response,
-                             exc=e,
-                             tb=traceback.format_exc())
+            response.exception()
+            if hasattr(self, "onError"):
+                self.onError(
+                    args=req_params,
+                    postdata=postdata,
+                    response=response,
+                    exc=e.__class__.__name__,
+                    tb=traceback.format_exc(),
+                )
 
-        return response
+        return response.render()
+
+    def track(self, tx, data={}, user=None):
+        data = data.copy()
+        data.update(
+            {
+                "source_ip": self.session["source_ip"],
+                "referer": self.session["referer"],
+                "user_agent": self.session["user_agent"],
+                "device_type": self.session["device_type"],
+                "sys_modified_by": self.session["email_address"],
+            }
+        )
+        tx.table(helpers.get_tracking_table(user or self.session)).insert(data)
 
     def OnActionDefault(self, tx, args, postdata, response):
-        response['body'] = to_json({'event': self.event, 'postdata': postdata})
+        return {"event": self.event, "postdata": postdata}
 
-    def onError(self, tx, args, postdata, response, exc, tb):
-        pprint.pprint({
-            'message': 'Unhandled Exception',
-            'exception': str(exc),
-            'traceback': traceback.format_exc()
-        })
+    def OnActionTracking(self, tx, args, postdata, response):
+        self.track(tx, postdata.get("payload", {}).get("data", {}))
 
+    def enqueue(self, tx, action, payload={}):
+        return enqueue(tx, action, payload, self.session["email_address"])

velocity/aws/handlers/response.py

@@ -1,10 +1,8 @@
 from velocity.misc.format import to_json
 import sys
 import traceback
-import os
+from velocity.aws import DEBUG
 
-DEBUG = (os.environ.get('ENV') != 'production') or \
-        (os.environ.get('DEBUG') == 'Y')
 
 variants = [
     'success',

velocity/aws/handlers/sqs_handler.py

@@ -1,48 +1,85 @@
 from velocity.misc.format import to_json
 import json
+import sys
+import os
 import traceback
+from velocity.aws import DEBUG
 
 class SqsHandler:
     def __init__(self, event, context):
         self.event = event
         self.context = context
         self.serve_action_default = True
+        self.skip_action = False
+
+    def log(self, tx, message, function=None):
+        if not function:
+            function = "<Unknown>"
+            idx = 0
+            while True:
+                try:
+                    temp = sys._getframe(idx).f_code.co_name
+                except ValueError as e:
+                    break
+                if temp in ["x", "log", "_transaction"]:
+                    idx += 1
+                    continue
+                function = temp
+                break
+
+        data = {
+            "app_name": os.environ["ProjectName"],
+            "referer": "SQS",
+            "user_agent": "QueueHandler",
+            "device_type": "Lambda",
+            "function": function,
+            "message": message,
+            "sys_modified_by": "lambda:BackOfficeQueueHandler",
+        }
+        tx.table("sys_log").insert(data)
 
     def serve(self, tx):
-        records = self.event.get('Records', [])
-        print(f"Handling batch of {len(records)} records from SQS")
+        records = self.event.get("Records", [])
         for record in records:
-            print(f"Start MessageId {record.get('messageId')}")
-            attrs = record.get('attributes')
+            attrs = record.get("attributes")
             try:
                 postdata = {}
-                if record.get('body'):
-                    postdata = json.loads(record.get('body'))
-                if hasattr(self, 'beforeAction'):
+                if record.get("body"):
+                    postdata = json.loads(record.get("body"))
+                if hasattr(self, "beforeAction"):
                     self.beforeAction(attrs=attrs, postdata=postdata)
                 actions = []
-                action = postdata.get('action')
+                action = postdata.get("action")
                 if action:
                     actions.append(
-                        f"on action {action.replace('-', ' ').replace('_', ' ')}"
-                        .title().replace(' ', ''))
+                        f"on action {action.replace('-', ' ').replace('_', ' ')}".title().replace(
+                            " ", ""
+                        )
+                    )
                 if self.serve_action_default:
-                    actions.append('OnActionDefault')
+                    actions.append("OnActionDefault")
                 for action in actions:
+                    if self.skip_action:
+                        return
                     if hasattr(self, action):
                         getattr(self, action)(attrs=attrs, postdata=postdata)
                         break
-                if hasattr(self, 'afterAction'):
+                if hasattr(self, "afterAction"):
                     self.afterAction(attrs=attrs, postdata=postdata)
             except Exception as e:
-                if hasattr(self, 'onError'):
-                    self.onError(attrs=attrs,
-                                 postdata=postdata,
-                                 exc=e,
-                                 tb=traceback.format_exc())
+                if hasattr(self, "onError"):
+                    self.onError(
+                        attrs=attrs,
+                        postdata=postdata,
+                        exc=e.__class__.__name__,
+                        tb=traceback.format_exc(),
+                    )
 
     def OnActionDefault(self, tx, attrs, postdata):
         print(
-            "Action handler not found. Calling default action `SqsHandler.OnActionDefault` with the following parameters for tx, attrs, and postdata:"
-        )
-        print({'tx': tx, 'attrs': attrs, 'postdata': postdata})
+            f"""
+            [Warn] Action handler not found. Calling default action `SqsHandler.OnActionDefault` with the following parameters for attrs, and postdata:
+            attrs: {str(attrs)}
+            postdata: {str(postdata)}
+            """
+        )

{velocity_python-0.0.6.dist-info → velocity_python-0.0.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: velocity-python
-Version: 0.0.6
+Version: 0.0.7
 Summary: A rapid application development library for interfacing with data storage
 Author-email: Paul Perez <pperez@codeclubs.org>
 Project-URL: Homepage, https://codeclubs.org/projects/velocity

{velocity_python-0.0.6.dist-info → velocity_python-0.0.7.dist-info}/RECORD

@@ -1,10 +1,10 @@
 velocity/__init__.py,sha256=zOzb6090Y-kq0sFFXB-uZl9DFpojkKMKiXgXSbh57HQ,55
-velocity/aws/__init__.py,sha256=9hVUbHjWh1HvD-xolBY_jLnV_bARlHyxkl5_l2nnINg,713
+velocity/aws/__init__.py,sha256=Bfov3WUtS0JvxFx1jEcG_ni9n5TeqIlOifVsamaV_8w,713
 velocity/aws/context.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 velocity/aws/handlers/__init__.py,sha256=OVqAh3Ln501EnmWUz2fq4lhokGHga6KSPA82yU4kRQs,119
-velocity/aws/handlers/lambda_handler.py,sha256=3mCFnFXCZReyOcNoFhj2kyXjokT8TsDb8Rc_IDasg6Q,4586
-velocity/aws/handlers/response.py,sha256=mO4bW595_AVuYE0r-YXD2T5kZyVyCywCg4yAQJJowUo,3988
-velocity/aws/handlers/sqs_handler.py,sha256=OBpmK0_ZTollITPFcQifpUfwjwNmR6Apkl0LKn2QJdk,2033
+velocity/aws/handlers/lambda_handler.py,sha256=8_YgqB_zw4cu_H8kIAMw3wyp0IRLtKipZTWWrKWbXlg,6019
+velocity/aws/handlers/response.py,sha256=VN4uFI_wtHiqAyBLX-eeQlmCwj1hdIvokvZLjsOyCnw,3915
+velocity/aws/handlers/sqs_handler.py,sha256=VB9XMZVqUTeXxgUJswa8Hl8wYGtEKgFp3lQKJE5Icx0,3014
 velocity/db/__init__.py,sha256=vrn2AFNAKaqTdnPwLFS0OcREcCtzUCOodlmH54U7ADg,200
 velocity/db/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 velocity/db/core/column.py,sha256=vB7dCrcIWFvRG1fgyzDQnEu7j_0KYxUuDSTCWVCYlPc,6153
@@ -31,8 +31,8 @@ velocity/misc/format.py,sha256=2GwdWOpUqmQjtLJKrHjQjTMFrATF6G1IXkLI1xVPB7U,2301
 velocity/misc/mail.py,sha256=fKfJtEkRKO3f_JbHNw9ThxKHJnChlBMsQwmEDFgj264,2096
 velocity/misc/merge.py,sha256=hBYPJy6lGL8lzb0wK1i1oEAyzragTWKF1rW9_PGHm6s,918
 velocity/misc/timer.py,sha256=wMbV-1yNXeCJo_UPi5sTSslu9hPzokLaIKgd98tyG_4,536
-velocity_python-0.0.6.dist-info/LICENSE,sha256=aoN245GG8s9oRUU89KNiGTU4_4OtnNmVi4hQeChg6rM,1076
-velocity_python-0.0.6.dist-info/METADATA,sha256=O5NgDrfTR3lEHRqwLihLNwQ4qcVq7iJvf0aRXloUnUo,8521
-velocity_python-0.0.6.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-velocity_python-0.0.6.dist-info/top_level.txt,sha256=JW2vJPmodgdgSz7H6yoZvnxF8S3fTMIv-YJWCT1sNW0,9
-velocity_python-0.0.6.dist-info/RECORD,,
+velocity_python-0.0.7.dist-info/LICENSE,sha256=aoN245GG8s9oRUU89KNiGTU4_4OtnNmVi4hQeChg6rM,1076
+velocity_python-0.0.7.dist-info/METADATA,sha256=yBaBcYrDdsBYj28SCNjTo_bQdc0SZ9awpIRxNX6Jzp0,8521
+velocity_python-0.0.7.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+velocity_python-0.0.7.dist-info/top_level.txt,sha256=JW2vJPmodgdgSz7H6yoZvnxF8S3fTMIv-YJWCT1sNW0,9
+velocity_python-0.0.7.dist-info/RECORD,,