Commits

lakin.wecker  committed 84b659d

adding a new PgLoadPlain command.

  • Participants
  • Parent commits 5f95183

Comments (0)

Files changed (1)

File baste/__init__.py

 #-------------------------------------------------------------------------------
 class PgRestore(object):
     #---------------------------------------------------------------------------
+    def __init__(self, file, db, user, format="custom"):
+        self.file = file
+        self.db = db
+        self.user = user
+        self.format = format
+
+    #---------------------------------------------------------------------------
+    def __call__(self):
+        """
+        Uses the pg_restore command to restore the database from the given file.
+        """
+        local(
+            "pg_restore --clean --no-owner --no-privileges --format=%s --host=localhost --username=%s --dbname=%s %s" % (
+                self.format, self.user, self.db, self.file
+            )
+        )
+
+#-------------------------------------------------------------------------------
+class PgLoadPlain(object):
+    #---------------------------------------------------------------------------
     def __init__(self, file, db, user):
         self.file = file
         self.db = db
 
     #---------------------------------------------------------------------------
     def __call__(self):
-        """
-        Uses the pg_restore command to restore the database from the given file.
-        """
-        local("pg_restore --clean --no-owner --no-privileges --format=custom --host=localhost --username=%s --dbname=%s %s" % (self.user, self.db, self.file))
+        """Uses psql to load a plain dump format"""
+        local(
+            "bzcat %s | psql --host=localhost --username=%s --dbname=%s" % (
+                self.file, self.user, self.db
+            )
+        )
 
 #-------------------------------------------------------------------------------
 class PgShell(object):