When restricting dbtoyaml to schemas/tables, don't exclude non-schema…

… objects.
perseas · Dec 14, 2011 · f03810b · f03810b
1 parent 5f51dae
commit f03810b
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/pyrseas/dbtoyaml.py b/pyrseas/dbtoyaml.py
@@ -42,16 +42,17 @@ def main(host='localhost', port=5432, schema=None):
     if options.schema:
         skey = 'schema ' + options.schema
         for sch in dbmap.keys():
-            if sch != skey:
+            if sch[:7] == 'schema ' and sch != skey:
                 del dbmap[sch]
     if options.tablist:
         ktablist = ['table ' + tbl for tbl in options.tablist]
         for sch in dbmap.keys():
-            for tbl in dbmap[sch].keys():
-                if tbl not in ktablist:
-                    del dbmap[sch][tbl]
-            if not dbmap[sch]:
-                del dbmap[sch]
+            if sch[:7] == 'schema ':
+                for tbl in dbmap[sch].keys():
+                    if tbl not in ktablist:
+                        del dbmap[sch][tbl]
+                if not dbmap[sch]:
+                    del dbmap[sch]
 
     print yaml.dump(dbmap, default_flow_style=False)