move connection parameters in environment class

Author: DanielePalaia

README.md

@@ -43,8 +43,8 @@ A connection to the RabbitMQ AMQP 1.0 server can be established using the Enviro
 For example:
 
 ```python
-    environment = Environment()
-    connection = environment.connection("amqp://guest:guest@localhost:5672/")
+    environment = Environment("amqp://guest:guest@localhost:5672/")
+    connection = environment.connection()
     connection.dial()
 ```
 

examples/getting_started/getting_started.py

@@ -63,7 +63,7 @@ def on_link_closed(self, event: Event) -> None:
 
 
 def create_connection(environment: Environment) -> Connection:
-    connection = environment.connection("amqp://guest:guest@localhost:5672/")
+    connection = environment.connection()
     # in case of SSL enablement
     # ca_cert_file = ".ci/certs/ca_certificate.pem"
     # client_cert = ".ci/certs/client_certificate.pem"
@@ -87,7 +87,7 @@ def main() -> None:
     routing_key = "routing-key"
 
     print("connection to amqp server")
-    environment = Environment()
+    environment = Environment(uri="amqp://guest:guest@localhost:5672/")
     connection = create_connection(environment)
 
     management = connection.management()

examples/reconnection/reconnection_example.py

@@ -21,9 +21,6 @@
     QuorumQueueSpecification,
 )
 
-environment = Environment()
-
-
 # here we keep track of the objects we need to reconnect
 @dataclass
 class ConnectionConfiguration:
@@ -42,6 +39,7 @@ class ConnectionConfiguration:
 def on_disconnection():
 
     print("disconnected")
+    global environment
     exchange_name = "test-exchange"
     queue_name = "example-queue"
     routing_key = "routing-key"
@@ -69,6 +67,8 @@ def on_disconnection():
         )
 
 
+environment = Environment(uri="amqp://guest:guest@localhost:5672/", on_disconnection_handler=on_disconnection)
+
 class MyMessageHandler(AMQPMessagingHandler):
 
     def __init__(self):
@@ -122,10 +122,7 @@ def create_connection() -> Connection:
     # ]
     # connection = Connection(uris=uris, on_disconnection_handler=on_disconnected)
 
-    connection = environment.connection(
-        uri="amqp://guest:guest@localhost:5672/",
-        on_disconnection_handler=on_disconnection,
-    )
+    connection = environment.connection()
     connection.dial()
 
     return connection

examples/streams/example_with_streams.py

@@ -67,7 +67,7 @@ def on_link_closed(self, event: Event) -> None:
 
 
 def create_connection(environment: Environment) -> Connection:
-    connection = environment.connection("amqp://guest:guest@localhost:5672/")
+    connection = environment.connection()
     # in case of SSL enablement
     # ca_cert_file = ".ci/certs/ca_certificate.pem"
     # client_cert = ".ci/certs/client_certificate.pem"
@@ -88,7 +88,7 @@ def main() -> None:
     queue_name = "example-queue"
 
     print("connection to amqp server")
-    environment = Environment()
+    environment = Environment("amqp://guest:guest@localhost:5672/")
     connection = create_connection(environment)
 
     management = connection.management()

examples/tls/tls_example.py

@@ -65,16 +65,7 @@ def on_link_closed(self, event: Event) -> None:
 
 def create_connection(environment: Environment) -> Connection:
     # in case of SSL enablement
-    ca_cert_file = ".ci/certs/ca_certificate.pem"
-    client_cert = ".ci/certs/client_certificate.pem"
-    client_key = ".ci/certs/client_key.pem"
-    connection = environment.connection(
-        "amqps://guest:guest@localhost:5671/",
-        ssl_context=SslConfigurationContext(
-            ca_cert=ca_cert_file,
-            client_cert=ClientCert(client_cert=client_cert, client_key=client_key),
-        ),
-    )
+    connection = environment.connection()
     connection.dial()
 
     return connection
