Add virtual host to connection

Add virtual host to connection

Author: Arstotzk

b_rabbit/b_rabbit.py

@@ -43,15 +43,19 @@ def __init__(
             port: int = 5672,
             user: str = None,
             password: str = None,
+            virtual_host: str = None,
         ):
 
             """
             Wrapper class to store the connection to server globally.
             :param str host: Hostname of RabbitMQ Server
             :param int port: Port of RabbitMQ Server
             """
-
-            if user != None:
+            if virtual_host != None:
+                self.connection = rabbitpy.Connection(
+                    "amqp://{}:{}@{}:{}/{}".format(user, password, host, str(port), virtual_host)
+                )
+            elif user != None:
                 self.connection = rabbitpy.Connection(
                     "amqp://{}:{}@{}:{}/".format(user, password, host, str(port))
                 )