web-dev-qa-db-fra.com

Comment vérifier si un utilisateur postgres existe?

createuser permet la création d'un utilisateur (ROLE) dans PostgreSQL. Existe-t-il un moyen simple de vérifier si cet utilisateur (nom) existe déjà? Sinon, createuser renvoie une erreur:

createuser: creation of new role failed: ERROR:  role "USR_NAME" already exists

MISE À JOUR: La solution doit être de préférence exécutable à partir de Shell, afin de faciliter l'automatisation à l'intérieur d'un script.

74
m33lky
SELECT 1 FROM pg_roles WHERE rolname='USR_NAME'

Et en termes de ligne de commande (grâce à Erwin):

psql postgres -tAc "SELECT 1 FROM pg_roles WHERE rolname='USR_NAME'"

Donne 1 si trouvé et rien d'autre.

C'est:

psql postgres -tAc "SELECT 1 FROM pg_roles WHERE rolname='USR_NAME'" | grep -q 1 || createuser ...
129

Suivant la même idée que de vérifier si une base de données existe

psql -t -c '\du' | cut -d \| -f 1 | grep -qw <user_to_check>

et vous pouvez l'utiliser dans un script comme celui-ci:

if psql -t -c '\du' | cut -d \| -f 1 | grep -qw <user_to_check>; then
    # user exists
    # $? is 0
else
    # ruh-roh
    # $? is 1
fi
4
matt

J'espère que cela aidera ceux d'entre vous qui pourraient le faire dans python.
J'ai créé un script/une solution de travail complet sur un GitHubGist - voir l’URL au-dessous de cet extrait de code.

# ref: https://stackoverflow.com/questions/8546759/how-to-check-if-a-postgres-user-exists
check_user_cmd = ("SELECT 1 FROM pg_roles WHERE rolname='%s'" % (deis_app_user))

# our create role/user command and vars
create_user_cmd = ("CREATE ROLE %s WITH LOGIN CREATEDB PASSWORD '%s'" % (deis_app_user, deis_app_passwd))

# ref: https://stackoverflow.com/questions/37488175/simplify-database-psycopg2-usage-by-creating-a-module
class RdsCreds():
    def __init__(self):
        self.conn = psycopg2.connect("dbname=%s user=%s Host=%s password=%s" % (admin_db_name, admin_db_user, db_Host, admin_db_pass))
        self.conn.set_isolation_level(0)
        self.cur = self.conn.cursor()

    def query(self, query):
        self.cur.execute(query)
        return self.cur.rowcount > 0

    def close(self):
        self.cur.close()
        self.conn.close()

db = RdsCreds()
user_exists = db.query(check_user_cmd)

# PostgreSQL currently has no 'create role if not exists'
# So, we only want to create the role/user if not exists 
if (user_exists) is True:
    print("%s user_exists: %s" % (deis_app_user, user_exists))
    print("Idempotent: No credential modifications required. Exiting...")
    db.close()
else:
    print("%s user_exists: %s" % (deis_app_user, user_exists))
    print("Creating %s user now" % (deis_app_user))
    db.query(create_user_cmd)
    user_exists = db.query(check_user_cmd)
    db.close()
    print("%s user_exists: %s" % (deis_app_user, user_exists))

Fournit un rôle/utilisateur à distance idempotent (RDS) pour créer un rôle/un utilisateur à partir de python sans modules CM, etc.

2
cmcc