给定特定输出时如何触发 'return'

Question

我正在 Python 中开发身份验证脚本。我需要 PyMySQL 检查用户是否存在，我该怎么做？

我阅读了文档，但没有找到问题的答案。

def check():
    connection = pymysql.connect(host='localhost',
                        user='user',
                        password='passwd',
                        db='db',
                        charset='utf8mb4',
                        cursorclass=pymysql.cursors.DictCursor)

    userexist = # PyMySQL check code here
    if userexist=='True':
       return 'good'

Answer 1

不清楚您是要检查 pymysql 用户还是数据库中的用户。此代码将解决这两种情况。 pymysql 将引发异常：pymysql.err.OperationalError 如果用户不存在。因此，您需要尝试一下，除了像这样的子句：

def check():
    try:
        connection = pymysql.connect(host='localhost',
                                     user='user',
                                     password='passwd',
                                     db='db',
                                     charset='utf8mb4',
                                 cursorclass=pymysql.cursors.DictCursor)
    except pymysql.err.OperationalError:
        print('User with entered connection credentials does not exists')
        return False
    try:
        with connection.cursor() as cursor:
            cursor.execute("select...") # Fill with your select statement
            result = cursor.fetchone()
            if result is None:
                connection.close()
                return False
    finally:
        connection.close()
    return 'good'

给定特定输出时如何触发 'return'

How to trigger 'return' when specific output given

python

pymysql