@@ -85,7 +76,17 @@ def main() -> None:
     exchange_name = "test-exchange"
     queue_name = "example-queue"
     routing_key = "routing-key"
-    environment = Environment()
+    ca_cert_file = ".ci/certs/ca_certificate.pem"
+    client_cert = ".ci/certs/client_certificate.pem"
+    client_key = ".ci/certs/client_key.pem"
+
+    environment = Environment(
+        "amqps://guest:guest@localhost:5671/",
+        ssl_context=SslConfigurationContext(
+            ca_cert=ca_cert_file,
+            client_cert=ClientCert(client_cert=client_cert, client_key=client_key),
+        ),
+    )
 
     print("connection to amqp server")
     connection = create_connection(environment)

rabbitmq_amqp_python_client/environment.py

@@ -23,46 +23,52 @@ class Environment:
         _connections (list[Connection]): List of active connections managed by this environment
     """
 
-    def __init__(self):  # type: ignore
+    def __init__(
+        self,  # single-node mode
+        uri: Optional[str] = None,
+        # multi-node mode
+        uris: Optional[list[str]] = None,
+        ssl_context: Optional[SslConfigurationContext] = None,
+        on_disconnection_handler: Optional[CB] = None,  # type: ignore
+    ):
         """
         Initialize a new Environment instance.
 
         Creates an empty list to track active connections.
+
+        Args:
+            uri: Single node connection URI
+            uris: List of URIs for multi-node setup
+            ssl_context: SSL configuration for secure connections
+            on_disconnection_handler: Callback for handling disconnection events
+
         """
+        self._uri = uri
+        self._uris = uris
+        self._ssl_context = ssl_context
+        self._on_disconnection_handler = on_disconnection_handler
         self._connections: list[Connection] = []
 
     def connection(
         self,
-        # single-node mode
-        uri: Optional[str] = None,
-        # multi-node mode
-        uris: Optional[list[str]] = None,
-        ssl_context: Optional[SslConfigurationContext] = None,
-        on_disconnection_handler: Optional[CB] = None,  # type: ignore
     ) -> Connection:
         """
         Create and return a new connection.
 
         This method supports both single-node and multi-node configurations, with optional
         SSL/TLS security and disconnection handling.
 
