PyQT 跟我学做密码管理器(5)
PyQT 跟我学做密码管理器(5) —— 备份数据到邮箱
声明:本文借鉴https://blog.csdn.net/bigbennyguo/article/details/50755207 修改而来
前言
借鉴的文章是基于python2+pyqt4,本文是基于python3+pyqt5
环境
编辑器:Sublime Text3
操作系统:win10
Python3
所需要的库:主要是PyQT5,因为加了一点拓展功能,所以还有sqlite3,Image,pyinstaller
pip3 install pyqt5
pip3 install pyqt5-tools
pip3 install sqilte3
pip3 install Image
pip3 install pyinstaller
首先是pyqt5, 这个是我们的主要工具,用来做界面的,pyqt5-tools这个库主要是是在用QT Creator设计好界面之后,将ui文件转化为py文件;
sqlite3这个库可以安也可以不安,因为pyqt5里面有一个模块QtSql,它支持sqlite的操作;
Image这个库主要是用来处理一下图片资源,以及最后程序的图标的
pyinstaller 最后将程序转化为可执行的exe文件
邮箱登陆
python自带的smtblib库可以登陆邮箱,本文主要介绍如何登陆邮箱
如果是公司自己的邮箱,可能直接使用smtplib.SMTP().login()也许就登陆上去了
每个公司的邮箱可能不一样,这里介绍qq/163的邮箱登陆
使用python来登陆qq/163邮箱,第一件事,得从邮箱里面获取授权码
授权码应该是一个16位字母组成的
拿到授权码之后,我们就可以进行邮箱登陆了
import smtplib
from email.mime.text import MIMEText
from email.mime.multipart import MIMEMultipart
class SendEmail():
global send_user
global email_host
global password
global user_list
global sub
password = "" #授权码
email_host = "smtp.qq.com" #邮箱host,如果是163的,则是smtp.163.com
send_user = "123@qq.com" #自己的qq邮箱账户或者163账户
user_list = ['123@qq.com','123@163.com'] #收件人列表
sub = "密码管理系统-密码备份" #邮件主题
def __init__(self):
self.server = smtplib.SMTP_SSL()
self.server.connect(email_host,465)
code,resp=self.server.login(send_user,password) #登陆邮箱
if code in (235, 503): #登陆成功的话,code应该为235,已经登陆过的话,code应该是503
print('login success')
else:
print('没有登陆成功')
return
如果登陆成功之后呢,就可以开始构造邮件了,构造邮件用到了python自带的email模块
如下:
user = "shape" + "<" + send_user + ">" #组装邮件消息头,包括收件人,发件人,邮件主题
self.message = MIMEMultipart('mixed')
self.message['Subject'] = sub
self.message['From'] = user
self.message['To'] = ";".join(user_list)
邮件正文:
text_plain = MIMEText(content,'plain', 'utf-8')
邮件附件:
#构造附件
sendfile=open(filename,'rb').read()
text_att = MIMEText(sendfile, 'base64', 'utf-8')
text_att["Content-Type"] = 'application/octet-stream'
new_name=filename.split('\\')[-1]
print(new_name)
text_att.add_header('Content-Disposition', 'attachment', filename=new_name)
完整的邮件类如下:
import smtplib
from email.mime.text import MIMEText
from email.mime.multipart import MIMEMultipart
class SendEmail():
global send_user
global email_host
global password
global user_list
global sub
password = "" #授权码
email_host = "smtp.qq.com" #邮箱host,如果是163的,则是smtp.163.com
send_user = "123@qq.com" #自己的qq邮箱账户或者163账户
user_list = ['123@qq.com','123@163.com'] #收件人列表
sub = "密码管理系统-密码备份" #邮件主题
def __init__(self):
self.server = smtplib.SMTP_SSL()
self.server.connect(email_host,465)
code,resp=self.server.login(send_user,password) #登陆邮箱
if code in (235, 503): #登陆成功的话,code应该为235,已经登陆过的话,code应该是503
print('login success')
else:
print('没有登陆成功')
return
user = "shape" + "<" + send_user + ">" #组装邮件消息头,包括收件人,发件人,邮件主题
self.message = MIMEMultipart('mixed')
self.message['Subject'] = sub
self.message['From'] = user
self.message['To'] = ";".join(user_list)
def send_textmail(self,content): #发送文本邮件
text_plain = MIMEText(content,'plain', 'utf-8')
self.message.attach(text_plain)
def send_Enclosuremail(self,filename): #发送附件
#构造附件
sendfile=open(filename,'rb').read()
text_att = MIMEText(sendfile, 'base64', 'utf-8')
text_att["Content-Type"] = 'application/octet-stream'
new_name=filename.split('\\')[-1]
print(new_name)
text_att.add_header('Content-Disposition', 'attachment', filename=new_name)
self.message.attach(text_att)
def sendmailforbackup(self,content,file):
self.send_textmail(content)
self.send_Enclosuremail(file)
#self.server.sendmail(send_user,user_list,self.message.as_string())
def close(self): #关闭邮件sever
print('mail class close')
self.server.close()
为密码管理器添加备份功能
其实在上一篇文章中,备份功能的接口已经预留好了,仅仅是没有实现它
要做的就是,从数据库中读取所有的数据,然后处理一下数据内容,将其转化为字符串
将数据库文件作为附件同时通过邮件发送出去
def backupAction_def(self):
cur = self.conn.cursor()
cur.execute('SELECT * FROM INFO')
data = cur.fetchall()
print(len(data))
a=' '.join(['Website', 'Username', 'Password', 'Url'])+'\n'
for items in data:
for item in items:
a=a+str(item)+' '
a =a+'\n'
myEmail=SendEmail()
myEmail.sendmailforbackup(a,self.dbpath)
myEmail.close()
小结
至此,这个密码管理器的所有功能全部完成。
目前就剩临门一脚了,将代码整合,打包生成exe文件,这个我们下一篇文章再说
目前的代码如下:
# -*- coding: utf-8 -*-
# @Date : 2018-12-17 16:50:23
# @Author : Jimy_Fengqi (jmps515@163.com)
# @Link : https://blog.csdn.net/qiqiyingse
# @Version : V1.0
# @pyVersion: 3.6
import sys,sqlite3,os
from PyQt5.QtWidgets import *
from PyQt5.QtGui import QIcon, QFont
from PyQt5.QtCore import *
import images
import smtplib
from email.mime.text import MIMEText
from email.mime.multipart import MIMEMultipart
class SendEmail():
global send_user
global email_host
global password
global user_list
global sub
password = "" #授权码
email_host = "smtp.qq.com" #邮箱host,如果是163的,则是smtp.163.com
send_user = "123@qq.com" #自己的qq邮箱账户或者163账户
user_list = ['123@qq.com','123@163.com'] #收件人列表
sub = "密码管理系统-密码备份" #邮件主题
def __init__(self):
self.server = smtplib.SMTP_SSL()
self.server.connect(email_host,465)
code,resp=self.server.login(send_user,password) #登陆邮箱
if code in (235, 503): #登陆成功的话,code应该为235,已经登陆过的话,code应该是503
print('login success')
else:
print('没有登陆成功')
return
user = "shape" + "<" + send_user + ">" #组装邮件消息头,包括收件人,发件人,邮件主题
self.message = MIMEMultipart('mixed')
self.message['Subject'] = sub
self.message['From'] = user
self.message['To'] = ";".join(user_list)
def send_textmail(self,content): #发送文本邮件
text_plain = MIMEText(content,'plain', 'utf-8')
self.message.attach(text_plain)
def send_Enclosuremail(self,filename): #发送附件
#构造附件
sendfile=open(filename,'rb').read()
text_att = MIMEText(sendfile, 'base64', 'utf-8')
text_att["Content-Type"] = 'application/octet-stream'
new_name=filename.split('\\')[-1]
print(new_name)
text_att.add_header('Content-Disposition', 'attachment', filename=new_name)
self.message.attach(text_att)
def sendmailforbackup(self,content,file):
self.send_textmail(content)
self.send_Enclosuremail(file)
#self.server.sendmail(send_user,user_list,self.message.as_string())
def close(self): #关闭邮件sever
print('mail class close')
self.server.close()
class PasswordManagerSystem(QMainWindow):
def __init__(self):
super(PasswordManagerSystem, self).__init__()
self.initToolbar()
self.initDB()
self.initGrid()
#self.current_row = 0
self.setGeometry(300, 300, 650, 300)
self.setWindowTitle('密码管理器')
self.setWindowIcon(QIcon(':images/logo.jpg'))
def initToolbar(self):
newAction = QAction(QIcon(':images/new.png'), 'New Ctrl+N', self)
editAction = QAction(QIcon(':images/edit.png'), 'Edit Ctrl+E', self)
delAction = QAction(QIcon(':images/del.png'), 'Delete', self)
backupAction = QAction(QIcon(':images/backup.png'), 'Backup Ctrl+B', self)
newAction.setShortcut('Ctrl+N')
editAction.setShortcut('Ctrl+E')
delAction.setShortcut('Delete')
backupAction.setShortcut('Ctrl+B')
newAction.triggered.connect(self.newAction_def)
editAction.triggered.connect(self.editAction_def)
delAction.triggered.connect(self.delAction_def)
backupAction.triggered.connect(self.backupAction_def)
self.tb_new = self.addToolBar('New')
self.tb_edit = self.addToolBar('Edit')
self.tb_del = self.addToolBar('Del')
self.tb_backup = self.addToolBar('Backup')
self.tb_new.addAction(newAction)
self.tb_edit.addAction(editAction)
self.tb_del.addAction(delAction)
self.tb_backup.addAction(backupAction)
def backupAction_def(self):
cur = self.conn.cursor()
cur.execute('SELECT * FROM INFO')
data = cur.fetchall()
print(len(data))
a=' '.join(['Website', 'Username', 'Password', 'Url'])+'\n'
for items in data:
for item in items:
a=a+str(item)+' '
a =a+'\n'
myEmail=SendEmail()
myEmail.sendmailforbackup(a,self.dbpath)
myEmail.close()
def initDB(self):
home = os.path.expanduser('~')
if '.PasswordManageSystem' not in os.listdir(home):
os.mkdir(os.path.join(home, '.PasswordManageSystem'))
self.dbpath = os.path.join(home, '.PasswordManageSystem', 'PasswordManagement.db')
if os.path.exists(self.dbpath):
self.conn = sqlite3.connect(self.dbpath)
self.conn.isolation_level = None
else:
self.conn = sqlite3.connect(dbpath)
self.conn.isolation_level = None
self.conn.execute('''CREATE TABLE INFO
(ID int PRIMARY KEY NOT NULL,
WEBSITE char(255),
USERNAME char(255),
PASSWORD char(255),
URL char(255))''')
cur = self.conn.cursor()
cur.execute('SELECT * FROM INFO')
self.displayData = cur.fetchall()
cur.close()
self.current_row = len(self.displayData)
def initGrid(self):
self.grid = QTableWidget()
self.setCentralWidget(self.grid)
self.grid.setColumnCount(4)
self.grid.setRowCount(0)
column_width = [75, 150, 270, 150]
for column in range(4):
self.grid.setColumnWidth(column, column_width[column])
headerlabels = ['Website', 'Username', 'Password', 'Url']
self.grid.setHorizontalHeaderLabels(headerlabels)
self.grid.setEditTriggers(QAbstractItemView.NoEditTriggers)
self.grid.setSelectionBehavior(QAbstractItemView.SelectRows)
for row in range(len(self.displayData)):#根据数据库数据的数目,决定循环次数
self.grid.insertRow(row) #插入一行
data = self.displayData[row] #获取一条数据
for i in range(4):
item=data[i+1] #获取一条数据中的一个元素
#print(row, i, item)
new_item = QTableWidgetItem(item) #将数据转化为QTableWidgetItem
self.grid.setItem(row, i, new_item) #插入数据
def newAction_def(self):
data = self.showDialog()
if data[0]:
self.current_row += 1
self.conn.execute("INSERT INTO INFO VALUES(%d, '%s', '%s', '%s', '%s')"
% (self.current_row, data[1], data[2], data[3], data[4]))
self.grid.insertRow(self.current_row - 1)
for i in range(4):
new_item = QTableWidgetItem(data[i + 1])
self.grid.setItem(self.current_row - 1, i, new_item)
def editAction_def(self):
selected_row = self.grid.selectedItems()
if selected_row:
edit_row = self.grid.row(selected_row[0])
old_data = []
for i in range(4):
old_data.append(self.grid.item(edit_row, i).text())
new_data = self.showDialog(*old_data)
if new_data[0]:
self.conn.execute('''UPDATE INFO SET
WEBSITE = '%s', USERNAME = '%s',
PASSWORD = '%s', URL = '%s'
WHERE ID = '%d' '''
% (new_data[1], new_data[2], new_data[3], new_data[4], edit_row + 1))
for i in range(4):
new_item = QTableWidgetItem(new_data[i + 1])
self.grid.setItem(edit_row, i, new_item)
else:
self.showHint()
def delAction_def(self):
selected_row = self.grid.selectedItems()
if selected_row:
del_row = self.grid.row(selected_row[0])
self.grid.removeRow(del_row)
self.conn.execute("DELETE FROM INFO WHERE ID = %d" % (del_row + 1))
for index in range(del_row + 2, self.current_row + 1):
self.conn.execute("UPDATE INFO SET ID = %d WHERE ID = %d" % ((index - 1), index))
self.current_row -= 1
else:
self.showHint()
def showHint(self):
hint_msg = QMessageBox()
hint_msg.setText('No selected row!')
hint_msg.addButton(QMessageBox.Ok)
hint_msg.exec_()
def showDialog(self, ws = '', un = '', pw = '', url = ''):
edit_dialog = QDialog(self)
group = QGroupBox('Edit Info', edit_dialog)
lbl_website = QLabel('Website:', group)
le_website = QLineEdit(group)
le_website.setText(ws)
lbl_username = QLabel('Username:', group)
le_username = QLineEdit(group)
le_username.setText(un)
lbl_password = QLabel('Password:', group)
le_password = QLineEdit(group)
le_password.setText(pw)
lbl_url = QLabel('Url:', group)
le_url = QLineEdit(group)
le_url.setText(url)
ok_button = QPushButton('OK', edit_dialog)
cancel_button = QPushButton('CANCEL', edit_dialog)
ok_button.clicked.connect(edit_dialog.accept)
ok_button.setDefault(True)
cancel_button.clicked.connect(edit_dialog.reject)
group_layout = QVBoxLayout()
group_item = [lbl_website, le_website,
lbl_username, le_username,
lbl_password, le_password,
lbl_url, le_url]
for item in group_item:
group_layout.addWidget(item)
group.setLayout(group_layout)
group.setFixedSize(group.sizeHint())
button_layout = QHBoxLayout()
button_layout.addWidget(ok_button)
button_layout.addWidget(cancel_button)
dialog_layout = QVBoxLayout()
dialog_layout.addWidget(group)
dialog_layout.addLayout(button_layout)
edit_dialog.setLayout(dialog_layout)
edit_dialog.setFixedSize(edit_dialog.sizeHint())
if edit_dialog.exec_():
website = le_website.text()
username = le_username.text()
password = le_password.text()
url = le_url.text()
return True, website, username, password, url
return False, None, None, None, None
class Login():
pass
if __name__ == '__main__':
app = QApplication(sys.argv)
pwk = PasswordManagerSystem()
pwk.show()
app.exec_()
pwk.conn.close()
sys.exit(0)