djrobstep/migra
1
1
Fork 0
mirror of https://github.com/djrobstep/migra.git synced 2024-09-11 13:56:05 +03:00
Code Issues Projects Releases Wiki Activity

Add a --with-privileges flag to migra command and pass it to add_all_changes

Browse Source
This commit is contained in:
Alvaro Gutierrez Perez 2018-07-09 14:52:03 +02:00
parent 885161218f
commit ab6a9d6b6c
No known key found for this signature in database
GPG Key ID: 5186F8446CE87F52
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
migra/command.py
View File

@ -32,14 +32,21 @@ def parse_args(args):
dest='schema',
default=None,
help='Restrict output to statements for a particular schema',
),
)
parser.add_argument(
'--create-extensions-only',
dest='create_extensions_only',
action='store_true',
default=False,
help='Only output "create extension..." statements, nothing else.',
),
)
parser.add_argument(
'--with-privileges',
dest='with_privileges',
action='store_true',
default=False,
help='Also output privilege differences (ie. grant/revoke statements)',
)
parser.add_argument('dburl_from', help='The database you want to migrate.')
parser.add_argument(
'dburl_target', help='The database you want to use as the target.'
@ -60,7 +67,7 @@ def run(args, out=None, err=None):
if args.create_extensions_only:
m.add_extension_changes(drops=False)
else:
m.add_all_changes()
m.add_all_changes(privileges=args.with_privileges)
try:
if m.statements:
print(m.sql, file=out)