atk-common 1.21.0__py3-none-any.whl → 1.23.0__py3-none-any.whl

atk_common/__init__.py

@@ -1,5 +1,6 @@
 # __init__.py
-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.datetime_utils import 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, date_time_utc_column
 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,10 +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_column',
     'get_utc_date_time',
     'seconds_to_utc_timestamp',
     'get_utc_date_from_iso',
+    'sql',
+    'sql_with_record',
+    'convert_none_to_null',
+    'date_time_utc_column',
     'get_env_value',
     'get_message',
     'create_error_log',

atk_common/datetime_utils.py

@@ -1,8 +1,5 @@
 from datetime import datetime, timezone
 
-def date_time_utc_column(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))
 

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.21.0.dist-info → atk_common-1.23.0.dist-info}/METADATA

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

{atk_common-1.21.0.dist-info → atk_common-1.23.0.dist-info}/RECORD

@@ -1,6 +1,6 @@
-atk_common/__init__.py,sha256=rEx7ncUlputDjFHrtdEqPPVz1QIE9RhsDWWZWBe2GY0,1219
-atk_common/datetime_utils.py,sha256=bkWmLnXfSlr5IIlovdF3UEhAxFJEeskqKrav9uIpek8,512
-atk_common/db_utils.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+atk_common/__init__.py,sha256=G2r5JTcOHLwHFrvpLC4L-t3JlilWQq8-oSDMeUF3AYA,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.21.0.dist-info/licenses/license.txt,sha256=_0O6fWM00-wTurDjnZhUP_N5QiwGhItaQZqHq5eqadA,1063
+atk_common-1.23.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.21.0.dist-info/METADATA,sha256=FapJW4X2OPpdY8ubc09y6yzBV_EwBSDMDhtHiAJkkfk,1464
-atk_common-1.21.0.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
-atk_common-1.21.0.dist-info/top_level.txt,sha256=4CwRjkLnheIdI4jQwc4tK3dbRc58WqUmoqjkdDTWlME,41
-atk_common-1.21.0.dist-info/RECORD,,
+atk_common-1.23.0.dist-info/METADATA,sha256=t7TM3cs6MvcRyPuUtrhyfWjOYjB1P3hat9siC2IkcOM,1464
+atk_common-1.23.0.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
+atk_common-1.23.0.dist-info/top_level.txt,sha256=4CwRjkLnheIdI4jQwc4tK3dbRc58WqUmoqjkdDTWlME,41
+atk_common-1.23.0.dist-info/RECORD,,