def execute_and_map(self, ddlstmt): "Execute a DDL statement, commit, and return a map of the database" self.execute(ddlstmt) self.conn.commit() db = Database(DbConnection(self.name, self.user, host=self.host, port=self.port)) return db.to_map()
def main(host='localhost', port=5432, schema=None): """Convert database table specifications to YAML.""" parser = OptionParser("usage: %prog [options] dbname") parser.add_option('-H', '--host', dest='host', help="database server host or socket directory " "(default %default)") parser.add_option('-p', '--port', dest='port', type='int', help="database server port (default %default)") parser.add_option('-U', '--username', dest='username', help="database user name (default %default)") parser.add_option('-n', '--schema', dest='schema', help="only for named schema (default %default)") parser.add_option('-t', '--table', dest='tablist', action='append', help="only for named tables (default all)") parser.set_defaults(host=host, port=port, username=os.getenv("USER"), schema=schema) (options, args) = parser.parse_args() if len(args) > 1: parser.error("too many arguments") elif len(args) != 1: parser.error("database name not specified") dbname = args[0] db = Database(DbConnection(dbname, options.username, options.host, options.port)) dbmap = db.to_map() # trim the map of schemas/tables not selected if options.schema: skey = 'schema ' + options.schema for sch in dbmap.keys(): if 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] print yaml.dump(dbmap, default_flow_style=False)
def main(schema=None): """Convert database table specifications to YAML.""" parser = cmd_parser("Extract the schema of a PostgreSQL database in " "YAML format", __version__) parser.add_argument('-m', '--multiple-files', action='store_true', help='output to multiple files (metadata directory)') parser.add_argument('-O', '--no-owner', action='store_true', help='exclude object ownership information') parser.add_argument('-x', '--no-privileges', action='store_true', dest='no_privs', help='exclude privilege (GRANT/REVOKE) information') group = parser.add_argument_group("Object inclusion/exclusion options", "(each can be given multiple times)") group.add_argument('-n', '--schema', metavar='SCHEMA', dest='schemas', action='append', default=[], help="extract the named schema(s) (default all)") group.add_argument('-N', '--exclude-schema', metavar='SCHEMA', dest='excl_schemas', action='append', default=[], help="do NOT extract the named schema(s) " "(default none)") group.add_argument('-t', '--table', metavar='TABLE', dest='tables', action='append', default=[], help="extract the named table(s) (default all)") group.add_argument('-T', '--exclude-table', metavar='TABLE', dest='excl_tables', action='append', default=[], help="do NOT extract the named table(s) " "(default none)") parser.set_defaults(schema=schema) cfg = parse_args(parser) output = cfg['files']['output'] options = cfg['options'] if options.multiple_files and output: parser.error("Cannot specify both --multiple-files and --output") db = Database(cfg) dbmap = db.to_map() if not options.multiple_files: print(yamldump(dbmap), file=output or sys.stdout) if output: output.close()
def main(host='localhost', port=5432, schema=None): """Convert database table specifications to YAML.""" parser = ArgumentParser(parents=[parent_parser()], description="Extract the schema of a PostgreSQL " "database in YAML format") parser.add_argument('-n', '--schema', help="only for named schema (default %(default)s)") parser.add_argument('-t', '--table', dest='tablist', action='append', help="only for named tables (default all)") parser.set_defaults(host=host, port=port, username=os.getenv("USER"), schema=schema) args = parser.parse_args() db = Database(DbConnection(args.dbname, args.username, args.password, args.host, args.port)) dbmap = db.to_map() # trim the map of schemas/tables not selected if args.schema: skey = 'schema ' + args.schema for sch in dbmap.keys(): if sch[:7] == 'schema ' and sch != skey: del dbmap[sch] if args.tablist: ktablist = ['table ' + tbl for tbl in args.tablist] for sch in dbmap.keys(): 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] if args.output: fd = args.output sys.stdout = fd print(yaml.dump(dbmap, default_flow_style=False)) if args.output: fd.close()
def main(host='localhost', port=5432, schema=None): """Convert database table specifications to YAML.""" parser = ArgumentParser(parents=[parent_parser()], description="Extract the schema of a PostgreSQL " "database in YAML format") parser.add_argument('-n', '--schema', help="only for named schema (default %(default)s)") parser.add_argument('-t', '--table', dest='tablist', action='append', help="only for named tables (default all)") parser.set_defaults(host=host, port=port, username=os.getenv("USER"), schema=schema) args = parser.parse_args() pswd = (args.password and getpass.getpass() or '') db = Database(args.dbname, args.username, pswd, args.host, args.port) dbmap = db.to_map([args.schema], args.tablist) if args.output: fd = args.output sys.stdout = fd print(yaml.dump(dbmap, default_flow_style=False)) if args.output: fd.close()
def main(host='localhost', port=5432, schema=None): """Convert database table specifications to YAML.""" parser = ArgumentParser(parents=[parent_parser()], description="Extract the schema of a PostgreSQL " "database in YAML format") group = parser.add_argument_group("Object inclusion/exclusion options", "(each can be given multiple times)") group.add_argument('-n', '--schema', metavar='SCHEMA', dest='schemas', action='append', default=[], help="extract the named schema(s) (default all)") group.add_argument('-N', '--exclude-schema', metavar='SCHEMA', dest='excl_schemas', action='append', default=[], help="do NOT extract the named schema(s) " "(default none)") group.add_argument('-t', '--table', metavar='TABLE', dest='tables', action='append', default=[], help="extract the named table(s) (default all)") group.add_argument('-T', '--exclude-table', metavar='TABLE', dest='excl_tables', action='append', default=[], help="do NOT extract the named table(s) " "(default none)") parser.set_defaults(host=host, port=port, schema=schema, username=os.getenv("PGUSER") or os.getenv("USER")) args = parser.parse_args() pswd = (args.password and getpass.getpass() or None) db = Database(args.dbname, args.username, pswd, args.host, args.port) dbmap = db.to_map(schemas=args.schemas, tables=args.tables, exclude_schemas=args.excl_schemas, exclude_tables=args.excl_tables) print(yaml.dump(dbmap, default_flow_style=False), file=args.output or sys.stdout) if args.output: args.output.close()
def main(schema=None): """Convert database table specifications to YAML.""" parser = cmd_parser( "Extract the schema of a PostgreSQL database in " "YAML format", __version__) parser.add_argument('-m', '--multiple-files', action='store_true', help='output to multiple files (metadata directory)') parser.add_argument('-O', '--no-owner', action='store_true', help='exclude object ownership information') parser.add_argument('-x', '--no-privileges', action='store_true', dest='no_privs', help='exclude privilege (GRANT/REVOKE) information') group = parser.add_argument_group("Object inclusion/exclusion options", "(each can be given multiple times)") group.add_argument('-n', '--schema', metavar='SCHEMA', dest='schemas', action='append', default=[], help="extract the named schema(s) (default all)") group.add_argument('-N', '--exclude-schema', metavar='SCHEMA', dest='excl_schemas', action='append', default=[], help="do NOT extract the named schema(s) " "(default none)") group.add_argument('-t', '--table', metavar='TABLE', dest='tables', action='append', default=[], help="extract the named table(s) (default all)") group.add_argument('-T', '--exclude-table', metavar='TABLE', dest='excl_tables', action='append', default=[], help="do NOT extract the named table(s) " "(default none)") parser.set_defaults(schema=schema) cfg = parse_args(parser) output = cfg['files']['output'] options = cfg['options'] if options.multiple_files and output: parser.error("Cannot specify both --multiple-files and --output") db = Database(cfg) dbmap = db.to_map() if not options.multiple_files: print(yamldump(dbmap), file=output or sys.stdout) if output: output.close()