atk-common 0.6.0__tar.gz → 1.0.0__tar.gz

{atk_common-0.6.0 → atk_common-1.0.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: atk_common
-Version: 0.6.0
+Version: 1.0.0
 Summary: ATK common methods
 Home-page: https://github.com/pypa/atk_common
 Author: Roger

{atk_common-0.6.0 → atk_common-1.0.0}/setup.py

@@ -36,7 +36,7 @@ setup(
     # For a discussion on single-sourcing the version across setup.py and the
     # project code, see
     # https://packaging.python.org/guides/single-sourcing-package-version/
-    version="0.6.0",  # Required
+    version="1.0.0",  # Required
     # This is a one-line description or tagline of what your project does. This
     # corresponds to the "Summary" metadata field:
     # https://packaging.python.org/specifications/core-metadata/#summary

{atk_common-0.6.0 → atk_common-1.0.0}/src/atk_common.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: atk_common
-Version: 0.6.0
+Version: 1.0.0
 Summary: ATK common methods
 Home-page: https://github.com/pypa/atk_common
 Author: Roger

{atk_common-0.6.0 → atk_common-1.0.0}/src/atk_common.egg-info/SOURCES.txt

@@ -10,7 +10,10 @@ src/atk_common.egg-info/requires.txt
 src/atk_common.egg-info/top_level.txt
 src/atk_package/__init__.py
 src/atk_package/datetime_utils.py
+src/atk_package/env_utils.py
+src/atk_package/error_utils.py
 src/atk_package/log_utils.py
 src/atk_package/rabbitmq_consumer.py
+src/atk_package/response_utils.py
 src/shared_python_atk_enforcement/__init__.py
 tests/test_log_utils.py

atk_common-1.0.0/src/atk_package/__init__.py

@@ -0,0 +1,21 @@
+# __init__.py
+from atk_package.rabbitmq_consumer import RabbitMQConsumer
+from atk_package.datetime_utils import date_time_utc, get_utc_date_time
+from atk_package.env_utils import get_env_value
+from atk_package.error_utils import get_message, get_error_entity, handle_error, get_response_error
+from atk_package.log_utils import add_log_item, add_log_item_http
+from atk_package.response_utils import create_save_resp
+
+__all__ = [
+    'RabbitMQConsumer',
+    'date_time_utc',
+    'get_utc_date_time',
+    'get_env_value',
+    'get_message',
+    'get_error_entity',
+    'handle_error',
+    'get_response_error',
+    'add_log_item',
+    'add_log_item_http',
+    'create_save_resp'
+]

atk_common-1.0.0/src/atk_package/env_utils.py

@@ -0,0 +1,11 @@
+import os
+from atk_package.log_utils import add_log_item
+
+def get_env_value(key, default_value):
+    try:
+        val = os.environ[key]
+        add_log_item(key + ':' + val)
+        return val
+    except (Exception) as error:
+        add_log_item(key + ':' + default_value)
+        return default_value

atk_common-1.0.0/src/atk_package/error_utils.py

@@ -0,0 +1,40 @@
+from datetime import datetime
+import json
+from atk_package.datetime_utils import get_utc_date_time
+from atk_package.log_utils import add_log_item
+from atk_package.response_utils import create_save_resp
+
+def get_message(error):
+    if hasattr(error, 'message'):
+        return str(error.message)
+    else:
+        return str(error)
+
+def get_error_entity(app, error, component, method, error_type):
+    data = {}
+    created = get_utc_date_time()
+    data['exceptionType'] = str(type(error))
+    data['errorType'] = error_type
+    data['message'] = get_message(error)
+    data['component'] = component
+    data['method'] = method
+    data['timestamp'] = created
+    return app.response_class(
+        response=json.dumps(data),
+        status=500,
+        mimetype='application/json'
+    )
+
+def handle_error(resp, status):
+    if resp.status_code == 500:
+        add_log_item(resp.json().get('message'))
+        return create_save_resp(status, resp.status_code, resp.json())
+    else:
+        add_log_item(resp.text)
+        return create_save_resp(status, resp.status_code, resp.text)
+
+def get_response_error(resp):
+    if resp.status_code == 500:
+        return resp.json()
+    else:
+        return resp.text

atk_common-1.0.0/src/atk_package/response_utils.py

@@ -0,0 +1,6 @@
+def create_save_resp(status, status_code, response_msg):
+    data = {}
+    data['status'] = status
+    data['statusCode'] = status_code
+    data['responseMsg'] = response_msg
+    return data

atk_common-0.6.0/src/atk_package/__init__.py

@@ -1,12 +0,0 @@
-# __init__.py
-from atk_package.rabbitmq_consumer import RabbitMQConsumer
-from atk_package.datetime_utils import date_time_utc, get_utc_date_time
-from atk_package.log_utils import add_log_item, add_log_item_http
-
-__all__ = [
-    'RabbitMQConsumer',
-    'date_time_utc',
-    'get_utc_date_time',
-    'add_log_item',
-    'add_log_item_http'
-]