我是 python 新手,正在尝试将这些文件夹中的文件夹名称和文本文件添加到数据库中。问题是我不知道如何将“textfiles”和“opendirectory”添加到数据库中。请查看此代码并帮助我。谢谢
#!/usr/bin/python
from easygui import *
import sys, glob, os, sqlite3
msgbox("Please choose your folder ","Welcome to MTT", ok_button ="Choose")
opendirectory = diropenbox("Welcome", "MTT",None)
con = sqlite3.connect('test.db')
cur = con.cursor()
cur.execute('DROP TABLE IF EXISTS folder')
cur.execute('DROP TABLE IF EXISTS file')
cur.execute('CREATE TABLE folder( folderid INTEGER PRIMARY KEY, foldername VARCHAR(120))')
cur.execute('CREATE TABLE file( fileid INTEGER PRIMARY KEY, folderid INTEGER, dataname VARCHAR(120), FOREIGN KEY(folderid) REFERENCES foldername(folderid))')
con.commit()
def main():
for dirpath,dirnames,filenames in os.walk(opendirectory):
for textfiles in filenames:
print textfiles
print opendirectory
cur.execute ('INSERT INTO folder (folderid, foldername) VALUES (null,opendirector)')
cur.execute('INSERT INTO file(fileid, dataname) VALUES(null,textfiles)')
cur.execute('SELECT * FROM folder')
print cur.fetchall()
main()
print 'success'