用 eric6 与 PyQt5 实现python的极速GUI编程（系列04）---- PyQt5自带教程：地址簿(address book)

# -*- coding: utf-8 -*-

import pickle

from PyQt5.QtCore import pyqtSlot, QFile, QIODevice, Qt, QTextStream
from PyQt5.QtWidgets import QWidget, QDialog, QLabel, QLineEdit, QPushButton, QHBoxLayout,  QMessageBox, QFileDialog

from Ui_addressbook import Ui_Dialog

class SortedDict(dict):
class Iterator(object):
def __init__(self, sorted_dict):
self._dict = sorted_dict
self._keys = sorted(self._dict.keys())
self._nr_items = len(self._keys)
self._idx = 0

def __iter__(self):
return self

def next(self):
if self._idx >= self._nr_items:
raise StopIteration

key = self._keys[self._idx]
value = self._dict[key]
self._idx += 1

return key, value

__next__ = next

def __iter__(self):
return SortedDict.Iterator(self)

iterkeys = __iter__

class FindDialog(QDialog):
def __init__(self, parent=None):
super(FindDialog, self).__init__(parent)

findLabel = QLabel('输入要查找的联系人姓名：')
self.lineEdit = QLineEdit()

self.findButton = QPushButton('查找')
self.findText = ''

layout = QHBoxLayout()
layout.addWidget(findLabel)
layout.addWidget(self.lineEdit)
layout.addWidget(self.findButton)

self.setLayout(layout)
self.setWindowTitle('查找联系人')

self.findButton.clicked.connect(self.findClicked)
self.findButton.clicked.connect(self.accept)

def findClicked(self):
text = self.lineEdit.text()

if not text:
QMessageBox.information(self, '姓名不能为空', '请输入一个姓名')
return

self.findText = text
self.lineEdit.clear()
self.hide()

def getFindText(self):
return self.findText

class Dialog(QDialog, Ui_Dialog):
NavigationMode, AddingMode, EditingMode = range(3)

def __init__(self, parent=None):
super(Dialog, self).__init__(parent)
self.setupUi(self)

self.pushButton_submit.hide()
self.pushButton_cancel.hide()

self.contacts = SortedDict()
self.oldName = ''
self.oldAddress = ''
self.currentMode = self.NavigationMode

self.dialog = FindDialog()

@pyqtSlot()
def on_pushButton_add_clicked(self):
self.oldName = self.lineEdit_name.text()
self.oldAddress = self.textEdit_address.toPlainText()

self.lineEdit_name.clear()
self.textEdit_address.clear()

self.updateInterface(self.AddingMode)

@pyqtSlot()
def on_pushButton_edit_clicked(self):
self.oldName = self.lineEdit_name.text()
self.oldAddress = self.textEdit_address.toPlainText()

self.updateInterface(self.EditingMode)

@pyqtSlot()
def on_pushButton_remove_clicked(self):
name = self.lineEdit_name.text()
address = self.textEdit_address.toPlainText()

if name in self.contacts:
button = QMessageBox.question(self, '确定删除','你真的确定要删除 {} 吗？'.format(name), QMessageBox.Yes | QMessageBox.No)

if button == QMessageBox.Yes:
self.on_pushButton_previous_clicked()
del self.contacts[name]

QMessageBox.information(self, '删除成功','{}已经从你的地址簿删除了！'.format(name))

self.updateInterface(self.NavigationMode)

@pyqtSlot()
def on_pushButton_find_clicked(self):
self.dialog.show()

if self.dialog.exec_() == QDialog.Accepted:
contactName = self.dialog.getFindText()

if contactName in self.contacts:
self.lineEdit_name.setText(contactName)
self.textEdit_address.setText(self.contacts[contactName])
else:
QMessageBox.information(self, '找不到','抱歉，{}不在你的地址簿内！'.format(contactName))
return

self.updateInterface(self.NavigationMode)

@pyqtSlot()
def on_pushButton_submit_clicked(self):
name = self.lineEdit_name.text()
address = self.textEdit_address.toPlainText()

if name == "" or address == "":
QMessageBox.information(self, '不能为空', '请输入姓名及地址！')
return

if self.currentMode == self.AddingMode:
if name not in self.contacts:
self.contacts[name] = address
QMessageBox.information(self, '添加成功', '{} 已经添加到你的地址簿！'.format(name))
else:
QMessageBox.information(self, '添加失败', '{} 已经存在于你的地址簿！'.format(name))
return

elif self.currentMode == self.EditingMode:
if self.oldName != name:
if name not in self.contacts:
QMessageBox.information(self, '编辑成功','{} 已经被编辑到你的地址簿！'.format(self.oldName))

del self.contacts[self.oldName]
self.contacts[name] = address
else:
QMessageBox.information(self, '编辑失败','抱歉，{} 已经存在于你的地址簿！'.format(name))
return
elif self.oldAddress != address:
QMessageBox.information(self, '编辑成功','{} 已经被编辑到你的地址簿！'.format(name))
self.contacts[name] = address

self.updateInterface(self.NavigationMode)

@pyqtSlot()
def on_pushButton_cancel_clicked(self):
self.lineEdit_name.setText(self.oldName)
self.textEdit_address.setText(self.oldAddress)

