Paweł Domański
79 lines
1.0 KiB
Bash
79 lines
1.0 KiB
Bash
#!/bin/bash
|
|
|
|
DBNAME="postgres"
|
|
USER="-U postgres"
|
|
PORT="-p 5432"
|
|
|
|
usage() {
|
|
echo "Usage: $0 [-d <dbname> -h <host> -U <user> -p <port> u]"
|
|
exit 1
|
|
}
|
|
|
|
while getopts "d:h:p:uU:" OPT;
|
|
do case "${OPT}" in
|
|
d) DBNAME=$OPTARG
|
|
;;
|
|
h) HOST="-h $OPTARG"
|
|
;;
|
|
p) PORT="-p $OPTARG"
|
|
;;
|
|
U) USER="-U $OPTARG"
|
|
;;
|
|
u) usage
|
|
;;
|
|
[?]) usage
|
|
esac;
|
|
done
|
|
|
|
|
|
allow_logins()
|
|
{
|
|
|
|
CNT=$(psql $HOST $PORT $USER $DBNAME <<_QUERY_
|
|
|
|
UPDATE pg_authid au
|
|
SET rolcanlogin = TRUE
|
|
FROM pg_lockout lo
|
|
WHERE au.rolname = lo.rolname;
|
|
|
|
_QUERY_
|
|
|
|
)
|
|
|
|
}
|
|
|
|
current_user()
|
|
{
|
|
|
|
CUSER=$(psql -t $HOST $PORT $USER $DBNAME <<_QUERY_
|
|
|
|
SELECT TRIM(BOTH FROM current_user);
|
|
|
|
_QUERY_
|
|
|
|
)
|
|
|
|
}
|
|
|
|
|
|
|
|
###############################################################
|
|
# MAIN #
|
|
###############################################################
|
|
|
|
current_user
|
|
|
|
if [ $CUSER != "postgres" ]
|
|
then
|
|
echo "You must be postgres to execute this script."
|
|
echo "Aborting."
|
|
exit 1
|
|
fi
|
|
|
|
allow_logins
|
|
|
|
echo "Users allowed to login = $CNT"
|
|
|
|
exit 0
|
|
|