Merge pull request #17 from stalkerg/small_fixes

minor fixes (by @stalkerg)

Author: funbringer

.travis.yml

@@ -12,7 +12,7 @@ install:
    - pip install flake8
 
 script:
-   - flake8 --ignore=W191,F401,E501 .
+   - flake8 --ignore=W191,F401,E501,F403 .
 
 notifications:
   email:

testgres/__init__.py

@@ -1 +1 @@
-from .testgres import PostgresNode, get_new_node, clean_all, stop_all
+from .testgres import *

testgres/testgres.py

@@ -179,7 +179,7 @@ def get_bin_path(self, filename):
         else:
             return os.path.join(pg_config.get("BINDIR"), filename)
 
-    def init(self, allows_streaming=False):
+    def init(self, allows_streaming=False, initdb_params=[]):
         """ Performs initdb """
 
         postgres_conf = os.path.join(self.data_dir, "postgresql.conf")
@@ -195,7 +195,7 @@ def init(self, allows_streaming=False):
         initdb = self.get_bin_path("initdb")
         with open(os.path.join(self.logs_dir, "initdb.log"), "a") as file_out:
             ret = subprocess.call(
-                [initdb, self.data_dir, "-N"],
+                [initdb, self.data_dir, "-N"] + initdb_params,
                 stdout=file_out,
                 stderr=subprocess.STDOUT
             )