如何检查postgres用户是否存在？

Question

createuser 允许在 PostgreSQL 中创建用户 (ROLE)。有没有一种简单的方法来检查该用户（名称）是否已经存在？否则 createuser 返回并显示错误：

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

更新：该解决方案最好可以从 shell 执行，以便更容易在脚本内实现自动化。

Answer 1

SELECT 1 FROM pg_roles WHERE rolname='USR_NAME'

就命令行而言（感谢 Erwin）：

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

如果找到则返回 1，除此之外别无其他。

那是：

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

Answer 2

遵循相同的想法检查数据库是否存在

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

并且你可以在这样的脚本中使用它：

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

Answer 3

要完全在单个 psql 命令中完成此操作：

DO $BEGIN
IF NOT EXISTS (SELECT FROM pg_catalog.pg_roles WHERE rolname = 'USR_NAME')
THEN CREATE ROLE USR_NAME;
END IF;
END$;

Answer 4

psql -qtA -c "\du USR_NAME" | psql -qtA -c "\du USR_NAME" |剪切-d“|” -f 1

[[ -n $(psql -qtA -c "\du ${1}" | cut -d "|" -f 1) ]] &&回显“存在”|| echo“不存在”

Answer 5

希望这对那些可能在 python 中执行此操作的人有所帮助。
我在 GitHubGist 上创建了一个完整的工作脚本/解决方案 - 请参阅此代码片段下面的 URL。

# 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))

提供幂等远程（RDS）PostgreSQL从python创建角色/用户，无需CM模块等

Answer 6

受到接受的答案的启发，不幸的是这对我不起作用（直接psql调用时出错） ，然后我做了这样的事情：

if ! echo "SELECT * FROM pg_roles WHERE rolname='USR_NAME'" | psql -h localhost -U postgres | grep -E "[1-9][0-9]* rows"; then
  # User not found, create it
  if ! echo "CREATE USER USR_NAME WITH PASSWORD 'USR_NAME' CREATEDB SUPERUSER" | psql -h localhost -U postgres; then
    echo "Error creating USR_NAME"
    exit 1
  fi
fi

即使我认为 grep -E "1 rows" 在这里是安全的，因为我们不应该有超过一个同名的用户，我更喜欢保留 grep -乙“[1-9][0-9]* rows” 以获得返回成功的通用“我得到 1 个或多个结果”。
如果失败，我会输入 exit 1 ，因为我所在的脚本需要创建此用户才能正常运行。

Answer 7

PHP 的完整片段：

• 此外，最好在列名“rolname”前面提及系统目录/表名“pg_roles”。像这样：“pg_roles.rolname”。

• 并提及它，因为它让我困惑：它实际上拼写为“rolname”而不是“rolename”。

// PHP Version 8.1.13
// PostgreSQL 15

try {
    // database connection
    $pdo = new PDO("pgsql:host=$host;port=$port;dbname=$dbname", $dbusername, $dbuserkeyword, [PDO::ATTR_ERRMODE => PDO::ERRMODE_EXCEPTION]);

    if ($pdo) { // went fine

        // a query to ask if user available in table (system catalog) "pg_roles" and column "rolname"
        $myResult = $pdo -> query("SELECT * FROM pg_roles WHERE pg_roles.rolname='{$theuseriamlookingfor}'");

        // retrieve the values from response in an Array
        $myResult = $myResult -> fetchAll();

        // output the first value from that Array
        echo $myResult[0]["rolname"];
    } 

} catch (PDOException $e) {
    // output the error-message
    echo $e->getMessage();
}