class QtRecorder(Recorder):
def __init__(self, output_path: str, parent: QWidget):
super().__init__(output_path)
from PyQt5.QtMultimedia import QAudioRecorder
self._recorder = QAudioRecorder(parent)
audio = self._recorder.audioSettings()
audio.setSampleRate(44100)
audio.setChannelCount(1)
self._recorder.setEncodingSettings(
audio,
self._recorder.videoSettings(),
"audio/x-wav",
)
self._recorder.setOutputLocation(QUrl.fromLocalFile(self.output_path))
self._recorder.setMuted(True)
def start(self, on_done: Callable[[], None]) -> None:
self._recorder.record()
super().start(on_done)
def stop(self, on_done: Callable[[str], None]):
self._recorder.stop()
super().stop(on_done)
def on_timer(self):
duration = self._recorder.duration()
if duration >= 300:
# disable mute after recording starts to avoid clicks/pops
if self._recorder.isMuted():
self._recorder.setMuted(False)
class QVoiceRecorder(object):
def __init__(self):
pass
def initAudioInput(self, filepath):
self.recorder = QAudioRecorder()
self.settings = QAudioEncoderSettings()
self.settings.setCodec("audio/vorbis")
self.recorder.setContainerFormat("ogg")
#self.settings.setQuality(QtMultimedia.HighQuality)
self.recorder.setEncodingSettings(self.settings)
url = QtCore.QUrl.fromLocalFile(
QtCore.QFileInfo(filepath).absoluteFilePath())
self.recorder.setOutputLocation(url)
def start(self):
self.recorder.record()
def stop(self):
self.recorder.pause()
self.recorder.stop()
self.recorder.stop()
class RecordAudio(object):
recorder = None
def __init__(self):
pass
def record(self, filename):
url = QtCore.QUrl.fromLocalFile(
QtCore.QFileInfo(filename).absoluteFilePath())
#content = QMediaObject(url)
#self.recorder = QAudioRecorder()
#source = QAudioInput()
#source = QMediaObject()
self.recorder = QAudioRecorder()
settings = QAudioEncoderSettings()
settings.setChannelCount(1)
settings.setSampleRate(44100)
settings.setBitRate(32)
settings.setCodec("audio/vorbis")
#settings.setEncodingMode(QtMultimedia.ConstantQualityEnconding)
self.recorder.setContainerFormat("ogg")
self.recorder.setEncodingSettings(settings)
self.recorder.setOutputLocation(url)
#container = None
#self.recorder.setEncodingSettings(settings, container)
def record(self):
self.recorder.record()
def stop(self):
self.recorder.stop()
class AudioRecorder(QWidget):
def __init__(self):
super().__init__()
self.initializeUI()
def initializeUI(self):
"""Initialize the window and display its contents to the screen."""
self.setFixedSize(360, 540)
self.setWindowTitle('9.1 - Audio Recorder')
self.audio_path = "" # Empty variable for path to audio file
self.setupWindow()
self.setupSystemTrayIcon()
self.show()
def setupWindow(self):
"""Set up widgets in the main window and the QAudioRecorder instance."""
# Set up two push buttons (the app's first "screen")
self.select_path_button = QPushButton("Select Audio Path")
self.select_path_button.setObjectName("SelectFile")
self.select_path_button.setFixedWidth(140)
self.select_path_button.clicked.connect(self.selectAudioPath)
self.start_button = QPushButton()
self.start_button.setObjectName("StartButton")
self.start_button.setEnabled(False)
self.start_button.setFixedSize(105, 105)
self.start_button.clicked.connect(self.startRecording)
# Set up the labels and stop button (the app's second "screen")
self.recording_label = QLabel("Recording...")
self.recording_label.setFont(QFont("Helvetica [Cronyx]", 32))
self.recording_label.setVisible(False)
self.recording_label.setAlignment(Qt.AlignHCenter)
self.time_label = QLabel("00:00")
self.time_label.setFont(QFont("Helvetica [Cronyx]", 18))
self.time_label.setObjectName("Time")
self.time_label.setVisible(False)
self.time_label.setAlignment(Qt.AlignHCenter)
self.stop_button = QPushButton()
self.stop_button.setObjectName("StopButton")
self.stop_button.setFixedSize(65, 65)
self.stop_button.setVisible(False)
self.stop_button.clicked.connect(self.stopRecording)
# Set up the main layout
self.main_v_box = QVBoxLayout()
self.main_v_box.setAlignment(Qt.AlignHCenter)
self.main_v_box.addWidget(self.select_path_button)
# Force select_path_button to be centered in the window
self.main_v_box.setAlignment(self.select_path_button, Qt.AlignCenter)
self.main_v_box.addStretch(3)
self.main_v_box.addWidget(self.start_button)
self.main_v_box.setAlignment(self.start_button, Qt.AlignCenter)
self.main_v_box.addWidget(self.recording_label)
self.main_v_box.addWidget(self.time_label)
self.main_v_box.addStretch(3)
self.main_v_box.addWidget(self.stop_button)
self.main_v_box.setAlignment(self.stop_button, Qt.AlignCenter)
self.main_v_box.addStretch(1)
self.setLayout(self.main_v_box) # Set the beginning layout
# Specify audio encoder settings
audio_settings = QAudioEncoderSettings()
# Depending upon your platform or the codecs that you have available, you
# will need to change the codec. For Linux users if you are having issues
# use "audio/x-vorbis", and then select the .ogg extension when saving
# the file
audio_settings.setCodec("audio/wav")
audio_settings.setQuality(QMultimedia.HighQuality)
# Create instance of QAudioRecorder for recording audio
self.audio_recorder = QAudioRecorder()
# Uncomment to discover possible codecs supported on your platform
#print(self.audio_recorder.supportedAudioCodecs())
self.audio_recorder.setEncodingSettings(audio_settings)
self.audio_recorder.durationChanged.connect(self.displayTime)
def setupSystemTrayIcon(self):
"""Set up system tray icon and context menu. User can re-open the window if
it was closed or quit the application using the tray menu."""
self.tray_icon = QSystemTrayIcon(
QIcon(":/resources/images/mic_icon.png"))
# Create the actions and context menu for the tray icon
tray_menu = QMenu()
open_act = tray_menu.addAction("Open")
open_act.triggered.connect(self.show)
tray_menu.addSeparator()
quit_act = tray_menu.addAction("Quit")
quit_act.triggered.connect(QApplication.quit)
self.tray_icon.setContextMenu(tray_menu)
self.tray_icon.show()
def selectAudioPath(self):
"""Open file dialog and choose the directory for saving the audio file."""
path, _ = QFileDialog.getSaveFileName(self, "Save Audio File",
os.getenv("HOME"), "WAV (*.wav)")
if path:
self.audio_path = path
self.start_button.setEnabled(True)
else:
QMessageBox.information(self, "Error", "No directory selected.",
QMessageBox.Ok)
def startRecording(self):
"""Set up the audio output file location, reset widget states.
Also starts the timer and begins recording. """
self.audio_recorder.setOutputLocation(
QUrl.fromLocalFile(self.audio_path))
# Set widget states
self.select_path_button.setVisible(False)
self.start_button.setVisible(False)
self.recording_label.setVisible(True)
self.time_label.setVisible(True)
self.time_label.setText("00:00") # Update the label
self.stop_button.setVisible(True)
# Start the timer and begin recording
self.audio_recorder.record()
def stopRecording(self):
"""Stop recording, stop the timer, and reset widget states."""
self.audio_recorder.stop()
# Reset widget states
self.select_path_button.setVisible(True)
self.start_button.setVisible(True)
self.recording_label.setVisible(False)
self.time_label.setVisible(False)
self.stop_button.setVisible(False)
def displayTime(self, duration):
"""Calculate the time displayed in the time_label widget."""
minutes, seconds = self.convertTotalTime(duration)
time_recorded = "{:02d}:{:02d}".format(minutes, seconds)
self.time_label.setText(time_recorded)
def convertTotalTime(self, time_in_milli):
"""Convert time from milliseconds."""
minutes = (time_in_milli / (1000 * 60)) % 60
seconds = (time_in_milli / 1000) % 60
return int(minutes), int(seconds)
def closeEvent(self, event):
"""Display a message in the system tray when the main window has been closed."""
self.tray_icon.showMessage("Notification",
"Audio Recorder is still running.", 8000)
class MainWindow_EXEC():
def __init__(self):
#-------------------Init QT Setup---------------------------
app = QtWidgets.QApplication(sys.argv)
self.MainWindow = QtWidgets.QMainWindow()
self.ui = Ui_MainWindow()
self.ui.setupUi(self.MainWindow)
app.setStyleSheet(qdarkstyle.load_stylesheet_pyqt5())
#------------------Exporting Setup------------------------------
self.ui.export_midi.clicked.connect(self.openDirectory_midi)
self.ui.export_midi.setFocusPolicy(QtCore.Qt.NoFocus)
self.ui.export_audio.clicked.connect(self.openDirectory_audio)
self.ui.export_audio.setFocusPolicy(QtCore.Qt.NoFocus)
#------------------Metronome Setup------------------------------
self.ui.metronome_button.clicked.connect(self.metro_thread)
#------------------Recording Setup------------------------------
self.ui.start_stop_rec.clicked.connect(self.start_stop_recording)
self.ui.play_gui.clicked.connect(self.play)
# QAudio setup
self.settings = QAudioEncoderSettings()
self.settings.setBitRate(16)
self.settings.setChannelCount(1)
self.audioRecorder = QAudioRecorder()
self.audioRecorder.setEncodingSettings(self.settings)
self.file_path = os.path.abspath(
os.path.join(os.path.dirname(__file__),
resource_path("resources/output.wav")))
self.url = QUrl.fromLocalFile(self.file_path)
self.audioRecorder.setOutputLocation(self.url)
#------------------Audio Terrain Gui Setup------------------------------
self.terrain = Terrain()
self.terrain.update()
self.terrain_widget = self.terrain.getwidget()
self.layout = QtGui.QGridLayout()
self.layout.setContentsMargins(0, 0, 0, 0)
self.ui.t_widget.setLayout(self.layout)
self.layout.addWidget(self.terrain_widget, 0, 0, 1, 1)
#------------------Audio Trimmer Setup------------------------------
self.ui.audio_trimmer.clicked.connect(self.trim_audio)
if os.path.isfile("resources/output.wav"):
self.y, self.sr = librosa.load(
resource_path("resources/output.wav"), sr=44100)
else:
new_wav = AudioSegment.empty()
new_wav.export("resources/output.wav", format="wav")
self.y, self.sr = librosa.load(
resource_path("resources/output.wav"), sr=44100)
self.duration = round(
librosa.core.get_duration(y=self.y, sr=self.sr) * self.sr)
self.maxv = np.iinfo(np.int16).max
self.win = pg.GraphicsLayoutWidget()
self.p = self.win.addPlot()
#removes X & Y Axis and disables mouse movement
self.p.showAxis('bottom', show=False)
self.p.showAxis('left', show=False)
self.p.setMouseEnabled(x=False, y=False)
self.region = pg.LinearRegionItem(brush=(100, 100, 100, 60),
bounds=(0, self.duration))
self.region.setRegion([0, self.duration])
self.p.addItem(self.region, ignoreBounds=True)
self.p.plot(self.y, pen="w")
self.layout.addWidget(self.win)
self.win.hide()
#------------------Midi Setup------------------------------
self.ui.convert_midi.clicked.connect(self.convertMidi)
self.ui.midi_play.clicked.connect(self.midiplayer_thread)
self.ui.tempo_slider.valueChanged[int].connect(self.tempo_value)
self.ui.midi_loop.toggle()
# default bpm is 120
self.current_tempo = 120
self.detected_tempo = 120
#------------------Drum Kit Selector Setup----------------------
self.ui.drum_kits.clicked.connect(self.select_drumkit)
self.drum_number = 0
self.drum_folders = [
'Drum_Kit_1', 'Drum_Kit_2', 'Drum_Kit_3', 'Drum_Kit_4'
]
self.drum_current = self.drum_folders[self.drum_number]
#------------------EXEC Window---------------------------------
self.MainWindow.show()
sys.exit(app.exec_())
#---------------------------------------------------------------
#------------------Functions----------------------------------
#------------------Drum Kit Selector------------------------------
def select_drumkit(self):
if self.drum_number < 3:
self.drum_number += 1
self.drum_current = self.drum_folders[self.drum_number]
self.ui.drum_kits.setText(self.drum_current.replace("_", " "))
else:
self.drum_number = 0
self.drum_current = self.drum_folders[self.drum_number]
self.ui.drum_kits.setText(self.drum_current.replace("_", " "))
#------------------Audio Trimmer------------------------------
def trim_audio(self):
# Switch to Trimmer widget
self.layout.removeWidget(self.terrain_widget)
self.terrain_widget.hide()
self.win.show()
self.trim_values = self.region.getRegion()
self.updateaudio()
# Trims signal array with region values
self.y = self.y[round(self.trim_values[0]):round(self.trim_values[1])]
# save the new signal values to wav
librosa.output.write_wav(resource_path("resources/output.wav"),
(self.y * self.maxv).astype(np.int16),
self.sr)
self.updateplot()
def updateplot(self):
# Replot the trimmed wav and update region bounds
self.duration = round(
librosa.core.get_duration(y=self.y, sr=self.sr) * self.sr)
self.p.plot(clear=True)
self.p.plot(self.y, pen="w")
self.region = pg.LinearRegionItem(brush=(100, 100, 100, 50),
bounds=(0, self.duration))
self.p.addItem(self.region, ignoreBounds=True)
self.region.setRegion([0, self.duration])
def updateaudio(self):
self.y, self.sr = librosa.load(resource_path("resources/output.wav"),
sr=44100)
#------------------Metronome Threading------------------------------
def metro_thread(self):
if self.ui.metronome_button.isChecked():
print('metronome is On')
self.thread = QThread(
) # a new thread to run our background tasks in
self.worker = Worker(
self.current_tempo) # a new worker to perform those tasks
self.worker.moveToThread(
self.thread
) # move the worker into the thread, do this first before connecting the signals
self.thread.started.connect(
self.worker.work
) # begin our worker object's loop when the thread starts running
self.thread.start()
else:
print('metronome is Off')
self.stop_loop()
self.worker.finished.connect(
self.loop_finished
) # do something in the gui when the worker loop ends
self.worker.finished.connect(
self.thread.quit) # tell the thread it's time to stop running
self.worker.finished.connect(self.thread.wait)
self.worker.finished.connect(
self.worker.deleteLater
) # have worker mark itself for deletion
self.thread.finished.connect(self.thread.deleteLater)
def stop_loop(self):
self.worker.working = False
def loop_finished(self):
# print('Worker Finished')
pass
#---------------------------------------------------------
#------------------ MIDI ------------------------------
def tempo_value(self, value):
self.current_tempo = value
def convertMidi(self):
self.ui.convert_midi.setEnabled(False)
self.thread2 = QThread()
self.worker2 = ConvertMidi_Worker()
self.worker2.moveToThread(self.thread2)
self.thread2.started.connect(self.worker2.work)
self.thread2.start()
self.worker2.finished.connect(self.convert_finished)
self.worker2.finished.connect(self.thread2.quit)
self.worker2.finished.connect(self.thread2.wait)
self.worker2.finished.connect(self.worker2.deleteLater)
self.thread2.finished.connect(self.thread2.deleteLater)
def convert_finished(self, tempo):
self.detected_tempo = tempo
self.ui.tempo_slider.setValue(self.detected_tempo)
self.ui.convert_midi.clearFocus()
self.ui.convert_midi.setEnabled(True)
print('Midi Conversion finished')
def midiplayer_thread(self):
if self.ui.midi_play.isChecked() and self.ui.midi_loop.isChecked(
) == False:
self.ui.midi_play.setEnabled(False)
self.win.hide()
self.terrain_widget.show()
self.terrain.animate()
self.thread3 = QThread()
self.worker3 = MidiPlayer_Worker(self.current_tempo,
self.drum_current)
self.worker3.moveToThread(self.thread3)
self.thread3.started.connect(self.worker3.workonce)
self.thread3.start()
self.worker3.finished2.connect(self.midi_loop_finished2)
self.worker3.finished2.connect(self.thread3.quit)
self.worker3.finished2.connect(self.thread3.wait)
self.worker3.finished2.connect(self.worker3.deleteLater)
self.thread3.finished.connect(self.thread3.deleteLater)
elif self.ui.midi_play.isChecked() and self.ui.midi_loop.isChecked(
) == True:
self.win.hide()
self.terrain_widget.show()
self.start_Midi_Thread()
self.terrain.animate()
elif self.ui.midi_play.isChecked() == False:
self.terrain.stop_animate()
self.stop_Midi_Thread()
def start_Midi_Thread(self):
self.thread3 = QThread()
self.worker3 = MidiPlayer_Worker(self.current_tempo, self.drum_current)
self.worker3.moveToThread(self.thread3)
self.thread3.started.connect(self.worker3.work)
self.thread3.start()
def stop_Midi_Thread(self):
self.worker3.working = False
self.worker3.stop()
self.worker3.finished.connect(self.midi_loop_finished)
self.worker3.finished.connect(self.thread3.quit)
self.worker3.finished.connect(self.thread3.wait)
self.worker3.finished.connect(self.worker3.deleteLater)
self.thread3.finished.connect(self.thread3.deleteLater)
print('done')
def midi_loop_finished(self):
print('Midi loop Finished')
def midi_loop_finished2(self):
print('Midi Player Finished')
self.ui.midi_play.toggle()
self.ui.midi_play.setEnabled(True)
self.terrain.stop_animate()
#---------------------------------------------------------
#------------------ Recorder & Player ------------------------------
def start_stop_recording(self):
if self.ui.start_stop_rec.isChecked():
self.ui.play_gui.setEnabled(False)
self.ui.audio_trimmer.setEnabled(False)
self.win.hide()
self.terrain_widget.show()
self.layout.addWidget(self.terrain_widget)
self.audioRecorder.record()
self.terrain.update()
self.terrain.animate()
print('Recording...')
else:
self.ui.play_gui.setEnabled(True)
self.ui.audio_trimmer.setEnabled(True)
self.terrain.stop_animate()
self.audioRecorder.stop()
self.layout.removeWidget(self.terrain_widget)
self.terrain_widget.hide()
self.updateaudio()
self.win.show()
self.updateplot()
print('Stop Recording')
def play(self):
if self.ui.play_gui.isChecked():
self.win.hide()
self.terrain_widget.show()
self.player = QSound(resource_path("resources/output.wav"))
self.terrain.animate()
self.player.play()
# if self.player.isFinished():
# self.ui.play_gui.toggle()
# print('done')
else:
self.terrain.stop_animate()
self.player.stop()
self.player.deleteLater()
#------------------ Exporting ------------------------------
def openDirectory_midi(self):
self.openDirectoryDialog = QtGui.QFileDialog.getExistingDirectory(
self.MainWindow, "Save Midi File")
if self.openDirectoryDialog:
self.saveMidi(self.openDirectoryDialog)
else:
pass
def openDirectory_audio(self):
self.openDirectoryDialog = QtGui.QFileDialog.getExistingDirectory(
self.MainWindow, "Save Audio File")
if self.openDirectoryDialog:
self.saveAudio(self.openDirectoryDialog)
else:
pass
def saveMidi(self, directory):
shutil.copy("resources/beatbox.mid", directory)
def saveAudio(self, directory):
shutil.copy("resources/output.wav", directory)
class App(MainWidget):
client = Client('localhost', 5000)
client_thread = ClientThread(client) # 后台线程
def __init__(self):
super(App, self).__init__()
self.loginButton.clicked.connect(self.showLoginDialog)
self.client_thread.text_signal.connect(self.showText) # 显示文本消息
self.client_thread.usr_signal.connect(self.showUserList) # 更新在线用户
self.client_thread.file_signal.connect(self.showFile) # 显示文件消息
self.emojis.emoji_signal.connect(self.addEmoji)
# 通过QListWidget的当前item变化来切换QStackedWidget中的序号
self.userListWidget.currentRowChanged.connect(self.dialogChanged)
self.usrList = [] # 保存上一次的在线用户列表
self.groupList = [] # 群组列表
self.md5 = hashlib.md5() # 用于加密密码
# 录音机
self.recorder = QAudioRecorder(self)
settings = QAudioEncoderSettings()
settings.setChannelCount(2)
settings.setSampleRate(16000)
self.recorder.setEncodingSettings(settings)
# 处理对话用户改变
def dialogChanged(self, i):
# 通过QListWidget的当前item变化来切换QStackedWidget中的序号
self.stackedWidget.setCurrentIndex(i)
# enable当前chatWidget的所有按钮
self.stackedWidget.currentWidget().emojiButton.setEnabled(True)
self.stackedWidget.currentWidget().sendButton.setEnabled(True)
self.stackedWidget.currentWidget().fileButton.setEnabled(True)
self.stackedWidget.currentWidget().voiceButton.setEnabled(True)
try:
# 设置图标
self.userListWidget.currentItem().setIcon(
qta.icon('fa.check-circle', color='lightgreen'))
except:
pass
# 显示登陆对话框
def showLoginDialog(self):
# 登录对话框
self.d = LoginDialog()
self.dialog = QDialog()
self.d.setupUi(self.dialog)
self.dialog.setWindowFlag(QtCore.Qt.FramelessWindowHint)
self.dialog.show()
# 点击确认就调用login()进行登录操作
self.d.buttonBox.accepted.connect(self.login)
# 登录的相关操作
def login(self):
passwd = self.d.lineEdit_2.text()
if len(passwd) == 0:
print('密码不能为空！')
QMessageBox.critical(self, "错误", "密码不能为空！", QMessageBox.Ok)
return
try:
self.client.connect()
except:
print('连接服务器失败！')
QMessageBox.critical(self, "错误", "连接服务器失败！", QMessageBox.Ok)
return
self.client_thread.start() # 启动线程
self.name = self.d.lineEdit.text() # 用户名
self.client.name = self.name
# md5加密密码
self.md5.update(passwd.encode('utf-8'))
hashed_passwd = self.md5.hexdigest()
data = {'name': self.name, 'passwd': hashed_passwd}
self.nameLabel.setText('<u>' + self.name)
self.client.add_msg(msgtype.LOGIN, '', data) # 登录信息发给服务器验证
self.stackedWidget.removeWidget(
self.stackedWidget.findChild(ChatWidget,
name='unlogin')) # 移除未登录显示的界面
# 关闭登录按钮
self.loginButton.close()
# 为用户建立文件夹
if not os.path.exists('./temp/%s' % self.client.name):
os.makedirs('./temp/%s' % self.client.name)
# 发送文字消息
def sendText(self):
text = self.stackedWidget.currentWidget().msgEdit.toPlainText()
if not len(text):
return # 长度为零，不发送
try:
to = self.userListWidget.selectedItems()[0].text()
except: # 出现异常是因为没有选择对话用户
return
if to != self.name:
data = {'to': to, 'text': text}
self.client.add_msg(msgtype.TEXT, '', data)
self.stackedWidget.currentWidget().msgEdit.clear() # 清空输入框
# 显示在聊天记录中
item = QListWidgetItem()
item.setText(text)
item.setBackground(QColor('#F19483'))
item.setTextAlignment(Qt.AlignRight)
self.stackedWidget.currentWidget().msgList.addItem(item)
# 显示收到的文字消息
def showText(self, msg):
from_user = msg['from']
text = msg['text']
item = QListWidgetItem()
if from_user in self.usrList: # 一对一
item.setText(text)
elif from_user in self.groupList: # 群聊
user = msg['user'] # 发言用户
if user == self.name:
return
item.setText('<%s>' % user + text)
item.setTextAlignment(Qt.AlignLeft)
item.setBackground(QColor('#F7D420'))
chat = self.stackedWidget.findChild(
ChatWidget, name=from_user) # 找到对应的chatWidget插入新的消息
chat.msgList.addItem(item)
chat.msgList.setCurrentRow(chat.msgList.count() - 1)
if from_user in self.usrList:
if self.stackedWidget.currentIndex(
) != self.usrList.index(from_user) + len(self.groupList):
# 如果当前没有和该用户对话，就显示未读图标，并弹出通知
user = self.userListWidget.item(
self.usrList.index(from_user) + len(self.groupList))
user.setIcon(qta.icon('fa.circle', color='orange'))
self.tray.showMessage(
from_user, text, qta.icon('fa5.comment-alt',
color='white'))
elif from_user in self.groupList:
if self.stackedWidget.currentIndex() != self.groupList.index(
from_user):
# 如果当前没有看这个群聊，就显示未读图标，并弹出通知
user = self.userListWidget.item(
self.groupList.index(from_user))
user.setIcon(qta.icon('fa.circle', color='orange'))
self.tray.showMessage(
from_user, text, qta.icon('fa5.comment-alt',
color='white'))
# 更新在线用户列表和对应的chatWidget
def showUserList(self, usr_list):
users = usr_list['users']
groups = usr_list['groups']
# 群组
for i in groups:
if i not in self.groupList:
# 添加右侧的chatWidget
chatWidget = ChatWidget()
chatWidget.setObjectName(i)
chatWidget.emojiButton.clicked.connect(self.emojis.show)
chatWidget.sendButton.clicked.connect(self.sendText)
chatWidget.fileButton.clicked.connect(self.sendFile)
chatWidget.voiceButton.pressed.connect(self.startRecord)
chatWidget.voiceButton.released.connect(self.stopRecord)
self.stackedWidget.addWidget(chatWidget)
# 添加到左侧的用户列表
item = QListWidgetItem(i)
item.setSizeHint(QSize(16777215, 60))
item.setTextAlignment(Qt.AlignCenter)
item.setText(i)
item.setIcon(qta.icon('fa.check-circle', color='lightgreen'))
self.userListWidget.addItem(item)
# 清空用户列表
for i in range(len(groups), self.userListWidget.count()):
self.userListWidget.takeItem(len(groups))
# 用户
for i in users:
# 添加到用户列表
item = QListWidgetItem(i)
item.setSizeHint(QSize(16777215, 60))
item.setTextAlignment(Qt.AlignCenter)
item.setText(i)
item.setIcon(qta.icon('fa.check-circle', color='lightgreen'))
self.userListWidget.addItem(item)
# 对于下线的用户，移除对应的chatWidget
for i in self.stackedWidget.findChildren(ChatWidget):
if i.objectName() not in users and i.objectName() not in groups:
self.stackedWidget.removeWidget(i)
for i in users:
if i not in self.usrList: # 新用户
# 添加右侧的chatWidget
chatWidget = ChatWidget()
chatWidget.setObjectName(i)
chatWidget.emojiButton.clicked.connect(self.emojis.show)
chatWidget.sendButton.clicked.connect(self.sendText)
chatWidget.fileButton.clicked.connect(self.sendFile)
chatWidget.voiceButton.pressed.connect(self.startRecord)
chatWidget.voiceButton.released.connect(self.stopRecord)
self.stackedWidget.addWidget(chatWidget)
self.usrList = users # 更新用户列表
self.groupList = groups # 更新群组列表
# 发送文件
def sendFile(self):
fileName, filetype = QFileDialog.getOpenFileName(self, "选取文件", "./")
if len(fileName) == 0:
return
to = self.userListWidget.selectedItems()[0].text() # 发送给的用户
if to != self.name:
self.client.add_msg(msgtype.FILE, '', {
'to': to,
'filename': fileName
})
# 文件显示在聊天记录中
item = QListWidgetItem()
item.setSizeHint(QSize(200, 80))
fileWidget = FileWidget(fileName, 0)
item.setBackground(QColor('#F19483'))
fileWidget.fileButton.setObjectName(fileName)
fileWidget.fileButton.clicked.connect(self.openFile)
self.stackedWidget.currentWidget().msgList.addItem(item)
self.stackedWidget.currentWidget().msgList.setItemWidget(
item, fileWidget)
# 显示收到的文件
def showFile(self, msg):
from_user = msg['from']
filename = msg['filename']
chat = self.stackedWidget.findChild(
ChatWidget, name=from_user) # 找到对应的chatWidget插入新的消息
if from_user in self.usrList: # 一对一
if self.stackedWidget.currentIndex(
) != self.usrList.index(from_user) + len(self.groupList):
# 如果当前没有和该用户对话，就显示未读图标
user = self.userListWidget.item(
self.usrList.index(from_user) + len(self.groupList))
user.setIcon(qta.icon('fa.circle', color='orange'))
self.tray.showMessage(from_user, filename,
qta.icon('fa5.file', color='white'))
elif from_user in self.groupList: # 群聊
item = QListWidgetItem()
item.setText('<%s>' % msg['user'])
item.setBackground(QColor('#F7D420'))
item.setSizeHint(QSize(200, 40))
chat.msgList.addItem(item)
if self.stackedWidget.currentIndex() != self.groupList.index(
from_user):
user = self.userListWidget.item(
self.groupList.index(from_user))
user.setIcon(qta.icon('fa.circle', color='orange'))
self.tray.showMessage(from_user, filename,
qta.icon('fa5.file', color='white'))
# 文件显示在聊天记录中
item = QListWidgetItem()
item.setBackground(QColor('#F7D420'))
item.setSizeHint(QSize(200, 80))
fileWidget = FileWidget(filename, 1)
fileWidget.fileButton.setObjectName('./temp/%s/' % self.name +
filename)
if filename.endswith('.wav'): # 语音消息
fileWidget.fileButton.clicked.connect(self.playVoice)
else:
fileWidget.fileButton.clicked.connect(self.openFile)
chat.msgList.addItem(item)
chat.msgList.setItemWidget(item, fileWidget)
# 打开点击的文件
def openFile(self):
filename = self.sender().objectName()
QDesktopServices.openUrl(QUrl.fromLocalFile(filename))
# 开始录音
def startRecord(self):
print('开始录音...')
self.stackedWidget.currentWidget().voiceButton.setIcon(
qta.icon('fa.spinner')) # 设置图标
# 设置文件名
self.voiceFileName = os.path.abspath('.') + '\\temp\\%s\\%s.wav' % (
self.name, time.strftime('%Y%m%d%H%M%S', time.localtime()))
print(self.voiceFileName)
self.recorder.setOutputLocation(QUrl.fromLocalFile(self.voiceFileName))
self.recorder.record() # 开始录音
# 停止录音并发送
def stopRecord(self):
self.stackedWidget.currentWidget().voiceButton.setIcon(
qta.icon('fa.microphone'))
self.recorder.stop()
print('录音结束')
to = self.userListWidget.selectedItems()[0].text()
if to != self.name:
self.client.add_msg(msgtype.FILE, '', {
'to': to,
'filename': self.voiceFileName
})
item = QListWidgetItem()
item.setSizeHint(QSize(200, 80))
item.setBackground(QColor('#F19483'))
fileWidget = FileWidget(self.voiceFileName, 0)
fileWidget.fileButton.setObjectName(self.voiceFileName)
fileWidget.fileButton.clicked.connect(self.playVoice)
self.stackedWidget.currentWidget().msgList.addItem(item)
self.stackedWidget.currentWidget().msgList.setItemWidget(
item, fileWidget)
# 播放语音
def playVoice(self):
print('playing ' + self.sender().objectName())
sound = QtMultimedia.QSound(self.sender().objectName()) # 取得文件名
sound.play()
loop = QEventLoop()
loop.exec()
# 向输入框中添加选中的表情
def addEmoji(self, emoji):
self.stackedWidget.currentWidget().msgEdit.setText(
self.stackedWidget.currentWidget().msgEdit.toPlainText() + emoji)