Python tkinter - получить идентификатор SQLite из ttk.combobox - PullRequest
Новая
       91

Python tkinter - получить идентификатор SQLite из ttk.combobox

1 голос
/ 01 мая 2020

У меня есть ttk.combobox, заполненный данными из базы данных. Когда пользователь выбирает данные, мне нужно получить идентификатор выбранных данных.

У меня есть решение для этого, но я уверен, что есть более элегантное и простое решение для этого, но я не могу найти его. Я разделяю SQL идентификатор строки с помощью '.', А из Combobox я превращаю строку в список, разделенный на '.' и получить список [0] для идентификатора.

Пример кода: 

from tkinter import *
from tkinter import ttk
import sqlite3
import DataBasePM

DBProjectManager2=DataBasePM.DBProjectManager2

def DropDownProjectView():
    con=sqlite3.connect(DBProjectManager2)
    con.row_factory = lambda curs, row:str(row[0])+". "+ row[1] #split ID with '.' 
    curs= con.cursor()
    curs.execute( """SELECT idProject, ProjectName 
                FROM Project WHERE idStatus=1""")
    rows=curs.fetchall()
    con.close()
    return rows

def GetIDFromDropDown(pickedString):
                 GetID=pickedString
                 GetID = list(GetID.split(".")) #id is before '.'
                 GetID=(int(GetID[0])) 
                 print(GetID)


root = Tk()
root.title("Tkinter ex")
root.geometry("400x400")

project_name_drop =  ttk.Combobox (root, value=DropDownProjectView() )
project_name_drop.pack()

buttonA=Button(root, text="get ID",command=lambda: GetIDFromDropDown(project_name_drop.get()))
buttonA.pack()

root.mainloop()

1 Ответ

0 голосов
/ 05 мая 2020

Вы можете вернуть словарь вместо списка из DropDownProjectView(): 

def DropDownProjectView():
    con=sqlite3.connect(DBProjectManager2)
    # return two items in each record: dropdown-item, id
    con.row_factory = lambda curs, row: (str(row[0])+". "+row[1], row[0])
    curs= con.cursor()
    curs.execute("SELECT idProject, ProjectName FROM Project WHERE idStatus=1")
    # build a dictionary with dropdown-item as key and id as value
    rows = {r[0]:r[1] for r in curs}
    con.close()
    return rows

Затем используйте list(rows.keys()) в качестве элементов раскрывающегося списка: 

rows = DropDownProjectView()
project_name_drop =  ttk.Combobox (root, value=list(rows.keys()))
project_name_drop.pack()

Наконец, используйте rows[pickedString] для получить идентификатор в GetIDFromDropDown(): 

def GetIDFromDropDown(pickedString):
    # cater exceptional case
    if pickedString in rows:
        id = rows[pickedString]
        print(id)
    else:
        print("invalid option: '%s'" % pickedString)
...