ry-pg-utils 1.0.8__tar.gz → 1.0.10__tar.gz

{ry_pg_utils-1.0.8/src/ry_pg_utils.egg-info → ry_pg_utils-1.0.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ry-pg-utils
-Version: 1.0.8
+Version: 1.0.10
 Summary: Utility functions for PostgreSQL
 Author: Ross Yeager
 Author-email: ryeager12@email.com

ry_pg_utils-1.0.10/VERSION

@@ -0,0 +1 @@
+1.0.10

{ry_pg_utils-1.0.8 → ry_pg_utils-1.0.10}/src/ry_pg_utils/ipc/channels.py

@@ -3,10 +3,9 @@ from ry_redis_bus.channels import Channel
 from ry_pg_utils.pb_types.database_pb2 import DatabaseConfigPb  # pylint: disable=no-name-in-module
 from ry_pg_utils.pb_types.database_pb2 import (  # pylint: disable=no-name-in-module
     DatabaseNotificationPb,
-    DatabaseSettingsPb,
 )
 
 # Channels
 DATABASE_CHANNEL = Channel("DATABASE_CHANNEL", DatabaseConfigPb)
-DATABASE_CONFIG_CHANNEL = Channel("DATABASE_CONFIG_CHANNEL", DatabaseSettingsPb)
+DATABASE_CONFIG_CHANNEL = Channel("DATABASE_CONFIG_CHANNEL", DatabaseConfigPb)
 DATABASE_NOTIFY_CHANNEL = Channel("DATABASE_NOTIFY_CHANNEL", DatabaseNotificationPb)

{ry_pg_utils-1.0.8 → ry_pg_utils-1.0.10}/src/ry_pg_utils/pb_types/database_pb2.py

@@ -14,25 +14,17 @@ _sym_db = _symbol_database.Default()
 from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0e\x64\x61tabase.proto\x12\x08\x64\x61tabase\x1a\x1fgoogle/protobuf/timestamp.proto\"m\n\nPostgresPb\x12\x10\n\x08\x64\x61tabase\x18\x01 \x01(\t\x12\x0c\n\x04user\x18\x02 \x01(\t\x12\x10\n\x08password\x18\x03 \x01(\t\x12\x0c\n\x04host\x18\x04 \x01(\t\x12\x0c\n\x04port\x18\x05 \x01(\x05\x12\x11\n\tbackendId\x18\x06 \x01(\t\"f\n\x11PostgresMessagePb\x12)\n\x05utime\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12&\n\x08postgres\x18\x02 \x01(\x0b\x32\x14.database.PostgresPb\"\xb2\x01\n\x12\x44\x61tabaseSettingsPb\x12\x17\n\x0fprimaryDatabase\x18\x01 \x01(\x08\x12<\n\x08postgres\x18\x02 \x03(\x0b\x32*.database.DatabaseSettingsPb.PostgresEntry\x1a\x45\n\rPostgresEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12#\n\x05value\x18\x02 \x01(\x0b\x32\x14.database.PostgresPb:\x02\x38\x01\"u\n\x10\x44\x61tabaseConfigPb\x12)\n\x05utime\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x36\n\x10\x64\x61tabaseSettings\x18\x02 \x01(\x0b\x32\x1c.database.DatabaseSettingsPb\"\x8e\x01\n\x16\x44\x61tabaseNotificationPb\x12)\n\x05utime\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x12\n\ntable_name\x18\x02 \x01(\t\x12\x14\n\x0c\x63hannel_name\x18\x03 \x01(\t\x12\x0e\n\x06\x61\x63tion\x18\x04 \x01(\t\x12\x0f\n\x07payload\x18\x05 \x01(\tb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0e\x64\x61tabase.proto\x12\x08\x64\x61tabase\x1a\x1fgoogle/protobuf/timestamp.proto\"m\n\nPostgresPb\x12\x10\n\x08\x64\x61tabase\x18\x01 \x01(\t\x12\x0c\n\x04user\x18\x02 \x01(\t\x12\x10\n\x08password\x18\x03 \x01(\t\x12\x0c\n\x04host\x18\x04 \x01(\t\x12\x0c\n\x04port\x18\x05 \x01(\x05\x12\x11\n\tbackendId\x18\x06 \x01(\t\"f\n\x11PostgresMessagePb\x12)\n\x05utime\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12&\n\x08postgres\x18\x02 \x01(\x0b\x32\x14.database.PostgresPb\"\x8e\x01\n\x16\x44\x61tabaseNotificationPb\x12)\n\x05utime\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x12\n\ntable_name\x18\x02 \x01(\t\x12\x14\n\x0c\x63hannel_name\x18\x03 \x01(\t\x12\x0e\n\x06\x61\x63tion\x18\x04 \x01(\t\x12\x0f\n\x07payload\x18\x05 \x01(\tb\x06proto3')
 
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
 _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'database_pb2', globals())
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
-  _DATABASESETTINGSPB_POSTGRESENTRY._options = None
-  _DATABASESETTINGSPB_POSTGRESENTRY._serialized_options = b'8\001'
   _POSTGRESPB._serialized_start=61
   _POSTGRESPB._serialized_end=170
   _POSTGRESMESSAGEPB._serialized_start=172
   _POSTGRESMESSAGEPB._serialized_end=274
