我想知道,我想如何编写一些代码,这给了我在数据库中找到的元素的索引。
我有这样的功能:
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]))
但是它也不起作用。有什么建议吗?
答案 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)
然后可以将结果附加到记录列表中。