Add obk core views (#30)

* Fix tests and dataclass inerithance

* Fix tests and dataclass inerithance

* Fix tests and dataclass inerithance

* Update changelog

* Remove comments

* apply black

* Revert "apply black"

This reverts commit 03a5f8f.

* Apply linter

* Apply linter to tests

* remove black

* Add test async event handler

* Add new logger module

* add picpay logger

* Add docstrings

* Fix typos

* improve configurations

* Add new views and config envrionments

Co-authored-by: GBgustavodutra <[email protected]>
Co-authored-by: Hugo Saito <[email protected]>

Author: 3 people

events_protocol/__init__.py

@@ -1,2 +1,3 @@
-__version__ = "0.3.1"  # pragma: no cover
+__version__ = "0.3.2"  # pragma: no cover
+
 

events_protocol/core/builder.py

@@ -16,7 +16,7 @@ class EventBuilder(LoggableMixin):
     @classmethod
     def error_for(
         cls,
-        exception: typing.Union[EventException],
+        exception: EventException,
         event: typing.Optional[Event] = Event(name="", version=1, id=str(uuid4())),
         id_flow=str(uuid4()),
         loggable=True,

events_protocol/core/logging/mixins/__init__.py

@@ -1 +1 @@
-from logging import Logger
+from .loggable import LoggableMixin

events_protocol/core/urls.py

@@ -0,0 +1,5 @@
+from events_protocol.core.views.aiohttp import AIOHTTPHealthCheckView
+
+from events_protocol.core.views.event import EventView
+
+URL_PATTERNS = [(EventView, r"/events"), (AIOHTTPHealthCheckView, r"/health")]

events_protocol/core/utils/__init__.py

@@ -0,0 +1 @@
+from .config import config

events_protocol/core/utils/config.py

@@ -0,0 +1,27 @@
+import os
+import typing
+import glob
+
+from decouple import Config, RepositoryEnv, Undefined
+
+def config(
+    config_name: str,
+    default: typing.Any = None,
+    cast: typing.Type = None,
+    env_file=os.environ.get("APPLICATION_ENVIRONMENT") or "dev"
+):
+    try:
+        available_env_files = glob.glob("./**/*.env", recursive=True)
+        env_file_path = [file for file in available_env_files if file.endswith(f'/{env_file}.env')][0]
+        _config = Config(RepositoryEnv(env_file_path))
+    except IndexError as ex:
+        raise FileNotFoundError(f'Can\'t find environment file {env_file}.env')
+    except Exception as ex:
+        raise OSError(f'Can\'t parse env file {env_file}.env: {ex}')
+    
+    cfg = None
+    try:
+        cfg = _config(config_name, default=default or Undefined(), cast=cast or Undefined())
+    except Exception:
+        cfg = cast(default) if cast else default
+    return cfg

events_protocol/core/utils/encoder.py

@@ -0,0 +1,12 @@
+import datetime
+import json
+from uuid import UUID
+
+
+class JSONEncoder(json.JSONEncoder):
+    def default(self, obj):
+        if isinstance(obj, UUID):
+            return obj.hex
+        elif isinstance(obj, datetime.datetime):
+            return str(obj)
+        return json.JSONEncoder.default(self, obj)

events_protocol/core/utils/sync.py

@@ -0,0 +1,2 @@
+from .aiohttp import *
+from .event import *

events_protocol/core/views/__init__.py

@@ -0,0 +1,85 @@
+import json
+import tracemalloc
+import typing
+from http import HTTPStatus
+
+from aiohttp.web import Application, Request, Response, View
+from aiohttp.web_middlewares import _Middleware as AIOHTTPMiddleware
+
+from events_protocol.core.views.base import BaseHealth, BaseView
+from events_protocol.core.utils.encoder import JSONEncoder
+
+
+async def init_app(
+    routes: typing.List[typing.Tuple[View, str]],
+    middlewares: typing.Iterable[AIOHTTPMiddleware] = [],
+) -> Application:
+
+    tracemalloc.start()
+    app = Application(middlewares=middlewares)
+    for view, path in routes:
+        app.router.add_view(path, view)
+
+    return app
+
+
+_NOT_ALLOWED_AIOHTTP = Response(status=HTTPStatus.METHOD_NOT_ALLOWED)
+
+
+class AIOHTTPView(BaseView, View):
+    request: Request
+    body: str = None
+
+    def __init__(self, *args, **kwargs):
+        self.request = args[0]
+        super().__init__(*args, **kwargs)
+
+    def __iter__(self):
+        return self._iter().__await__()
+
+    async def get_body(self):
+        if not self.body and self.request.body_exists:
+            _body: bytes = await self.request.content.read(-1)
+            self.body = _body.decode("utf-8")
+        return self.body
+
+    async def get(self, *args, **kwargs):
+        return await self._get(*args, **kwargs)
+
+    async def _get(self, *args, **kwargs):
+        return _NOT_ALLOWED_AIOHTTP
+
+    async def put(self, *args, **kwargs):
+        await self.get_body()
+        return await self._put(*args, **kwargs)
+
+    async def _put(self, *args, **kwargs):
+        return _NOT_ALLOWED_AIOHTTP
+
+    async def post(self, *args, **kwargs):
+        await self.get_body()
+        return await self._post(*args, **kwargs)
+
+    async def _post(self, *args, **kwargs):
+        return _NOT_ALLOWED_AIOHTTP
+
+    async def delete(self, *args, **kwargs):
+        return await self._delete(*args, **kwargs)
+
+    async def _delete(self, *args, **kwargs):
+        return _NOT_ALLOWED_AIOHTTP
+
+    def get_query_args(self):
+        return self.request.rel_url.query or dict()
+
+    async def write_response(
+        self, http_status: HTTPStatus, response_body: dict, headers: dict = {},
+    ):
+        headers.update(self._base_header)
+        return Response(
+            body=json.dumps(response_body, cls=JSONEncoder), status=http_status, headers=headers,
+        )
+
+
+class AIOHTTPHealthCheckView(BaseHealth, AIOHTTPView):
+    pass