本文實例為大家分享了python圖書管理系統的具體代碼,供大家參考,具體內容如下
實現語言:python
圖形框架:DTK+2.0
數據庫框架:SQLite 3.0
本程序需要以下部件運行:
Python2.5.0、GTK+2.16、Pygtk 2.16、PyGobject 2.14、Pycairo 1.4
LibiaryManager.py
#!/usr/bin/python# -*- coding: utf-8 -*-import pygtkpygtk.require('2.0')import gtkimport sysimport gobjectimport sqlite3import timeimport datetimeclass LibManager: bookColumName = (u"書本號", u"書名", u"作者", u"出版社", u"價格", u"購入日期", u"分類", u"簡介", u"館藏數") readerColumName = (u"讀者號", u"讀者姓名", u"讀者身份", u"備注") def __init__(self): self.dbconn = sqlite3.connect('sqlitefile.db') self.cursor = self.dbconn.cursor() self.nowDate = time.strftime("%Y-%m-%d").decode("utf-8") inputBookInitData = (u"書名", u"作者", u"出版社", 0.0, self.nowDate, u"分類", u"簡介", 0) try: builder = gtk.Builder() builder.add_from_file("ui.glade") except BaseException, e: self.errorMessage("Fail to load UI file.") print e sys.exit(1) builder.connect_signals(self) self.window = builder.get_object("mainWindow") self.borrowView = builder.get_object("borrowView") self.borrowViewModel = builder.get_object("borrowViewModel") self.borrowViewInit() self.returnView = builder.get_object("returnView") self.returnViewModel = builder.get_object("returnViewModel") self.returnEntryReaderID = builder.get_object("returnReaderID") self.returnEntryBookID = builder.get_object("returnBookID") self.returnViewInit() self.queryView = builder.get_object("queryView") self.queryViewReaderModel = builder.get_object("queryViewReaderModel") self.queryViewBookModel = builder.get_object("queryViewBookModel") self.inputView = builder.get_object("inputView") self.inputViewBookModel = builder.get_object("inputViewBookModel") self.inputViewReaderModel = builder.get_object("inputViewReaderModel") self.inputViewModel = None def borrowViewInit(self): columName = (u"讀者號", u"書號", u"借出日期") columEditAttr = (True, True, False) for columnNum in range(len(columName)): renderer = gtk.CellRendererText() renderer.set_data("column", columnNum) renderer.set_property("editable", columEditAttr[columnNum]) if columEditAttr[columnNum]: renderer.connect("edited", self.on_borrow_cell_edited) column = gtk.TreeViewColumn(columName[columnNum], renderer, text=columnNum) column.set_resizable(True) self.borrowView.append_column(column) def returnViewInit(self): columName = (u"借書記錄", u"書本", u"讀者姓名", u"讀者資料", u"借出日期") for columnNum in range(len(columName)): renderer = gtk.CellRendererText() column = gtk.TreeViewColumn(columName[columnNum], renderer, text=columnNum) column.set_resizable(True) self.returnView.append_column(column) # QueryView def queryViewSetColumn(self, columName): self.queryRecord = 0 columns = self.queryView.get_columns() for c in columns: self.queryView.remove_column(c) self.queryViewModel.clear() self.queryView.set_model(self.queryViewModel) for columnNum in range(len(columName)): renderer = gtk.CellRendererText() renderer.set_property("editable", True) column = gtk.TreeViewColumn(columName[columnNum], renderer, text=columnNum) column.set_resizable(True) self.queryView.append_column(column) def queryReader(self): self.queryViewModel = self.queryViewReaderModel self.queryViewSetColumn(self.readerColumName) self.cursor.execute("SELECT COUNT(*) FROM reader") self.maxRecords = self.cursor.fetchone()[0] print self.maxRecords def queryBook(self): self.queryViewModel = self.queryViewBookModel self.queryViewSetColumn(self.bookColumName) self.cursor.execute("SELECT COUNT(*) FROM books") self.maxRecords = self.cursor.fetchone()[0] print self.maxRecords #inputView def inputViewSetColumn(self, columName): columns = self.inputView.get_columns() for c in columns: self.inputView.remove_column(c) self.inputViewModel.clear() self.inputView.set_model(self.inputViewModel) for columnNum in range(len(columName)): renderer = gtk.CellRendererText() renderer.set_property("editable", True) renderer.set_data("column", columnNum) renderer.connect("edited", self.on_input_cell_edited) column = gtk.TreeViewColumn(columName[columnNum], renderer, text=columnNum) column.set_resizable(True) self.inputView.append_column(column) def inputReader(self): self.inputViewModel = self.inputViewReaderModel self.inputViewSetColumn(self.readerColumName[1:]) def inputBook(self): self.inputViewModel = self.inputViewBookModel self.inputViewSetColumn(self.bookColumName[1:]) #Callback Functions def on_mainWindow_delete_event(self, widget, data=None): print "destroy signal occurred" gtk.main_quit() #BorrowView def on_borrowAdd_clicked(self, button, data = None): self.borrowViewModel.append((1001, 1, self.nowDate)) def on_borrowDel_clicked(self, button, data = None): selection = self.borrowView.get_selection() model, iter = selection.get_selected() if iter: model.remove(iter) def on_borrowClear_clicked(self, button, data = None): self.borrowViewModel.clear() def on_borrowSubmit_clicked(self, button, data = None): for row in self.borrowViewModel: self.cursor.execute("INSERT INTO borrow VALUES (NULL, ?, ?, ?)", row) self.dbconn.commit() self.infoMessage(u"借書記錄已經成功提交。") self.borrowViewModel.clear() def on_borrow_cell_edited(self, cell, path_string, new_text): model = self.borrowViewModel iter = model.get_iter_from_string(path_string) column = cell.get_data("column") try: id = int(new_text) except ValueError, e: print e self.errorMessage(u"請輸入純數字記錄。") return if column == 0: self.cursor.execute("SELECT * FROM reader where ReaderID = ?", [new_text]) if self.cursor.fetchone(): model.set(iter, column, id) else: self.errorMessage(u"不存在該讀者號: %d。" % id) if column == 1: self.cursor.execute("SELECT * FROM books where BookID = ?", [new_text]) if self.cursor.fetchone(): model.set(iter, column, id) else: self.errorMessage(u"不存在該書本號: %d。" % id) #QueryView def on_queryBooks_clicked(self, button, data = None): self.queryBook() self.cursor.execute("SELECT * FROM books LIMIT ? OFFSET ?", [15, 0]) for row in self.cursor: self.queryViewModel.append(row) def on_queryReaders_clicked(self, button, data = None): self.queryReader() self.cursor.execute("SELECT * FROM reader LIMIT ? OFFSET ?", [15, 0]) for row in self.cursor: self.queryViewModel.append(row) def on_queryPrev_clicked(self, button, data = None): if self.queryRecord == 0: self.errorMessage(u"已到最前頁。") return self.queryRecord -= 15 self.queryViewModel.clear() if self.queryViewModel.get_n_columns() == 9: self.cursor.execute("SELECT * FROM books LIMIT ? OFFSET ?",[15, self.queryRecord]) elif self.queryViewModel.get_n_columns() == 4: self.cursor.execute("SELECT * FROM reader LIMIT ? OFFSET ?",[15, self.queryRecord]) for row in self.cursor: self.queryViewModel.append(row) def on_queryNext_clicked(self, button, data = None): if self.queryRecord + 15 > self.maxRecords: self.errorMessage(u"已到最后頁。") return self.queryRecord += 15 self.queryViewModel.clear() if self.queryViewModel.get_n_columns() == 9: self.cursor.execute("SELECT * FROM books LIMIT ? OFFSET ?",[15, self.queryRecord]) elif self.queryViewModel.get_n_columns() == 4: self.cursor.execute("SELECT * FROM reader LIMIT ? OFFSET ?",[15, self.queryRecord]) for row in self.cursor: self.queryViewModel.append(row) def on_returnQuery_clicked(self, button, data = None): readerid = self.returnEntryReaderID.get_text() bookid = self.returnEntryBookID.get_text() if len(readerid) and len(bookid): self.cursor.execute("SELECT borrow.BorrowID, books.BookName, reader.ReaderName,/ reader.ReaderSchool, borrow.BorrowDate FROM borrow,books,reader/ WHERE books.BookID = borrow.BookID AND / reader.ReaderID = borrow.ReaderID AND/ borrow.ReaderID = ? AND borrow.BookID = ?", [readerid, bookid]) elif len(readerid): self.cursor.execute("SELECT borrow.BorrowID, books.BookName, reader.ReaderName,/ reader.ReaderSchool, borrow.BorrowDate FROM borrow,books,reader/ WHERE books.BookID = borrow.BookID AND / reader.ReaderID = borrow.ReaderID AND/ borrow.ReaderID = ?", [readerid]) elif len(bookid): self.cursor.execute("SELECT borrow.BorrowID, books.BookName, reader.ReaderName,/ reader.ReaderSchool, borrow.BorrowDate FROM borrow,books,reader/ WHERE books.BookID = borrow.BookID AND / reader.ReaderID = borrow.ReaderID AND/ borrow.BookID = ?", [bookid]) else: self.errorMessage(u"請輸入查詢條件。") return self.returnViewModel.clear() for row in self.cursor: self.returnViewModel.append(row) def on_returnCommit_clicked(self, button, data = None): selection = self.returnView.get_selection() model, iter = selection.get_selected() if iter: borrowDate = time.strptime(model.get_value(iter, 4), "%Y-%m-%d") timeDelta = datetime.date.today() - datetime.date(*borrowDate[0:3]) borrowDays = timeDelta.days borrowID = model.get_value(iter, 0) self.cursor.execute("DELETE FROM borrow WHERE BorrowID = ?", [borrowID]) self.dbconn.commit() readerName = model.get_value(iter, 2) bookName = model.get_value(iter, 1) self.infoMessage(u"讀者 %s 所借書《%s》歸還成功。 借出%d天" % (readerName, bookName, borrowDays)) model.remove(iter) #inputView def on_inputReader_clicked(self, button, data = None): self.inputReader() def on_inputBooks_clicked(self, button, data = None): self.inputBook() def on_inputAdd_clicked(self, button, data = None): if not self.inputViewModel: self.errorMessage(u"請先選擇錄入類型。") return count = self.inputViewModel.get_n_columns() if count == 3: inputReaderInitData = (u"姓名", u"身份", u"備注") self.inputViewModel.append(inputReaderInitData) elif count == 8: inputBookInitData = (u"書名", u"作者", u"出版社", 0.0, self.nowDate, u"分類", u"簡介", 0) self.inputViewModel.append(inputBookInitData) def on_inputDel_clicked(self, button, data = None): selection = self.inputView.get_selection() model, iter = selection.get_selected() if iter: model.remove(iter) def on_inputSubmit_clicked(self, button, data = None): count = self.inputViewModel.get_n_columns() for row in self.inputViewModel: #錄入讀者 if count == 3: self.cursor.execute(u"INSERT INTO reader VALUES (NULL, ?, ?, ?)", [s.decode("utf-8") for s in row]) #錄入書目 elif count == 8: dataRow = [] for obj in row: if isinstance(obj, str): dataRow.append(obj.decode("utf-8")) else: dataRow.append(obj) print dataRow self.cursor.execute(u"INSERT INTO books VALUES (NULL, ?, ?, ?, ?, ?, ?, ?, ?)",/ dataRow) self.dbconn.commit() self.infoMessage(u"記錄已經成功提交。") self.inputViewModel.clear() def on_input_cell_edited(self, cell, path_string, new_text): iter = self.inputViewModel.get_iter_from_string(path_string) column = cell.get_data("column") columnCount = self.inputViewModel.get_n_columns() if columnCount == 3: self.inputViewModel.set(iter, column, new_text) elif columnCount == 8: if self.inputViewModel.get_column_type(column) == gobject.TYPE_FLOAT: try: self.inputViewModel.set(iter, column, float(new_text)) except ValueError, e: self.errorMessage(u"請輸入價格實數。") return elif self.inputViewModel.get_column_type(column) == gobject.TYPE_UINT: try: self.inputViewModel.set(iter, column, int(new_text)) except ValueError, e: self.errorMessage(u"請輸入整數。") return else: self.inputViewModel.set(iter, column, new_text) def errorMessage(self, message): print message dialog = gtk.MessageDialog(None, gtk.DIALOG_MODAL | gtk.DIALOG_DESTROY_WITH_PARENT, gtk.MESSAGE_ERROR, gtk.BUTTONS_OK, message) dialog.run() dialog.destroy() def infoMessage(self, message): print message dialog = gtk.MessageDialog(None, gtk.DIALOG_MODAL | gtk.DIALOG_DESTROY_WITH_PARENT, gtk.MESSAGE_INFO, gtk.BUTTONS_OK, message) dialog.run() dialog.destroy() def main(self): self.window.show() gtk.main()if __name__ == "__main__": app = LibManager() app.main()
新聞熱點
疑難解答