django - 如何从 bash 测试 postgresql 凭据？

Question

我有一个从虚拟环境运行的 Django 应用程序。我正在编写一个 bash 脚本来测试虚拟环境。postgresql 用户名和密码是虚拟环境中的环境变量，Django 将其拉入settings.py。

为了测试与数据库的连接，我写了如下内容:

echo -n 'Testing database connection  ...'

export COMMAND="from psycopg2 import connect; import sys; c=connect(dbname='myappdb',\ 
user='$POSTGRESQL_USER', password='$POSTGRESQL_PWD', host='127.0.0.1', port=5432); \
sys.exit(c.closed)"

test `python -c "$COMMAND" 2>/dev/null && echo true `

其中 test 是一个函数:

function test {
  [ $1 ] && echo "[OK]" || echo "[FAIL]"
}

似乎即使使用错误的凭据，也不会引发异常并且 c.closed 始终为零。有什么方法可以测试凭据在虚拟环境中是否正确？

Answer 1

它是 bash，不是 python，所以不确定它是否适合您的需要:

vao@vao-X102BA:~$ psql -d "postgresql://p:goodpassword@localhost/t" -c "select now()"
              now
-------------------------------
 2017-07-14 11:42:40.712981+03
(1 row)

vao@vao-X102BA:~$ echo $?
0
vao@vao-X102BA:~$ psql -d "postgresql://p:badpassword@localhost/t" -c "select now()"
psql: FATAL:  password authentication failed for user "p"
FATAL:  password authentication failed for user "p"
vao@vao-X102BA:~$ echo $?
2

如果你想抑制输出，而不是 select now()，你可以使用 -c "\q" 退出