def configFilterTypeButtons(self,
name=u"default",
checked=False,
xPos=0,
yPos=0,
idButton=0):
button = QRadioButton(self)
button.setObjectName(name)
button.setEnabled(True)
button.setGeometry(QRect(xPos, yPos, 82, 17))
button.setCursor(QCursor(Qt.PointingHandCursor))
button.setChecked(checked)
lam = lambda id=idButton: self.buttonChecked(id)
self.connect(button, SIGNAL("clicked()"), lam)
self.filterTypeButtons.addButton(button, idButton)
class Music(QMainWindow):
def __init__(self):
super(Music, self).__init__()
self.setGeometry(20, 50, 522, 175)
self.setMinimumSize(522, 175)
self.setMaximumSize(522, 175)
self.setWindowTitle('Muse')
self.setWindowIcon(QIcon('arti.PNG'))
self.setFont(QFont('Roboto', 12))
palette = QPalette()
palette.setColor(palette.Window, QColor('#000000'))
palette.setColor(palette.WindowText, QColor('#FFFFFF'))
self.setPalette(palette)
self.menubar = QMenuBar(self)
self.menubar.setGeometry(0, 0, 682, 21)
self.menubar.setFont(QFont('Roboto', 10))
self.date_menu = QMenu(self.menubar)
self.date_menu.setTitle(str(datetime.now().strftime('%d-%m-%Y')))
self.setMenuBar(self.menubar)
self.menubar.addAction(self.date_menu.menuAction())
self.song = QLineEdit(self)
self.song.setPlaceholderText(
'Enter the name of song you want to search for:')
self.song.setGeometry(10, 30, 501, 31)
self.spotify = QRadioButton(self)
self.spotify.setText('Spotify')
self.spotify.setGeometry(120, 80, 101, 21)
self.gaana = QRadioButton(self)
self.gaana.setText('Gaana')
self.gaana.setGeometry(330, 80, 91, 21)
self.search = QPushButton(self)
self.search.setText('Search')
self.search.setGeometry(380, 130, 121, 31)
self.search.clicked.connect(lambda: self.on_click())
self.search.setCursor(QCursor(Qt.PointingHandCursor))
self.label = QLabel(self)
self.label.setGeometry(10, 130, 341, 31)
def on_click(self):
self.song_search = self.song.text()
if self.song.text():
if self.spotify.isChecked():
self.speak('Searching your song on Spotify.')
spotify_url = 'https://open.spotify.com/search/'
webbrowser.open(spotify_url + self.song_search)
elif self.gaana.isChecked():
self.speak('Searching your song on Gaana.')
gaana_url = 'https://gaana.com/search/'
webbrowser.open(gaana_url + self.song_search)
else:
self.speak('Please choose either Spotify or Gaana.')
else:
self.speak('Please type a song name first')
self.song.clear()
def speak(self, audio):
self.engine = pyttsx3.init('sapi5')
self.voices = self.engine.getProperty('voices')
self.engine.setProperty('voice', self.voices[1].id)
self.engine.setProperty('rate', 165)
self.engine.say(audio)
self.label.setText(audio)
self.engine.runAndWait()
class AddUserFace(QDialog):
def __init__(self, image):
super(AddUserFace, self).__init__()
self.setFixedSize(300, 275)
self.setWindowIcon(QIcon('icons/add.png'))
self.setWindowTitle('添加')
self.rb_select = QRadioButton("选择", self)
self.rb_select.setGeometry(70, 20, 50, 26)
self.rb_select.toggled.connect(self.toggle)
self.rb_new = QRadioButton('新建', self)
self.rb_new.setGeometry(120, 20, 50, 26)
self.rb_new.toggled.connect(self.toggle)
lbl_name = QLabel('名称', self)
lbl_name.setGeometry(10, 70, 50, 26)
lbl_name.setAlignment(Qt.AlignCenter)
users = DbHelper.query_users()
self.cb_user = QComboBox(self)
self.cb_user.setGeometry(70, 70, 200, 26)
self.le_user = QLineEdit(self)
self.le_user.setGeometry(70, 70, 200, 26)
if users is not None and len(users) > 0:
self.rb_select.setChecked(True)
for user in users:
self.cb_user.addItem(user[1], userData=user)
else:
self.rb_select.setDisabled(True)
self.rb_new.setChecked(True)
lbl_face = QLabel('人脸', self)
lbl_face.setGeometry(10, 140, 50, 26)
lbl_face.setAlignment(Qt.AlignCenter)
self.btn_save = QPushButton(self)
self.btn_save.setText('保存')
self.btn_save.setGeometry(10, 234, 280, 30)
self.btn_save.clicked.connect(self.save)
self.cache_faces = {}
self.face = None
faces = HaarcascadeDetective.get_faces(image)
index = 0
for face in faces:
viewer_face = QPushButton(self)
viewer_face.setGeometry(70 * (index + 1), 120, 60, 60)
viewer_face.setIconSize(QSize(56, 56))
img = cv2.cvtColor(face, cv2.COLOR_BGR2RGB)
image = QImage(img, img.shape[1], img.shape[0], img.shape[1] * 3,
QImage.Format_RGB888)
pix_map = QPixmap.fromImage(image)
viewer_face.setIcon(QIcon(pix_map))
viewer_face.clicked.connect(self.select_face)
self.cache_faces[viewer_face] = face
if index == 0:
self.face = face
viewer_face.setStyleSheet(
'border-color: rgb(255, 0, 0);border-style: outset;border-width: 2px;'
)
index += 1
if index > 3:
break
if index == 0:
Tool.show_msg_box("未检测到人脸信息")
def select_face(self):
sender = self.sender()
for btn in self.cache_faces:
btn.setStyleSheet('border-style: none')
sender.setStyleSheet(
"border-color: rgb(255, 0, 0);border-style: outset;border-width: 2px;"
)
self.face = self.cache_faces[btn]
def toggle(self):
if self.rb_new.isChecked():
self.cb_user.hide()
self.le_user.show()
elif self.rb_select.isChecked():
self.le_user.hide()
self.cb_user.show()
def save(self):
if self.face is None:
return
self.btn_save.setDisabled(True)
if self.rb_new.isChecked():
user_name = self.le_user.text().strip(' ')
if not user_name.replace(' ', '').encode("utf-8").isalpha():
Tool.show_msg_box('只支持英文字母和空格～')
self.le_name.setFocus()
return
user_id = DbHelper.insert_user(user_name)
else:
user_id = self.cb_user.currentData()[0]
if not os.path.exists("faces"):
os.mkdir("faces")
if not os.path.exists('faces/{}'.format(user_id)):
os.mkdir('faces/{}'.format(user_id))
face = 'faces/{}/{}.png'.format(user_id, time.time())
cv2.imwrite(face, self.face)
self.close()
