atk-common 1.20.0__py3-none-any.whl → 1.22.0__py3-none-any.whl

atk_common/__init__.py

@@ -1,5 +1,6 @@
 # __init__.py
-from atk_common.datetime_utils import date_time_utc, get_utc_date_time
+from atk_common.datetime_utils import date_time_utc_column, get_utc_date_time, seconds_to_utc_timestamp, get_utc_date_from_iso
+from atk_common.db_utils import sql, sql_with_record, convert_none_to_null
 from atk_common.env_utils import get_env_value
 from atk_common.error_utils import get_message, create_error_log, get_error_entity, handle_error, get_response_error, get_error_type
 from atk_common.http_utils import is_http_status_ok, is_http_status_internal, get_test_response
@@ -9,8 +10,13 @@ from atk_common.response_utils import create_response, is_response_ok
 from atk_common.docker_utils import get_current_container_info
 
 __all__ = [
-    'date_time_utc',
+    'date_time_utc_column',
     'get_utc_date_time',
+    'seconds_to_utc_timestamp',
+    'get_utc_date_from_iso',
+    'sql',
+    'sql_with_record',
+    'convert_none_to_null',
     'get_env_value',
     'get_message',
     'create_error_log',

atk_common/datetime_utils.py

@@ -1,7 +1,12 @@
 from datetime import datetime, timezone
 
-def date_time_utc(column):
-    return 'TO_CHAR({0} at time zone \'UTC\', \'yyyy-mm-dd hh24:mi:ss.ms"Z"\')'.format(column)
-
 def get_utc_date_time():
     return str(datetime.now(timezone.utc))
+
+def seconds_to_utc_timestamp(seconds):
+    return str(datetime.fromtimestamp(seconds, tz=timezone.utc))
+
+def get_utc_date_from_iso(date_time):
+    dt = datetime.fromisoformat(date_time)
+    dt_utc = dt.astimezone(timezone.utc)
+    return str(dt_utc.date())

atk_common/db_utils.py

@@ -0,0 +1,26 @@
+import psycopg2
+from psycopg2.extras import RealDictCursor
+
+def sql(connection, sql, get_one):
+    with connection:
+        with connection.cursor(cursor_factory=RealDictCursor) as cursor:
+            cursor.execute(sql)
+            if get_one:
+                return cursor.fetchone()
+            return cursor.fetchall()
+
+def sql_with_record(connection, sql, record):
+    with connection:
+        with connection.cursor(cursor_factory=RealDictCursor) as cursor:
+            cursor.execute(sql, record)
+            return cursor.fetchone()
+
+def convert_none_to_null(value):
+    if value is None or value == '':
+        return 'null'
+    if type(value) is str:
+        return '\'' + value + '\''
+    return value
+
+def date_time_utc_column(column):
+    return 'TO_CHAR({0} at time zone \'UTC\', \'yyyy-mm-dd hh24:mi:ss.ms"Z"\')'.format(column)

{atk_common-1.20.0.dist-info → atk_common-1.22.0.dist-info}/METADATA

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

{atk_common-1.20.0.dist-info → atk_common-1.22.0.dist-info}/RECORD

@@ -1,6 +1,6 @@
-atk_common/__init__.py,sha256=nG-PFquhz5ycFh1zY1zLqAFPOCbalD80_YmLcJ4vqr0,1093
-atk_common/datetime_utils.py,sha256=qsVF7l90P1-xukG2tV_jLqG9J_Yfl5wTpyfrdPBlyMo,239
-atk_common/db_utils.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+atk_common/__init__.py,sha256=1BDL1loiv5TTOOVAXL1ZhJ-yYox3ema2t_mGpNraUYg,1360
+atk_common/datetime_utils.py,sha256=tMXB3ZuRG8GyA25YLaNXhs4cVHF5zWdw9fXOh0Z9kQw,379
+atk_common/db_utils.py,sha256=6S-WlQltnYjrG0AJ1BXwLgkqT4T6cQmHnHI9ksCdy38,869
 atk_common/docker_utils.py,sha256=nkUhp2OhwJsm41JDujHDY-JxQfdDjy91duprH59Esfk,2084
 atk_common/env_utils.py,sha256=66v-S0vlkJNaun6ay1XcOGRtYJb4Ce4odJsWxuaHsQc,373
 atk_common/error_utils.py,sha256=7SN-Jz3i2MGIPqsexnjgbZg5poMnALLEiCJQ4za5im4,2398
@@ -23,7 +23,7 @@ atk_common/enums/section_role_enum.py,sha256=9rwep4Wmvyvnlrey08TRRoi8VQYAPkylpP0
 atk_common/enums/sensor_type_enum.py,sha256=UBa3LPyGTKgEJV578CqFlwXjnlXqVlybrvzdjsHhjSU,80
 atk_common/enums/speed_control_status_enum.py,sha256=qpURh0K1L1tSpbrzVnckoe4hUn1illIkbo7k4mLfzIM,182
 atk_common/enums/violation_type_enum.py,sha256=01qTHOj-O8bOc-nwIHVnxLosm4cusD_YuqXM3ZsiRRk,86
-atk_common-1.20.0.dist-info/licenses/license.txt,sha256=_0O6fWM00-wTurDjnZhUP_N5QiwGhItaQZqHq5eqadA,1063
+atk_common-1.22.0.dist-info/licenses/license.txt,sha256=_0O6fWM00-wTurDjnZhUP_N5QiwGhItaQZqHq5eqadA,1063
 atk_package/__init__.py,sha256=EceDa352WFHTdC_LxYANXf_TS9pGzK3MCx87bwAKTvg,811
 atk_package/datetime_utils.py,sha256=qsVF7l90P1-xukG2tV_jLqG9J_Yfl5wTpyfrdPBlyMo,239
 atk_package/env_utils.py,sha256=bXOrxM3fZUslqfmZt75iphbEJHbG4riJa8XOVzPwIII,313
@@ -36,7 +36,7 @@ atk_package/enums/__init__.py,sha256=oW0aVnwewfH6sNu14jjZ5_uP4iRN4pRmGdTNC204rHo
 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.20.0.dist-info/METADATA,sha256=tGgmjgTI8pwI6wAhOupxVqT12rZp6CAF-5gkYRk1BoI,1464
-atk_common-1.20.0.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
-atk_common-1.20.0.dist-info/top_level.txt,sha256=4CwRjkLnheIdI4jQwc4tK3dbRc58WqUmoqjkdDTWlME,41
-atk_common-1.20.0.dist-info/RECORD,,
+atk_common-1.22.0.dist-info/METADATA,sha256=RR_FNbXxiPxUclSI0J-9jX320CrewoFnDIEw-QUhsgM,1464
+atk_common-1.22.0.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
+atk_common-1.22.0.dist-info/top_level.txt,sha256=4CwRjkLnheIdI4jQwc4tK3dbRc58WqUmoqjkdDTWlME,41
+atk_common-1.22.0.dist-info/RECORD,,