-  _DATABASESETTINGSPB._serialized_start=277
-  _DATABASESETTINGSPB._serialized_end=455
-  _DATABASESETTINGSPB_POSTGRESENTRY._serialized_start=386
-  _DATABASESETTINGSPB_POSTGRESENTRY._serialized_end=455
-  _DATABASECONFIGPB._serialized_start=457
-  _DATABASECONFIGPB._serialized_end=574
-  _DATABASENOTIFICATIONPB._serialized_start=577
-  _DATABASENOTIFICATIONPB._serialized_end=719
+  _DATABASENOTIFICATIONPB._serialized_start=277
+  _DATABASENOTIFICATIONPB._serialized_end=419
 # @@protoc_insertion_point(module_scope)

{ry_pg_utils-1.0.8 → ry_pg_utils-1.0.10}/src/ry_pg_utils/pb_types/database_pb2.pyi

@@ -4,9 +4,7 @@ isort:skip_file
 """
 
 import builtins
-import collections.abc
 import google.protobuf.descriptor
-import google.protobuf.internal.containers
 import google.protobuf.message
 import google.protobuf.timestamp_pb2
 import typing
@@ -66,66 +64,6 @@ class PostgresMessagePb(google.protobuf.message.Message):
 
 global___PostgresMessagePb = PostgresMessagePb
 
-@typing.final
-class DatabaseSettingsPb(google.protobuf.message.Message):
-    """Message for DatabaseSettings"""
-
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
-
-    @typing.final
-    class PostgresEntry(google.protobuf.message.Message):
-        DESCRIPTOR: google.protobuf.descriptor.Descriptor
-
-        KEY_FIELD_NUMBER: builtins.int
-        VALUE_FIELD_NUMBER: builtins.int
-        key: builtins.str
-        @property
-        def value(self) -> global___PostgresPb: ...
-        def __init__(
-            self,
-            *,
-            key: builtins.str = ...,
-            value: global___PostgresPb | None = ...,
-        ) -> None: ...
-        def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ...
-        def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ...
-
-    PRIMARYDATABASE_FIELD_NUMBER: builtins.int
-    POSTGRES_FIELD_NUMBER: builtins.int
-    primaryDatabase: builtins.bool
-    @property
-    def postgres(self) -> google.protobuf.internal.containers.MessageMap[builtins.str, global___PostgresPb]: ...
-    def __init__(
-        self,
-        *,
-        primaryDatabase: builtins.bool = ...,
-        postgres: collections.abc.Mapping[builtins.str, global___PostgresPb] | None = ...,
-    ) -> None: ...
-    def ClearField(self, field_name: typing.Literal["postgres", b"postgres", "primaryDatabase", b"primaryDatabase"]) -> None: ...
-
-global___DatabaseSettingsPb = DatabaseSettingsPb
-
-@typing.final
-class DatabaseConfigPb(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
-
-    UTIME_FIELD_NUMBER: builtins.int
-    DATABASESETTINGS_FIELD_NUMBER: builtins.int
-    @property
-    def utime(self) -> google.protobuf.timestamp_pb2.Timestamp: ...
-    @property
-    def databaseSettings(self) -> global___DatabaseSettingsPb: ...
-    def __init__(
-        self,
-        *,
-        utime: google.protobuf.timestamp_pb2.Timestamp | None = ...,
-        databaseSettings: global___DatabaseSettingsPb | None = ...,
-    ) -> None: ...
-    def HasField(self, field_name: typing.Literal["databaseSettings", b"databaseSettings", "utime", b"utime"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing.Literal["databaseSettings", b"databaseSettings", "utime", b"utime"]) -> None: ...
-
-global___DatabaseConfigPb = DatabaseConfigPb
-
 @typing.final
 class DatabaseNotificationPb(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor

{ry_pg_utils-1.0.8 → ry_pg_utils-1.0.10}/src/ry_pg_utils/updater.py

@@ -11,7 +11,6 @@ from ryutils.verbose import Verbose
 from ry_pg_utils.connect import close_engine, init_database, is_database_initialized
 from ry_pg_utils.ipc import channels
 from ry_pg_utils.pb_types.database_pb2 import (  # pylint: disable=no-name-in-module
-    DatabaseSettingsPb,
     PostgresMessagePb,
     PostgresPb,
 )
@@ -21,20 +20,15 @@ DB_RETRY_TIME = 5.0
 
 
 def get_database_settings(
-    message_pb: DatabaseSettingsPb, backend_id: str, verbose: bool = False
+    raw_db_config_pb: PostgresMessagePb | None, verbose: bool = False
 ) -> T.Optional[PostgresInfo]:
-    raw_db_config_pb = message_pb.postgres.get(backend_id, None)
-
     if raw_db_config_pb is None:
         return None
 
-    if not message_pb.primaryDatabase:
-        return None
-
-    db_config_pb: PostgresPb = raw_db_config_pb
+    db_config_pb: PostgresPb = raw_db_config_pb.postgres
 
     if verbose:
-        log.print_normal(f"Received database settings {backend_id}:\n{db_config_pb}")
+        log.print_normal(f"Received database settings:\n{db_config_pb}")
 
     database_settings_msg = PostgresInfo(
         db_name=db_config_pb.database,
@@ -80,12 +74,11 @@ class DbUpdater(RedisClientBase):
         self.last_db_init_retry_time = 0.0
         self.use_local_db_only = args.use_local_db_only if args else True
         self.database_settings_msg: T.Optional[PostgresInfo] = None
-        self.backend_id = backend_id
         self.logging_error_db_callback = logging_error_db_callback
 
     @message_handler
-    def handle_database_config_message(self, message_pb: DatabaseSettingsPb) -> None:
-        database_settings = get_database_settings(message_pb, backend_id=self.backend_id)
+    def handle_database_config_message(self, message_pb: PostgresMessagePb) -> None:
+        database_settings = get_database_settings(message_pb)
 
         if database_settings is not None:
             self.database_settings_msg = database_settings

{ry_pg_utils-1.0.8 → ry_pg_utils-1.0.10/src/ry_pg_utils.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ry-pg-utils
-Version: 1.0.8
+Version: 1.0.10
 Summary: Utility functions for PostgreSQL
 Author: Ross Yeager
 Author-email: ryeager12@email.com

ry_pg_utils-1.0.8/VERSION

@@ -1 +0,0 @@
-1.0.8