1
0
mirror of https://github.com/dbcli/pgcli synced 2024-05-31 01:17:54 +00:00

Make user arg optional.

This commit is contained in:
Amjith Ramanujam 2015-01-14 10:07:35 -08:00
parent 8333603b63
commit ac93600aae

View File

@ -241,15 +241,15 @@ class PGCli(object):
default=False, help='Never prompt for password.') default=False, help='Never prompt for password.')
@click.option('-d', '--dbname', default='', envvar='PGDATABASE', @click.option('-d', '--dbname', default='', envvar='PGDATABASE',
help='database name to connect to.') help='database name to connect to.')
@click.argument('database', default='', envvar='PGDATABASE', nargs=1) @click.argument('database', default=lambda: None, envvar='PGDATABASE', nargs=1)
@click.argument('username', default='', envvar='PGUSER', nargs=1) @click.argument('username', default=lambda: None, envvar='PGUSER', nargs=1)
def cli(database, user, host, port, prompt_passwd, never_prompt, dbname, def cli(database, user, host, port, prompt_passwd, never_prompt, dbname,
username): username):
pgcli = PGCli(prompt_passwd, never_prompt) pgcli = PGCli(prompt_passwd, never_prompt)
# Choose which ever one has a valid database name. # Choose which ever one has a valid value.
database = database or dbname database = database or dbname
user = user or username user = username or user
if '://' in database: if '://' in database:
pgcli.connect_uri(database) pgcli.connect_uri(database)