class SettingsFormVK(QWidget):
updateUIuserSignal = pyqtSignal(bool, dict)
setPreloaderStateSignal = pyqtSignal(bool)
def __init__(self):
QWidget.__init__(self)
self.ui = Ui_settingsFormVK()
self.ui.setupUi(self)
self.updateUIuserSignal.connect(self.updateUIuser)
self.setPreloaderStateSignal.connect(self.setPreloaderState)
self.settings = QSettings()
self.settings.beginGroup("app")
self.vkHandle = VK()
cookie = unicode(self.settings.value("userVK/cookie").toString())
if cookie:
self.vkHandle.set_cookie(cookie)
self.preloader = PreloaderWidget(self)
self.updateUI()
def updateUI(self):
self.updateUIvk()
def updateUIvk(self):
userName = unicode(self.settings.value("userVK/username").toString())
cookie = unicode(self.settings.value("userVK/cookie").toString())
if userName and cookie:
self.ui.userNameLabel.setText(userName)
self.ui.logoffVkButton.setEnabled(True)
self.ui.loginVkButton.setEnabled(False)
else:
self.ui.loginVkButton.setEnabled(True)
self.ui.logoffVkButton.setEnabled(False)
if not self.vkHandle.check_connection():
self.ui.statusLabel.setText(u"Невозможно установить соединение")
else:
self.thread = GenericThread(self.getUserInfo)
self.thread.start()
@pyqtSlot(bool)
def setPreloaderState(self, state):
if state:
self.preloader.start()
else:
self.preloader.stop()
def getUserInfo(self):
self.setPreloaderStateSignal.emit(True)
try:
if self.checkCookies():
vkuser = self.vkHandle.getUser()
if vkuser:
self.updateUIuserSignal.emit(True, vkuser)
self.setPreloaderStateSignal.emit(False)
else:
self.updateUIuserSignal.emit(False, {})
self.setPreloaderStateSignal.emit(False)
finally:
self.setPreloaderStateSignal.emit(False)
@pyqtSlot(bool, dict)
def updateUIuser(self, authorized, userInfo):
if authorized:
userName = userInfo.get("username", u"")
self.settings.setValue("userVK/username", userName)
self.ui.userNameLabel.setEnabled(True)
self.ui.userNameLabel.setText(userName)
self.ui.logoffVkButton.setEnabled(True)
self.ui.loginVkButton.setEnabled(False)
self.ui.statusLabel.setText(u"Авторизован")
else:
self.ui.userNameLabel.setEnabled(False)
self.ui.userNameLabel.setText(u"")
self.ui.statusLabel.setText(u"Не авторизован")
self.ui.logoffVkButton.setEnabled(False)
self.ui.loginVkButton.setEnabled(True)
def checkCookies(self):
if self.vkHandle.check_connection():
if self.vkHandle.is_logged():
return True
else:
self.flushUserInfo()
return False
def flushUserInfo(self):
self.settings.setValue("userVK/username", "")
self.settings.setValue("userVK/cookie", "")
self.vkHandle.set_cookie("")
@pyqtSlot(str)
def accept_cookie(self, cookie):
self.settings.setValue("userVK/cookie", unicode(cookie))
self.vkHandle.set_cookie(cookie)
self.updateUI()
@pyqtSlot()
def on_loginVkButton_clicked(self):
self.loginVkWindow = LoginVk()
self.loginVkWindow.acceptCookieSignal.connect(self.accept_cookie)
self.loginVkWindow.show()
@pyqtSlot()
def on_logoffVkButton_clicked(self):
self.flushUserInfo()
self.updateUI()
class SingletonDownload(QThread):
def __new__(cls, creator, *args, **kwargs):
if not hasattr(cls, 'instance'):
cls.instance = super(SingletonDownload, cls).__new__(cls)
cls.instance.__init__(creator, *args, **kwargs)
return cls.instance
def __init__(self, creator, *args, **kwargs):
QThread.__init__(self)
self.creator = creator
self.args = args
self.kwargs = kwargs
self.settings = QSettings()
self.settings.beginGroup('app')
self.vkHandle = VK()
self.musicLibHandle = MusicLib()
def run(self):
try:
current_track = self.creator.current_track
if not 'artist' in current_track or not 'title' in current_track:
return
if not self.settings.contains('userVK/cookie'):
self.setStatusLabelTextSignal.emit(u'Не авторизован ВКонтакте, проверьте настройки')
return
self.creator.setProgressBarValueSignal.emit(5)
if not self.vkHandle.check_connection():
self.creator.setStatusLabelTextSignal.emit(u'Не удается соединиться с сервером ВКонтакте')
return
_cookie = self.settings.value('userVK/cookie').toString()
self.vkHandle.set_cookie(_cookie)
if not self.vkHandle.is_logged():
self.creator.setStatusLabelTextSignal.emit(u'Авторизационные данные ВКонтакте не корректны, проверьте настройки')
return
self.creator.setProgressBarValueSignal.emit(10)
_song = self.vkHandle.search_best(current_track['artist'], current_track['title'])
if not _song:
self.creator.setStatusLabelTextSignal.emit(u'Трек не найден в базе ВКонтакте')
else:
recieved_bytes, filetmp_path = self.downloadtemp(_song['url'])
if filetmp_path:
try:
file_tags = Tag(filetmp_path)
except ValueError:
pass
else:
file_tags.flush_tags()
file_tags.set_tags(**current_track)
song_title = u'{0} - {1}.mp3'.format(current_track['artist'], current_track['title'])
save_path = self.musicLibHandle.get_path()
if not os.path.exists(save_path):
os.mkdir(save_path)
shutil.move(filetmp_path, self.musicLibHandle.get_path() + '/' + song_title)
self.creator.setStatusLabelTextSignal.emit(u'Загрузка файла завершена')
else:
self.creator.setStatusLabelTextSignal.emit(u'Не удалось загрузить файл')
finally:
self.creator.setProgressBarValueSignal.emit(0)
self.creator.setStateDownloadButtonSignal.emit(True)
def downloadfile(self, url, filename, dirname='.', fd=None):
chunk_size = 40960
file_path = os.path.join(dirname, filename)
file_url = urllib2.urlopen(url)
total_size = file_url.info().getheader('Content-Length').strip()
total_size = int(total_size)
recieved_bytes = 0
if not fd:
file_data = open(file_path, 'wb')
else:
file_data = os.fdopen(fd, 'wb')
while True:
chunk = file_url.read(chunk_size)
if not chunk:
if total_size == recieved_bytes:
file_data.close()
return recieved_bytes, file_path
return None
file_data.write(chunk)
recieved_bytes += len(chunk)
percent = (float(recieved_bytes) / float(total_size) * 90) + 10
self.creator.setProgressBarValueSignal.emit(percent)
def downloadtemp(self, url):
audio_file_fd, audio_file_path = mkstemp(prefix='vkaudio', suffix='.mp3')
audio_file_dir = os.path.dirname(audio_file_path)
audio_file_name = os.path.basename(audio_file_path)
recieved_bytes, file_path = self.downloadfile(url, audio_file_name,
dirname=audio_file_dir,
fd=audio_file_fd)
if recieved_bytes and file_path:
return recieved_bytes, file_path
return None