self.updateInterface(self.NavigationMode)

@pyqtSlot()
def on_pushButton_load_clicked(self):
fileName, _ = QFileDialog.getOpenFileName(self, '打开地址簿', '', '地址簿文件 (*.abk);;所有文件 (*)')

if not fileName:
return

try:
in_file = open(str(fileName), 'rb')
except IOError:
QMessageBox.information(self, '不能打开文件','打开文件 {} 时发生错误！'.format(fileName))
return

self.contacts = pickle.load(in_file)
in_file.close()

if len(self.contacts) == 0:
QMessageBox.information(self, '文件中无联系人','你打开的文件中无联系人！')
else:
for name, address in self.contacts:
self.lineEdit_name.setText(name)
self.textEdit_address.setText(address)

self.updateInterface(self.NavigationMode)

@pyqtSlot()
def on_pushButton_save_clicked(self):
fileName, _ = QFileDialog.getSaveFileName(self, '保存地址簿', '', '地址簿文件 (*.abk);;所有文件 (*)')

if not fileName:
return

try:
out_file = open(str(fileName), 'wb')
except IOError:
QMessageBox.information(self, '不能打开文件','打开文件 {} 时发生错误！'.format(fileName))
return

pickle.dump(self.contacts, out_file)
out_file.close()

@pyqtSlot()
def on_pushButton_export_clicked(self):
name = str(self.lineEdit_name.text())
address = self.textEdit_address.toPlainText()

nameList = name.split()

if len(nameList) > 1:
firstName = nameList[0]
lastName = nameList[-1]
else:
firstName = name
lastName = ''

fileName, _ = QFileDialog.getSaveFileName(self, '导出联系', '', 'vCard 文件 (*.vcf);;所有文件 (*)')

if not fileName:
return

out_file = QFile(fileName)

if not out_file.open(QIODevice.WriteOnly):
QMessageBox.information(self, '不能打开文件', out_file.errorString())
return

out_s = QTextStream(out_file)

out_s << 'BEGIN:VCARD' << '\n'
out_s << 'VERSION:2.1' << '\n'
out_s << 'N:' << lastName << ';' << firstName << '\n'
out_s << 'FN:' << ' '.join(nameList) << '\n'

address.replace(';', '\\;')
address.replace('\n', ';')
address.replace(',', ' ')

out_s << 'ADR;HOME:;' << address << '\n'
out_s << 'END:VCARD' << '\n'

QMessageBox.information(self, '导出成功','{} 已经被导出为 vCard ！'.format(name))

@pyqtSlot()
def on_pushButton_previous_clicked(self):
name = self.lineEdit_name.text()

prev_name = prev_address = None
for this_name, this_address in self.contacts:
if this_name == name:
break

prev_name = this_name
prev_address = this_address
else:
self.lineEdit_name.clear()
self.textEdit_address.clear()
return

if prev_name is None:
for prev_name, prev_address in self.contacts:
pass

self.lineEdit_name.setText(prev_name)
self.textEdit_address.setText(prev_address)

@pyqtSlot()
def on_pushButton_next_clicked(self):
name = self.lineEdit_name.text()
it = iter(self.contacts)

try:
while True:
this_name, _ = it.next()

if this_name == name:
next_name, next_address = it.next()
break
except StopIteration:
next_name, next_address = iter(self.contacts).next()

self.lineEdit_name.setText(next_name)
self.textEdit_address.setText(next_address)

def updateInterface(self, mode):
self.currentMode = mode

if self.currentMode in (self.AddingMode, self.EditingMode):
self.lineEdit_name.setReadOnly(False)
self.lineEdit_name.setFocus(Qt.OtherFocusReason)
self.textEdit_address.setReadOnly(False)

self.pushButton_add.setEnabled(False)
self.pushButton_edit.setEnabled(False)
self.pushButton_remove.setEnabled(False)

self.pushButton_next.setEnabled(False)
self.pushButton_previous.setEnabled(False)

self.pushButton_submit.show()
self.pushButton_cancel.show()

self.pushButton_load.setEnabled(False)
self.pushButton_save.setEnabled(False)
self.pushButton_export.setEnabled(False)

elif self.currentMode == self.NavigationMode:
if not self.contacts:
self.lineEdit_name.clear()
self.textEdit_address.clear()

self.lineEdit_name.setReadOnly(True)
self.textEdit_address.setReadOnly(True)
self.pushButton_add.setEnabled(True)

number = len(self.contacts)
self.pushButton_edit.setEnabled(number >= 1)
self.pushButton_remove.setEnabled(number >= 1)
self.pushButton_find.setEnabled(number > 2)
self.pushButton_next.setEnabled(number > 1)
self.pushButton_previous.setEnabled(number >1 )

self.pushButton_submit.hide()
self.pushButton_cancel.hide()

self.pushButton_export.setEnabled(number >= 1)

self.pushButton_load.setEnabled(True)
self.pushButton_save.setEnabled(number >= 1)

if __name__ == '__main__':
import sys
from PyQt5.QtWidgets import QApplication

app = QApplication(sys.argv)
dlg = Dialog()
dlg.show()
sys.exit(app.exec_())