-        Args:
-            uri: Single node connection URI
-            uris: List of URIs for multi-node setup
-            ssl_context: SSL configuration for secure connections
-            on_disconnection_handler: Callback for handling disconnection events
-
         Returns:
             Connection: A new connection instance
 
         Raises:
             ValueError: If neither uri nor uris is provided
         """
         connection = Connection(
-            uri=uri,
-            uris=uris,
-            ssl_context=ssl_context,
-            on_disconnection_handler=on_disconnection_handler,
+            uri=self._uri,
+            uris=self._uris,
+            ssl_context=self._ssl_context,
+            on_disconnection_handler=self._on_disconnection_handler,
         )
         logger.debug("Environment: Creating and returning a new connection")
         self._connections.append(connection)

tests/conftest.py

@@ -15,7 +15,7 @@
 
 @pytest.fixture()
 def environment(pytestconfig):
-    environment = Environment()
+    environment = Environment(uri="amqp://guest:guest@localhost:5672/")
     try:
         yield environment
 
@@ -25,8 +25,8 @@ def environment(pytestconfig):
 
 @pytest.fixture()
 def connection(pytestconfig):
-    environment = Environment()
-    connection = environment.connection("amqp://guest:guest@localhost:5672/")
+    environment = Environment(uri="amqp://guest:guest@localhost:5672/")
+    connection = environment.connection()
     connection.dial()
     try:
         yield connection
@@ -37,17 +37,18 @@ def connection(pytestconfig):
 
 @pytest.fixture()
 def connection_ssl(pytestconfig):
-    environment = Environment()
     ca_cert_file = ".ci/certs/ca_certificate.pem"
     client_cert = ".ci/certs/client_certificate.pem"
     client_key = ".ci/certs/client_key.pem"
-    connection = environment.connection(
+
+    environment = Environment(
         "amqps://guest:guest@localhost:5671/",
         ssl_context=SslConfigurationContext(
             ca_cert=ca_cert_file,
             client_cert=ClientCert(client_cert=client_cert, client_key=client_key),
         ),
     )
+    connection = environment.connection()
     connection.dial()
     try:
         yield connection
@@ -58,8 +59,8 @@ def connection_ssl(pytestconfig):
 
 @pytest.fixture()
 def management(pytestconfig):
-    environment = Environment()
-    connection = environment.connection("amqp://guest:guest@localhost:5672/")
+    environment = Environment(uri="amqp://guest:guest@localhost:5672/")
+    connection = environment.connection()
     connection.dial()
     try:
         management = connection.management()
@@ -71,8 +72,8 @@ def management(pytestconfig):
 
 @pytest.fixture()
 def consumer(pytestconfig):
-    environment = Environment()
-    connection = environment.connection("amqp://guest:guest@localhost:5672/")
+    environment = Environment(uri="amqp://guest:guest@localhost:5672/")
+    connection = environment.connection()
     connection.dial()
     try:
         queue_name = "test-queue"

tests/test_connection.py

@@ -2,7 +2,6 @@
 
 from rabbitmq_amqp_python_client import (
     ClientCert,
-    Connection,
     ConnectionClosed,
     Environment,
     SslConfigurationContext,
@@ -20,43 +19,45 @@ def on_disconnected():
 
 
 def test_connection() -> None:
-    environment = Environment()
-    connection = environment.connection("amqp://guest:guest@localhost:5672/")
+    environment = Environment(uri="amqp://guest:guest@localhost:5672/")
+    connection = environment.connection()
     connection.dial()
     environment.close()
 
 
 def test_environment_context_manager() -> None:
-    with Environment() as environment:
-        connection = environment.connection("amqp://guest:guest@localhost:5672/")
+    with Environment(uri="amqp://guest:guest@localhost:5672/") as environment:
+        connection = environment.connection()
         connection.dial()
 
 
 def test_connection_ssl() -> None:
-    environment = Environment()
     ca_cert_file = ".ci/certs/ca_certificate.pem"
     client_cert = ".ci/certs/client_certificate.pem"
     client_key = ".ci/certs/client_key.pem"
-    connection = environment.connection(
+
+    environment = Environment(
         "amqps://guest:guest@localhost:5671/",
         ssl_context=SslConfigurationContext(
             ca_cert=ca_cert_file,
             client_cert=ClientCert(client_cert=client_cert, client_key=client_key),
         ),
     )
+
+    connection = environment.connection()
     connection.dial()
 
     environment.close()
 
 
 def test_environment_connections_management() -> None:
 
-    environment = Environment()
-    connection = environment.connection("amqp://guest:guest@localhost:5672/")
+    environment = Environment(uri="amqp://guest:guest@localhost:5672/")
+    connection = environment.connection()
     connection.dial()
-    connection2 = environment.connection("amqp://guest:guest@localhost:5672/")
+    connection2 = environment.connection()
     connection2.dial()
-    connection3 = environment.connection("amqp://guest:guest@localhost:5672/")
+    connection3 = environment.connection()
     connection3.dial()
 
     assert environment.active_connections == 3
@@ -89,17 +90,17 @@ def on_disconnected():
 
         # reconnect
         if connection is not None:
-            connection = Connection("amqp://guest:guest@localhost:5672/")
+            connection = environment.connection()
             connection.dial()
 
         nonlocal reconnected
         reconnected = True
 
-    environment = Environment()
-
-    connection = environment.connection(
+    environment = Environment(
         "amqp://guest:guest@localhost:5672/", on_disconnection_handler=on_disconnected
     )
+
+    connection = environment.connection()
     connection.dial()
 
     # delay