在PyCharm中运行《AI智能联系人管理》即可进入如图1所示的系统主界面。
说明:在运行程序前,先将当前的计算机连接互联网,并且需要先申请汉王云名片识别接口的服务器Key,并且复制该Key到项目的key.txt文件中替换原来的说明文字即可。
具体的操作步骤如下:
(1)添加联系人。在主界面中,单击“添加”按钮,将打开添加联系人窗口,在该窗口中,单击“选择名片”按钮,选择一张名片对应的图片,系统将自动识别该名片中的信息,并且显示到右侧的文本框中,效果如图2所示。如果没有名片也可手动输入联系人信息。
(2)搜索联系人。在主界面顶部的文本框中,输入要搜索的内容,单击“搜索”按钮,将从已经保存的联系人信息中查找相应的内容,如果找到对应的内容,将显示在下方,如图3所示,否则给出显示“没有搜索内容”的提示对话框,如图4所示。
说明:在查找联系人时,也可以单击索引字母按钮,按拼音索引来查找。
(3)编辑或删除联系人。在显示联系人时,每个联系人的右侧都有两个按钮(如图5所示),一个是“编辑”,另一个是“删除”,单击不同的按钮将实现相应的功能。
(4)查看联系人的分布。在主界面中,单击“查看联系人分布”按钮,将打开“联系人分布”窗口,在该窗口中将以饼图显示联系人的分布,如图6所示。
# qt5模块
from PyQt5 import QtCore, QtGui, QtWidgets
from PyQt5.QtWidgets import *
from PyQt5.QtGui import *
from PyQt5.QtWebEngineWidgets import *
from PyQt5.QtCore import *
# 自定义模块
import mainpage
import addpage
import gridlayout
import editpage
import pinyintool
# 内置模块
import sys
import requests, base64, json
import collections
import os
import re
# 第三方模块
import pandas as pd
from pandas import DataFrame
from PIL import Image
import phone
from pyecharts import Pie
# 获取文件的路径
cdir = os.getcwd()
# 文件路径
path=cdir+'/res/datafile/'
# 读取路径 判断是否创建了文件
if not os.path.exists(path):
# 根据路径建立文件夹
os.makedirs(path)
# 姓名 公司 电话 手机 邮件 地址 城市 分类 name comp tel mobile email addr city type
cardfile = pd.DataFrame(columns=['name', 'comp', 'tel', 'mobile', 'email', 'addr', 'city', 'type'])
# 生成xlsx文件
cardfile.to_excel(path+'名片信息表.xlsx', sheet_name='data', index=None)
# 编辑页面
class editWindow(QWidget,editpage.Ui_Form):
# 初始化方法
def __init__(self):
# 找到父类 首页面
super(editWindow, self).__init__()
# 初始化页面方法
self.setupUi(self)
# 为保存按钮添加事件
self.pushButton_2.clicked.connect(self.editkeep)
# 显示添加名片页面
def OPEN(self):
# 显示页面
self.show()
# 保存编辑内容
def editkeep(self):
# 获取按钮名称
indexName = self.pushButton_2.objectName()
# 获取表
pi_table = pd.read_excel(path + '名片信息表.xlsx', sheet_name='data')
# 获取控件信息 # 'name', 'comp', 'tel', 'mobile', 'email', 'addr', 'city', 'type'
name = self.lineEdit.text()
comp = self.lineEdit_2.text()
tel = self.lineEdit_3.text()
mobile = self.lineEdit_4.text()
email = self.lineEdit_5.text()
addr = self.lineEdit_6.text()
# 判断手机号是否为空
if mobile.strip():
# 根据手机号判断区域
try:
info = phone.Phone().find(int(mobile))
except Exception as e:
print("根据手机号判断区域时出错", e)
QMessageBox.critical(self, "错误:", "手机号码不正确!", QMessageBox.Ok) # 弹出提示对话框
self.lineEdit_4.setFocus() # 让手机文本框获得焦点
return
# 判断手机号是否正确返回信息
if info == None:
city = '其他'
else:
# 正确返回信息获取省
city = info['province']
else:
city = '其他'
# 判断姓名是否为空
if name.strip():
# 获取首字母拼音
type = pinyintool.getPinyin(name[0])
# 根据行号删除数据
datas = pi_table.drop(index=[int(indexName)], axis=0)
data = datas.append({'name': name,
'comp': comp,
'tel': tel,
'mobile': mobile,
'email': email,
'addr': addr,
'city': city,
'type': type
}, ignore_index=True)
# 更新xlsx文件
DataFrame(data).to_excel(path + '名片信息表.xlsx',
sheet_name='data', index=False)
self.close()
window.dataall()
else:
QMessageBox.information(self, '提示信息', '姓名不能为空')
pass
#首页列表样式
class griditem(QWidget,gridlayout.Ui_Form):
# 初始化方法
def __init__(self):
# 找到父类 首页面
super(griditem, self).__init__()
# 初始化页面方法
self.setupUi(self)
# 显示饼图类
class HtmlWindows(QMainWindow):
def __init__(self):
super(QMainWindow,self).__init__()
self.setGeometry(200, 200, 850, 500)
self.browser = QWebEngineView()
def set(self,title,hurl):
self.setWindowTitle(title)
d = os.path.dirname(os.path.realpath(sys.argv[0])) # 获取当前文件所在路径
d=re.sub(r'\\','/',d) # 将路径中的分隔符\替换为/
url=d+'/res/datafile/'+hurl
self.browser.load(QUrl(url))
self.setCentralWidget(self.browser)
# 首页页面
class parentWindow(QWidget,mainpage.Ui_Form):
# 初始化方法
def __init__(self):
# 找到父类 首页面
super(parentWindow, self).__init__()
# 初始化页面方法
self.setupUi(self)
# 标签按钮绑定事件
self.pushButton_3.clicked.connect(self.dataall)
self.pushButton_4.clicked.connect(lambda:self.dataother('A'))
self.pushButton_6.clicked.connect(lambda:self.dataother('B'))
self.pushButton_5.clicked.connect(lambda:self.dataother('C'))
self.pushButton_7.clicked.connect(lambda:self.dataother('D'))
self.pushButton_8.clicked.connect(lambda:self.dataother('E'))
self.pushButton_9.clicked.connect(lambda:self.dataother('F'))
self.pushButton_10.clicked.connect(lambda:self.dataother('G'))
self.pushButton_11.clicked.connect(lambda:self.dataother('H'))
self.pushButton_12.clicked.connect(lambda:self.dataother('I'))
self.pushButton_13.clicked.connect(lambda:self.dataother('J'))
self.pushButton_14.clicked.connect(lambda:self.dataother('K'))
self.pushButton_15.clicked.connect(lambda:self.dataother('L'))
self.pushButton_16.clicked.connect(lambda:self.dataother('M'))
self.pushButton_17.clicked.connect(lambda:self.dataother('N'))
self.pushButton_18.clicked.connect(lambda:self.dataother('O'))
self.pushButton_19.clicked.connect(lambda:self.dataother('P'))
self.pushButton_20.clicked.connect(lambda:self.dataother('Q'))
self.pushButton_21.clicked.connect(lambda:self.dataother('R'))
self.pushButton_22.clicked.connect(lambda:self.dataother('S'))
self.pushButton_23.clicked.connect(lambda:self.dataother('T'))
self.pushButton_24.clicked.connect(lambda:self.dataother('U'))
self.pushButton_25.clicked.connect(lambda:self.dataother('V'))
self.pushButton_26.clicked.connect(lambda:self.dataother('W'))
self.pushButton_27.clicked.connect(lambda:self.dataother('X'))
self.pushButton_28.clicked.connect(lambda:self.dataother('Y'))
self.pushButton_29.clicked.connect(lambda:self.dataother('Z'))
# 搜索按钮绑定事件
self.pushButton.clicked.connect(self.seachbtn)
self.pushButton_30.clicked.connect(self.lookpie)
# 显示全部数据
self.dataall()
# 显示饼图
def lookpie(self):
# 获取表数据
# 读取文件内容
pi_table = pd.read_excel(path + '名片信息表.xlsx', sheet_name='data')
# 获取城市信息
citattr = pi_table['city'].values
# 判断是否有城市信息
if len(citattr)==0:
# 没有信息提示
QMessageBox.information(self, '提示信息', '没有联系人')
return
attr = []
v1 = []
# 循环表里city项
for i in citattr:
# 判断city项是否包含在attr里
if not i in attr:
# 不包含在attr表里时候添加到attr里
attr.append(i)
# Counter(计数器)是对字典的补充,用于追踪值的出现次数。
d = collections.Counter(citattr)
# 循环城市列表
for k in attr:
# d[k] 是k在列表d中出现的次数
v1.append(d[k])
# 生成饼图
pie = Pie("联系人分布")
pie.add("", attr, v1, is_label_show=True)
pie.show_config()
pie.render(path+'联系人分布饼图.html')
# 显示饼图
htmlwidows.set('联系人分布','联系人分布饼图.html')
htmlwidows.show()
pass
#搜索功能
def seachbtn(self):
# 读取文件内容
pi_table = pd.read_excel(path + '名片信息表.xlsx', sheet_name='data')
seachk=self.lineEdit.text()
# 查询数据 用户名和公司如果有一个包含搜索内容筛选出来
cardArray = pi_table[(pi_table['name'].str.contains(seachk)) | (pi_table['comp'].str.contains(seachk))].values
tb = pi_table[(pi_table['name'].str.contains(seachk)) | (pi_table['comp'].str.contains(seachk))]
if len(cardArray) == 0:
QMessageBox.information(self, '提示信息', '没有搜索内容')
else:
# 每次点循环删除管理器的组件
while self.gridLayout.count():
# 获取第一个组件
item = self.gridLayout.takeAt(0)
widget = item.widget()
# 删除组件
widget.deleteLater()
i = -1
for n in range(len(cardArray)):
# x 确定每行显示的个数 0,1,2 每行3个
x = n % 3
# 当x为0的时候设置换行 行数+1
if x == 0:
i += 1
item = griditem()
item.label_8.setText('姓名:' + str(cardArray[n][0]))
item.label_9.setText('公司:' + str(cardArray[n][1]))
item.label_10.setText('电话:' + str(cardArray[n][2]))
item.label_11.setText('手机:' + str(cardArray[n][3]))
item.label_12.setText('邮箱:' + str(cardArray[n][4]))
item.label_13.setText('地址:' + str(cardArray[n][5]))
# 设置名称 为获取项目行数
item.pushButton.setObjectName(str(tb.index.tolist()[n]))
item.pushButton_3.setObjectName(str(tb.index.tolist()[n]))
# 为按钮绑定点击事件
item.pushButton.clicked.connect(self.edit)
item.pushButton_3.clicked.connect(self.deletedata)
# 动态给gridlayout添加布局
self.gridLayout.addWidget(item, i, x)
# 设置上下滑动控件可以滑动 把scrollAreaWidgetContents_2添加到scrollArea中
self.scrollAreaWidgetContents.setMinimumHeight(i * 200)
# girdlayout 添加到滑动控件中
self.scrollAreaWidgetContents.setLayout(self.gridLayout)
#显示全部数据
def dataall(self):
# 每次点循环删除管理器的组件
while self.gridLayout.count():
# 获取第一个组件
item = self.gridLayout.takeAt(0)
widget = item.widget()
# 删除组件
widget.deleteLater()
i=-1
# 读取文件内容
pi_table = pd.read_excel(path + '名片信息表.xlsx', sheet_name='data')
# 获取所有数据
cardArray = pi_table.values
# 循环数据
for n in range(len(cardArray)):
# x 确定每行显示的个数 0,1,2 每行3个
x = n % 3
# 当x为0的时候设置换行,即行数+1
if x == 0:
i += 1
item = griditem()
item.label_8.setText('姓名:'+str(cardArray[n][0]))
item.label_9.setText('公司:'+str(cardArray[n][1]))
item.label_10.setText('电话:'+str(cardArray[n][2]))
item.label_11.setText('手机:'+str(cardArray[n][3]))
item.label_12.setText('邮箱:'+str(cardArray[n][4]))
item.label_13.setText('地址:'+str(cardArray[n][5]))
# 设置名称 为获取项目行数
item.pushButton.setObjectName(str(pi_table.index.tolist()[n]))
item.pushButton_3.setObjectName(str(pi_table.index.tolist()[n]))
# 为按钮绑定点击事件
item.pushButton.clicked.connect(self.edit)
item.pushButton_3.clicked.connect(self.deletedata)
# 动态添加控件到gridlayout中
self.gridLayout.addWidget(item,i, x)
# 设置上下滑动控件可以滑动 把scrollAreaWidgetContents_2添加到scrollArea中
self.scrollAreaWidgetContents.setMinimumHeight(i*200)
# 设置gridlayout到滑动控件中
self.scrollAreaWidgetContents.setLayout(self.gridLayout)
pass
# 删除数据方法
def deletedata(self):
# 获取信号源 点击的按钮
sender = self.gridLayout.sender()
# 获取按钮名称
indexName = sender.objectName()
# 获取表信息
pi_table = pd.read_excel(path + '名片信息表.xlsx', sheet_name='data')
# 根据行号删除数据
data=pi_table.drop(index=[int(indexName)], axis=0)
# 更新xlsx文件
DataFrame(data).to_excel(path + '名片信息表.xlsx',
sheet_name='data', index=False)
# 显示全部数据
self.dataall()
# 编辑数据
def edit(self):
# 获取信号源 点击的按钮
sender = self.gridLayout.sender()
# 获取按钮名称
indexName=sender.objectName()
pi_table = pd.read_excel(path + '名片信息表.xlsx', sheet_name='data')
# 根据行号 获取数据
cardArray =pi_table.iloc[[indexName]].values
# 打开编辑页面
editWindow.OPEN()
# 设置数据
editWindow.lineEdit.setText(str(cardArray[0][0]))
editWindow.lineEdit_2.setText(str(cardArray[0][1]))
editWindow.lineEdit_3.setText(str(cardArray[0][2]))
editWindow.lineEdit_4.setText(str(cardArray[0][3]))
editWindow.lineEdit_5.setText(str(cardArray[0][4]))
editWindow.lineEdit_6.setText(str(cardArray[0][5]))
# 设置按钮名称
editWindow.pushButton_2.setObjectName(str(indexName))
pass
# 显示部分数据
def dataother(self,typeAZ):
# 每次点循环删除管理器的组件
while self.gridLayout.count():
# 获取第一个组件
item = self.gridLayout.takeAt(0)
# 获取布局
widget = item.widget()
# 删除组件
widget.deleteLater()
pass
# 读取文件内容
pi_table = pd.read_excel(path + '名片信息表.xlsx', sheet_name='data')
i = -1
# 筛选内容
cardArray = pi_table[pi_table['type'] == typeAZ].values
tb= pi_table[pi_table['type'] == typeAZ]
for n in range(len(cardArray)):
# x 确定每行显示的个数 0,1,2 每行3个
x = n % 3
# 当x为0的时候设置换行 行数+1
if x == 0:
i += 1
item = griditem()
item.label_8.setText('姓名:' + str(cardArray[n][0]))
item.label_9.setText('公司:' + str(cardArray[n][1]))
item.label_10.setText('电话:' + str(cardArray[n][2]))
item.label_11.setText('手机:' + str(cardArray[n][3]))
item.label_12.setText('邮箱:' + str(cardArray[n][4]))
item.label_13.setText('地址:' + str(cardArray[n][5]))
# 设置名称 为获取项目行数
item.pushButton.setObjectName(str(tb.index.tolist()[n]))
item.pushButton_3.setObjectName(str(tb.index.tolist()[n]))
# 为按钮绑定点击事件
item.pushButton.clicked.connect(self.edit)
item.pushButton_3.clicked.connect(self.deletedata)
# 动态设置控件
self.gridLayout.addWidget(item, i, x)
# 动态设置滑动控件滑动高度
self.scrollAreaWidgetContents.setMinimumHeight(i * 200)
# giridLayout 添加到滑动控件中
self.scrollAreaWidgetContents.setLayout(self.gridLayout)
#添加名片页面
class childWindow(QWidget,addpage.Ui_Form):
# 初始化方法
def __init__(self):
# 找到父类 添加名片页面
super(childWindow, self).__init__()
# 初始化页面方法
self.setupUi(self)
# 给选择名片按钮添加事件
self.pushButton.clicked.connect(self.openfile)
# 给保存按钮添加事件
self.pushButton_2.clicked.connect(self.keep)
#保存名片信息到文档
def keep(self):
pi_table = pd.read_excel(path + '名片信息表.xlsx', sheet_name='data')
# 获取输出框内容
name =self.lineEdit.text()
comp = self.lineEdit_2.text()
tel= self.lineEdit_3.text()
mobile= self.lineEdit_4.text()
email= self.lineEdit_5.text()
addr= self.lineEdit_6.text()
# 判断手机号是否为空
if mobile.strip():
# 根据手机号判断区域
try:
info = phone.Phone().find(int(mobile))
except Exception as e:
print("根据手机号判断区域时出错",e)
QMessageBox.critical(self,"错误:","手机号码不正确!",QMessageBox.Ok) # 弹出提示对话框
self.lineEdit_4.setFocus() # 让手机文本框获得焦点
return
# 判断手机号是否正确返回信息
if info==None:
city = '其他'
else:
# 正确返回信息获取省
city = info['province']
else:
city = '其他'
# 判断姓名是否为空
if name.strip():
# 获取首字母拼音
type=pinyintool.getPinyin(name[0])
# 添加数据
data = pi_table.append({'name': name,
'comp': comp,
'tel': tel,
'mobile': mobile,
'email': email,
'addr': addr,
'city': city,
'type': type,
}, ignore_index=True)
# 更新xlsx文件
DataFrame(data).to_excel(path + '名片信息表.xlsx',
sheet_name='data', index=False)
# 首页面显示全部数据
window.dataall()
# 关闭添加页面
self.close()
else:
QMessageBox.information(self, '提示信息', '姓名不能为空')
pass
# 显示添加名片页面
def OPEN(self):
# 移除控件上图片
self.label_2.setPixmap(QPixmap(""))
# 移除输入框内容
self.lineEdit.setText("")
self.lineEdit_2.setText("")
self.lineEdit_3.setText("")
self.lineEdit_4.setText("")
self.lineEdit_5.setText("")
self.lineEdit_6.setText("")
# 显示页面
self.show()
#选择名片按钮执行方法
def openfile(self):
# 启动选择文件对话空,查找jpg以及png图片
self.download_path = QFileDialog.getOpenFileName(self, "选择要识别的图片", "/", "Image Files(*.jpg *.png)")
# 判断是否选择图片
if not self.download_path[0].strip():
# 没有选择图片执行
# 消息对话框 information 提问对话框 question 警告对话框 warning
# 严重错误对话框 critical 关于对话框 about
QMessageBox.information(self, '提示信息', '没有选择名片图片')
pass
else:
# 选择图片执行
# 图片路径self.download_path[0]
# 理由pixmap解析图片
pixmap = QPixmap(self.download_path[0])
# 设置图片
self.label_2.setPixmap(pixmap)
# 让图片自适应label大小
self.label_2.setScaledContents (True)
try:
# 识别名片图片返回识别结果
content=self.recg()
except:
QMessageBox.information(self, '提示信息', '识别错误请重新选择图片')
cjson = json.loads(content)
# 判断是否正确返回内容
if cjson['code']=='0':
# 名称
self.dcontent(1, cjson)
# 公司
self.dcontent(2, cjson)
# 电话
self.dcontent(3, cjson)
# 手机
self.dcontent(4, cjson)
# 邮件
self.dcontent(5, cjson)
# 地址
self.dcontent(6, cjson)
else:
QMessageBox.information(self, '提示信息', '信息码-'+cjson['code']+
' 请去官网http://developer.hanvon.com/api/toAPIinfo.do?id=2&num= 查看原因')
pass
#设置识别显示的内容
def dcontent(self,k,count):
try:
if k==1:
self.lineEdit.setText(count['name'][0])
elif k==2:
self.lineEdit_2.setText(count['comp'][0])
elif k==3:
self.lineEdit_3.setText(count['tel'][0])
elif k==4:
self.lineEdit_4.setText(count['mobile'][0])
elif k==5:
self.lineEdit_5.setText(count['email'][0])
elif k==6:
self.lineEdit_6.setText(count['addr'][0])
except:
pass
# 信号槽 点击按钮显示大图功能
def wichbtn(self):
# 使用电脑中的看图工具打开图片
img = Image.open(self.download_path[0])
# 显示图片
img.show()
# 识别名片图片
def recg(self):
# 汉王云名片接口 API申请地址
# http://developer.hanvon.com
# 请将您申请的Key写到项目根目录下的key.txt文件中,并且将其写在第一行
with open('key.txt',"r") as file:
key = file.readline() # 读取写到key.txt文件中的您申请的Key
url = 'https://api.hanvon.com/rt/ws/v1/ocr/bcard/recg?key=%s&code=91f6a58d-e418-4e58-8ec2-61b583c55ba2' % key
img = Image.open(self.download_path[0])
# 设置模式L为灰度图
img2 = img.convert('L')
_w = img2.width
_h = img2.height
img2 = img2.resize((int(_w), int(_h)), Image.ANTIALIAS)
img2.save('card_gray.jpg')
base64img = base64.b64encode(open('card_gray.jpg', 'rb').read()).decode()
'''
lang:指定使用的语言。值为:auto(多语言)、chns(中文简体)、chnt(中文繁体)、en(英文)
color:指定颜色。值为gray(进行灰度处理)、color(原图像未做任何处理 上传图像的类型)
image:指定名片图像的base64串
'''
data = {"lang": 'auto', "color": 'gray', "image": base64img}
headers = {"Content-Type": "application/octet-stream"}
resp = requests.post(url, data=json.dumps(data), headers=headers)
return resp.text
if __name__=='__main__':
# 每一个PyQt5应用都必须创建一个应用对象
app=QApplication(sys.argv)
# 初始化页面
window=parentWindow()
# 显示首页
window.show()
child = childWindow()
# 初始化编辑页面
editWindow = editWindow()
# 显示联系人窗体
htmlwidows = HtmlWindows()
# 添加按钮添加事件
window.pushButton_2.clicked.connect(child.OPEN)
# 项目结束调用
sys.exit(app.exec_())
python例程-AI智能联系人管理的程序使用说明.zip