#!/usr/bin/env?python3
#CSDN：https://blog.csdn.net/m0_38106923
#郵箱：www.1345070491@qq.com
import?cv2
import?numpy?as?np

from?PyQt5.QtCore?import?pyqtSignal
from?PyQt5.QtGui?import?QIcon?QTextCursor
from?PyQt5.QtWidgets?import?QApplication?QWidget?QMessageBox?QTableWidgetItem?QAbstractItemView
from?PyQt5.uic?import?loadUi

import?logging
import?logging.config
import?os
import?shutil
import?sqlite3
import?sys
import?threading
import?multiprocessing

from?datetime?import?datetime


#?自定義數據庫記錄不存在異常
class?RecordNotFound（Exception）:
????pass


class?DataManageUI（QWidget）:
????logQueue?=?multiprocessing.Queue（）??#?日志隊列
????receiveLogSignal?=?pyqtSignal（str）??#?日志信號

????def?__init__（self）:
????????super（DataManageUI?self）.__init__（）
????????loadUi（‘./ui/DataManage.ui‘?self）
????????self.setWindowIcon（QIcon（‘./icons/icon.png‘））
????????self.setFixedSize（931?577）

????????#?設置tableWidget只讀，不允許修改
????????self.tableWidget.setEditTriggers（QAbstractItemView.NoEditTriggers）

????????#?數據庫
????????self.database?=?‘./Facebase.db‘
????????self.datasets?=?‘./datasets‘
????????self.isDbReady?=?False
????????self.initDbButton.clicked.connect（self.initDb）

????????#?用戶管理
????????self.queryUserButton.clicked.connect（self.queryUser）
????????self.deleteUserButton.clicked.connect（self.deleteUser）

????????#?直方圖均衡化
????????self.isEqualizeHistEnabled?=?False
????????self.equalizeHistCheckBox.stateChanged.connect（
????????????lambda:?self.enableEqualizeHist（self.equalizeHistCheckBox））

????????#?訓練人臉數據
????????self.trainButton.clicked.connect（self.train）

????????#?系統日志
????????self.receiveLogSignal.connect（lambda?log:?self.logOutput（log））
????????self.logOutputThread?=?threading.Thread（target=self.receiveLog?daemon=True）
????????self.logOutputThread.start（）

????#?是否執行直方圖均衡化
????def?enableEqualizeHist（self?equalizeHistCheckBox）:
????????if?equalizeHistCheckBox.isChecked（）:
????????????self.isEqualizeHistEnabled?=?True
????????else:
????????????self.isEqualizeHistEnabled?=?False

????#?初始化/刷新數據庫
????def?initDb（self）:
????????#?刷新前重置tableWidget
????????while?self.tableWidget.rowCount（）?>?0:
????????????self.tableWidget.removeRow（0）
????????try:
????????????if?not?os.path.isfile（self.database）:
????????????????raise?FileNotFoundError

????????????conn?=?sqlite3.connect（self.database）
????????????cursor?=?conn.cursor（）

????????????res?=?cursor.execute（‘SELECT?*?FROM?users‘）
????????????for?row_index?row_data?in?enumerate（res）:
????????????????self.tableWidget.insertRow（row_index）
????????????????for?col_index?col_data?in?enumerate（row_data）:
????????????????????self.tableWidget.setItem（row_index?col_index?QTableWidgetItem（str（col_data）））
????????????cursor.execute（‘SELECT?Count（*）?FROM?users‘）
????????????result?=?cursor.fetchone（）
????????????dbUserCount?=?result[0]
????????except?FileNotFoundError:
????????????logging.error（‘系統找不到數據庫文件{}‘.format（self.database））
????????????self.isDbReady?=?False
????????????self.initDbButton.setIcon（QIcon（‘./icons/error.png‘））

?屬性????????????大小?????日期????時間???名稱
-----------?---------??----------?-----??----
?????目錄???????????0??2019-03-29?13:52??Python+OpenCv實現AI人臉識別身份認證系統（3）——訓練人臉識別模型\
?????目錄???????????0??2019-03-29?13:52??Python+OpenCv實現AI人臉識別身份認證系統（3）——訓練人臉識別模型\config\
?????文件?????????380??2018-05-09?11:07??Python+OpenCv實現AI人臉識別身份認證系統（3）——訓練人臉識別模型\config\logging.cfg
?????文件?????????240??2018-05-09?11:07??Python+OpenCv實現AI人臉識別身份認證系統（3）——訓練人臉識別模型\config\telegramBot.cfg.bak
?????文件???????13457??2019-03-29?13:48??Python+OpenCv實現AI人臉識別身份認證系統（3）——訓練人臉識別模型\dataManage.py
?????目錄???????????0??2019-03-29?13:52??Python+OpenCv實現AI人臉識別身份認證系統（3）——訓練人臉識別模型\icons\
?????文件????????1343??2018-05-09?11:07??Python+OpenCv實現AI人臉識別身份認證系統（3）——訓練人臉識別模型\icons\error.png
?????文件????????1905??2018-05-09?11:07??Python+OpenCv實現AI人臉識別身份認證系統（3）——訓練人臉識別模型\icons\icon.png
?????文件????????1409??2018-05-09?11:07??Python+OpenCv實現AI人臉識別身份認證系統（3）——訓練人臉識別模型\icons\success.png
?????文件?????????772??2018-05-09?11:07??Python+OpenCv實現AI人臉識別身份認證系統（3）——訓練人臉識別模型\icons\warning.png
?????目錄???????????0??2019-03-29?13:51??Python+OpenCv實現AI人臉識別身份認證系統（3）——訓練人臉識別模型\ui\
?????文件???????11578??2018-05-09?11:07??Python+OpenCv實現AI人臉識別身份認證系統（3）——訓練人臉識別模型\ui\DataManage.ui