atk-common 1.0.0__py3-none-any.whl → 1.2.0__py3-none-any.whl

{atk_common-1.0.0.dist-info → atk_common-1.2.0.dist-info}/METADATA

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

atk_common-1.2.0.dist-info/RECORD

@@ -0,0 +1,17 @@
+atk_package/__init__.py,sha256=AkUS0g4hl_YQHI3hOUAxHk1B3KsdsJyDQn2qHuHcpd0,818
+atk_package/datetime_utils.py,sha256=qsVF7l90P1-xukG2tV_jLqG9J_Yfl5wTpyfrdPBlyMo,239
+atk_package/env_utils.py,sha256=4WRQGZz-j2eFqNK3wEK9k3wja0X4_mZmOPUPMX-vSDY,314
+atk_package/error_utils.py,sha256=70-bnNsd_ktHlrDB1o_d_qGUpYvUe25Ycz2myz8vU_o,1424
+atk_package/http_utils.py,sha256=5PjGYy-TiBbtNOElEDhsCs3BPDTfqg_aMQBNCuHfcrQ,90
+atk_package/log_utils.py,sha256=ihsbrooDT3cwdDa9VmSaY6bTPke4tTQm2LhZnmnmLQ4,420
+atk_package/rabbitmq_consumer.py,sha256=wmZQieeONkfXB5RmM6yiNkwl0_QK4oWWDBSfiD47CdA,1753
+atk_package/response_utils.py,sha256=ezqTSNdGwueFXzijKB7CKPWxZml39bqjAZOeIMWMntk,197
+atk_package/enums/__init__.py,sha256=t4f3eoekm9pF38yLLgg0s8niItF8Qi2ybC4MpgtxQOU,238
+atk_package/enums/command_status_enum.py,sha256=M2Nln27a_DbzI07-gfytWQk2X087JhkU6Fmard5qVHs,127
+atk_package/enums/speed_control_status_enum.py,sha256=qpURh0K1L1tSpbrzVnckoe4hUn1illIkbo7k4mLfzIM,182
+shared_python_atk_enforcement/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+atk_common-1.2.0.dist-info/METADATA,sha256=7JlDlP_he9T9YWhKMRoj4Np6EP3huqB0aO1Al_RFp78,1193
+atk_common-1.2.0.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+atk_common-1.2.0.dist-info/license.txt,sha256=_0O6fWM00-wTurDjnZhUP_N5QiwGhItaQZqHq5eqadA,1063
+atk_common-1.2.0.dist-info/top_level.txt,sha256=4HLkJ8mtbeZikG_9tMb5-dw9UWuAQk3xUN_izlisloQ,42
+atk_common-1.2.0.dist-info/RECORD,,

atk_package/__init__.py

@@ -1,13 +1,13 @@
 # __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.error_utils import get_message, get_error_entity, handle_error, get_response_error, get_error_type
+from atk_package.http_utils import is_status_code_ok
 from atk_package.log_utils import add_log_item, add_log_item_http
+from atk_package.rabbitmq_consumer import RabbitMQConsumer
 from atk_package.response_utils import create_save_resp
 
 __all__ = [
-    'RabbitMQConsumer',
     'date_time_utc',
     'get_utc_date_time',
     'get_env_value',
@@ -15,7 +15,10 @@ __all__ = [
     'get_error_entity',
     'handle_error',
     'get_response_error',
+    'get_error_type',
+    'is_status_code_ok',
     'add_log_item',
     'add_log_item_http',
+    'RabbitMQConsumer',
     'create_save_resp'
 ]

atk_package/enums/__init__.py

@@ -0,0 +1,8 @@
+# __init__.py
+from atk_package.enums.command_status_enum import CommandStatusType
+from atk_package.enums.speed_control_status_enum import SpeedControlStatusType
+
+__all__ = [
+    'CommandStatusType',
+    'SpeedControlStatusType'
+]

atk_package/enums/command_status_enum.py

@@ -0,0 +1,7 @@
+from enum import Enum
+ 
+class CommandStatusType(Enum):
+    CREATED = 1
+    COMPLETED = 2
+    EXPIRED = 3
+    FAILED = 4

atk_package/enums/speed_control_status_enum.py

@@ -0,0 +1,10 @@
+from enum import Enum
+ 
+class SpeedControlStatusType(Enum):
+    CREATED = 1
+    CONFIRMED = 2
+    STARTED = 3
+    FAILED = 4
+    PAUSED = 5
+    STOPPED = 6
+    DELETED = 7

atk_package/error_utils.py

@@ -10,15 +10,15 @@ def get_message(error):
     else:
         return str(error)
 
-def get_error_entity(app, error, component, method, error_type):
+def get_error_entity(app, error, component, method, error_type, status_code):
     data = {}
-    created = get_utc_date_time()
+    data['statusCode'] = status_code
     data['exceptionType'] = str(type(error))
     data['errorType'] = error_type
     data['message'] = get_message(error)
     data['component'] = component
     data['method'] = method
-    data['timestamp'] = created
+    data['timestamp'] = get_utc_date_time()
     return app.response_class(
         response=json.dumps(data),
         status=500,
@@ -38,3 +38,11 @@ def get_response_error(resp):
         return resp.json()
     else:
         return resp.text
+
+# Return values:
+# 1 - Connection error
+# 2 - Database error
+def get_error_type(conn):
+    if conn is None:
+        return 1
+    return 2

atk_package/http_utils.py

@@ -0,0 +1,2 @@
+def is_status_code_ok(status_code):
+    return status_code >= 200 and status_code < 300

atk_common-1.0.0.dist-info/RECORD

@@ -1,13 +0,0 @@
-atk_package/__init__.py,sha256=ehWDM4aBgAKPuWWvtRt0NZ3bsCrcTG2BJIaTyk0aKpw,699
-atk_package/datetime_utils.py,sha256=qsVF7l90P1-xukG2tV_jLqG9J_Yfl5wTpyfrdPBlyMo,239
-atk_package/env_utils.py,sha256=4WRQGZz-j2eFqNK3wEK9k3wja0X4_mZmOPUPMX-vSDY,314
-atk_package/error_utils.py,sha256=mU90ftizwEesGrWo1MhzyL3nkpnSWvVS0xCe_C8fr-Q,1249
-atk_package/log_utils.py,sha256=ihsbrooDT3cwdDa9VmSaY6bTPke4tTQm2LhZnmnmLQ4,420
-atk_package/rabbitmq_consumer.py,sha256=wmZQieeONkfXB5RmM6yiNkwl0_QK4oWWDBSfiD47CdA,1753
-atk_package/response_utils.py,sha256=ezqTSNdGwueFXzijKB7CKPWxZml39bqjAZOeIMWMntk,197
-shared_python_atk_enforcement/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-atk_common-1.0.0.dist-info/METADATA,sha256=B-Biz5jS1kf9b1DgqrfCiGlYrDkp1DXXZrVbTibMtI0,1193
-atk_common-1.0.0.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-atk_common-1.0.0.dist-info/license.txt,sha256=_0O6fWM00-wTurDjnZhUP_N5QiwGhItaQZqHq5eqadA,1063
-atk_common-1.0.0.dist-info/top_level.txt,sha256=4HLkJ8mtbeZikG_9tMb5-dw9UWuAQk3xUN_izlisloQ,42
-atk_common-1.0.0.dist-info/RECORD,,