如何将元素从数组传递到查询

时间:2019-04-29 13:36:37

标签: python python-3.x postgresql

我想知道,我想如何编写一些代码,这给了我在数据库中找到的元素的索引。

我有这样的功能:

def get_teachers_names(name, lastname):
    try:
        params = config()

        conn = psycopg2.connect(**params)
        cur = conn.cursor()

        select_query = "SELECT id FROM lecturers WHERE name = %s AND lastname = %s"
        record_to_find = (name, lastname)

        cur.execute(select_query, record_to_find)
        records = list(cur)

        conn.commit()
        cur.close()

        return records

    except (Exception, psycopg2.DatabaseError) as error:
        print(error)

当我做类似的事情时:

index = database.get_teachers_names('Name1', 'Surname1')

我得到我想要的索引。但是当我有一个列表时:

names = ('Name1', 'Name2')
surnames = ('Surname1', 'Surname2')

而且我尝试使用以下方法:

index = database.get_teachers_names(names[0], surnames[0]

我得到[]作为结果。我也尝试过:

index = database.get_teachers_names(''.join(names[0], ''.join(surnames[0]))

但是它也不起作用。有什么建议吗?

1 个答案:

答案 0 :(得分:0)

当您传递简单的字符串时,它会起作用,因为它们会正确填充查询字符串:

index = database.get_teachers_names('Name1', 'Surname1')

但是,当您传递列表时,它不会在以下位置正确解析为字符串:

"SELECT id FROM lecturers WHERE name = %s AND lastname = %s"

我认为对您来说最好的方法是遍历列表并为每个条目执行此操作: cur.execute(select_query,record_to_find)

然后可以将结果附加到记录列表中。