Support specifying postgresql drivers (#156)

* Support specifying postgresql drivers

* Add "postgresql-pg8000" extra dep

* Update setup.py

Co-authored-by: Till Hoffmann <[email protected]>

* Update postgres.py

* Update postgres.py

* fix whitespace

Co-authored-by: Till Hoffmann <[email protected]>

Author: KarolisL

testcontainers/postgres.py

@@ -32,12 +32,18 @@ class PostgresContainer(DbContainer):
     POSTGRES_PASSWORD = os.environ.get("POSTGRES_PASSWORD", "test")
     POSTGRES_DB = os.environ.get("POSTGRES_DB", "test")
 
-    def __init__(self, image="postgres:latest", port=5432, user=None, password=None, dbname=None):
+    def __init__(self,
+                 image="postgres:latest",
+                 port=5432, user=None,
+                 password=None,
+                 dbname=None,
+                 driver="psycopg2"):
         super(PostgresContainer, self).__init__(image=image)
         self.POSTGRES_USER = user or self.POSTGRES_USER
         self.POSTGRES_PASSWORD = password or self.POSTGRES_PASSWORD
         self.POSTGRES_DB = dbname or self.POSTGRES_DB
         self.port_to_expose = port
+        self.driver = driver
 
         self.with_exposed_ports(self.port_to_expose)
 
@@ -47,7 +53,7 @@ def _configure(self):
         self.with_env("POSTGRES_DB", self.POSTGRES_DB)
 
     def get_connection_url(self, host=None):
-        return super()._create_connection_url(dialect="postgresql+psycopg2",
+        return super()._create_connection_url(dialect="postgresql+{}".format(self.driver),
                                               username=self.POSTGRES_USER,
                                               password=self.POSTGRES_PASSWORD,
                                               db_name=self.POSTGRES_DB,