def update(self):
self.is_update_running = True
self.stop_uptodate_thread()
self.btnUpdate.hide()
self.btnCancel.show()
self.btnSetRootFolder.hide()
self.set_progress_bar(0, 0, "Downloading: %p%")
self.build_loader = BuildLoader(
self, self.uptodate_thread.download_url, self.uptodate_thread.strptime)
self.build_loader.finished.connect(self.finished)
self.build_loader.progress_changed.connect(self.set_progress_bar)
self.build_loader.block_abortion.connect(lambda: self.btnCancel.hide())
self.btnCancel.clicked.connect(self.build_loader.stop)
self.build_loader.start()
def update(self):
self.is_update_running = True
self.uptodate_thread.is_running = False
self.uptodate_thread.terminate()
self.uptodate_thread.wait()
self.btnUpdate.hide()
self.btnCancel.show()
self.btnSetRootFolder.setEnabled(False)
self.set_progress_bar(0, "Downloading: %p%")
self.build_loader = BuildLoader(
self.settings.value('root_folder'), self.uptodate_thread.download_url)
self.build_loader.finished.connect(self.finished)
self.build_loader.progress_changed.connect(self.set_progress_bar)
self.btnCancel.clicked.connect(self.build_loader.stop)
self.build_loader.start()
class B3dVersionMangerMainWindow(QMainWindow,
main_window_design.Ui_MainWindow):
def __init__(self, app):
super().__init__()
self.app = app
self.setupUi(self)
# Read icons
self.app_icon = QIcon(":/icons/app.svg")
self.star_icon = QIcon(":/icons/star.svg")
self.star_inv_icon = QIcon(":/icons/star_inv.svg")
self.trash_icon = QIcon(":/icons/delete.svg")
self.quit_icon = QIcon(":/icons/quit_inv.svg")
self.fake_icon = QIcon(":/icons/fake.svg")
# Read settings
self.settings = QSettings('b3d_version_manager', 'settings')
is_register_blend = self.settings.value('is_register_blend', type=bool)
self.is_run_minimized = self.settings.value('is_run_minimized',
type=bool)
is_run_on_startup = self.settings.value('is_run_on_startup', type=bool)
root_folder = self.settings.value('root_folder')
if (not root_folder) or (not os.path.isdir(root_folder)):
exe_path = os.path.dirname(sys.executable)
self.settings.setValue('root_folder', exe_path)
# Custom title bar
self.title.setText("%s %s" % (QApplication.applicationName(),
QApplication.applicationVersion()))
self.btnClose.clicked.connect(self.close)
self.btnMinimize.clicked.connect(self.showMinimized)
# Custom menu bar
self.actionToggleRegisterBlend.setChecked(is_register_blend)
self.actionToggleRegisterBlend.triggered.connect(
self.toggle_register_blend)
self.actionToggleRunMinimized.setChecked(self.is_run_minimized)
self.actionToggleRunMinimized.triggered.connect(
self.toggle_run_minimized)
self.actionToggleRunOnStartup.setChecked(is_run_on_startup)
self.actionToggleRunOnStartup.triggered.connect(
self.toggle_run_on_startup)
self.actionQuit.triggered.connect(self.quit)
self.menubar.hide()
self.btnFile.setMenu(self.menuFile)
# Root folder layout
self.labelRootFolder.setText(self.settings.value('root_folder'))
self.btnSetRootFolder.clicked.connect(self.set_root_folder)
self.btnOpenRootFolder.clicked.connect(
lambda: os.startfile(self.settings.value('root_folder')))
# Tray layout
self.blender_action = QAction(self.star_inv_icon, "Blender", self)
show_action = QAction("Show", self)
hide_action = QAction("Hide", self)
quit_action = QAction(self.quit_icon, "Quit", self)
self.blender_action.triggered.connect(self.open_latest_b3d)
show_action.triggered.connect(self.bring_to_front)
hide_action.triggered.connect(self.hide)
quit_action.triggered.connect(self.quit)
self.tray_menu = QMenu()
self.tray_menu.addAction(self.blender_action)
self.tray_menu.addSeparator()
self.tray_menu.addAction(show_action)
self.tray_menu.addAction(hide_action)
self.tray_menu.addAction(quit_action)
self.tray_icon = QSystemTrayIcon(self.app_icon, self)
self.tray_icon.setContextMenu(self.tray_menu)
self.tray_icon.messageClicked.connect(self.bring_to_front)
self.tray_icon.activated.connect(lambda btn: self.bring_to_front()
if btn == 3 else False)
self.tray_icon.show()
# Version layout
self.btnUpdate.clicked.connect(self.update)
self.set_task_visible(False)
self.layouts = []
self.latest_local = None
self.collect_versions()
self.draw_list_versions()
# Custom drag behaviour
self.old_pos = self.pos()
self.pressed = False
# Update task
self.is_update_running = False
self.uptodate_silent = False
self.uptodate_thread = CheckForUpdates(self)
self.uptodate_thread.new_version_obtained.connect(
self.show_new_version)
self.uptodate_thread.start()
self.taskbar_progress = None
def showEvent(self, event):
# Setup taskbar
if not self.taskbar_progress:
self.task_button = QWinTaskbarButton(self)
self.task_button.setWindow(self.windowHandle())
self.taskbar_progress = self.task_button.progress()
self.taskbar_progress.setVisible(True)
self.taskbar_progress.setValue(0)
def open_latest_b3d(self):
if self.layouts:
self.layouts[0].open()
def show_new_version(self, display_name):
if (display_name == self.progressBar.text()):
return
self.set_task_visible(True)
self.set_progress_bar(0, 0, display_name)
if self.isHidden() and not self.uptodate_silent:
self.tray_icon.showMessage(
"Blender Version Manager",
"New version of Blender 2.8 is avaliable!",
QSystemTrayIcon.Information, 4000)
def set_task_visible(self, is_visible):
if is_visible:
self.progressBar.show()
self.btnUpdate.show()
self.btnCancel.hide()
else:
self.progressBar.hide()
self.btnUpdate.hide()
self.btnCancel.hide()
def is_running_task(self):
if self.is_update_running:
QMessageBox.information(self, "Warning",
"Update task in progress!", QMessageBox.Ok)
return True
else:
return False
def toggle_run_minimized(self, is_checked):
self.settings.setValue('is_run_minimized', is_checked)
def toggle_register_blend(self, is_checked):
self.settings.setValue('is_register_blend', is_checked)
def cleanup_layout(self, layout):
while layout.count():
item = layout.takeAt(0)
widget = item.widget()
if widget is not None:
widget.deleteLater()
else:
self.cleanup_layout(item.layout())
def collect_versions(self):
self.layouts.clear()
root_folder = self.settings.value('root_folder')
dirs = next(os.walk(root_folder))[1]
versions = []
for dir in dirs:
if os.path.isfile(os.path.join(root_folder, dir, "blender.exe")):
versions.append(dir)
for ver in versions:
b3d_item_layout = B3dItemLayout(root_folder, ver, False, self)
self.layouts.append(b3d_item_layout)
def draw_list_versions(self):
if len(self.layouts) > 0:
self.layouts.sort(key=lambda ver: ver.mtime, reverse=True)
self.latest_local = self.layouts[0].git
self.blender_action.setVisible(True)
for b3d_item_layout in self.layouts:
self.layoutListVersions.removeItem(b3d_item_layout)
b3d_item_layout.set_is_latest(False)
self.layouts[0].set_is_latest(True)
for b3d_item_layout in self.layouts:
self.layoutListVersions.addLayout(b3d_item_layout)
else:
self.latest_local = None
label = QLabel("No Local Versions Found!")
label.setStyleSheet("color: white; font-size: 10pt;")
label.setAlignment(Qt.AlignCenter)
label.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
self.layoutListVersions.addWidget(label)
self.blender_action.setVisible(False)
def set_root_folder(self):
root_folder = self.settings.value('root_folder')
dir = QFileDialog.getExistingDirectory(self, "Choose Root Folder",
root_folder)
if dir and (dir != root_folder):
self.settings.setValue('root_folder', dir)
self.labelRootFolder.setText(dir)
self.cleanup_layout(self.layoutListVersions)
self.collect_versions()
self.draw_list_versions()
def update(self):
self.is_update_running = True
self.uptodate_thread.is_running = False
self.uptodate_thread.terminate()
self.uptodate_thread.wait()
self.btnUpdate.hide()
self.btnCancel.show()
self.btnSetRootFolder.hide()
self.set_progress_bar(0, 0, "Downloading: %p%")
self.build_loader = BuildLoader(self,
self.uptodate_thread.download_url)
self.build_loader.finished.connect(self.finished)
self.build_loader.progress_changed.connect(self.set_progress_bar)
self.build_loader.block_abortion.connect(lambda: self.btnCancel.hide())
self.btnCancel.clicked.connect(self.build_loader.stop)
self.build_loader.start()
def finished(self, version):
self.build_loader.terminate()
self.build_loader.wait()
self.btnSetRootFolder.show()
self.set_task_visible(False)
self.is_update_running = False
if version:
root_folder = self.settings.value('root_folder')
b3d_item_layout = B3dItemLayout(root_folder, version, True, self)
self.layouts.append(b3d_item_layout)
self.tray_icon.showMessage("Blender Version Manager",
"Update finished!",
QSystemTrayIcon.Information, 4000)
self.draw_list_versions()
self.uptodate_thread = CheckForUpdates(self)
self.uptodate_thread.new_version_obtained.connect(
self.show_new_version)
self.set_progress_bar(0, 0, "")
self.uptodate_thread.start()
def set_progress_bar(self, progress_bar_val, taskbar_val, format):
self.progressBar.setFormat(format)
self.progressBar.setValue(progress_bar_val * 100)
if self.taskbar_progress:
self.taskbar_progress.setValue(taskbar_val * 100)
def quit(self):
if not self.is_running_task():
self.tray_icon.hide()
self.app.quit()
def closeEvent(self, event):
if self.actionToggleRunMinimized.isChecked():
event.ignore()
self.hide()
elif self.is_running_task():
event.ignore()
else:
self.tray_icon.hide()
event.accept()
def mousePressEvent(self, event):
self.old_pos = event.globalPos()
self.pressing = True
def mouseMoveEvent(self, event):
if self.pressing:
delta = QPoint(event.globalPos() - self.old_pos)
self.move(self.x() + delta.x(), self.y() + delta.y())
self.old_pos = event.globalPos()
def mouseReleaseEvent(self, QMouseEvent):
self.pressing = False
def toggle_run_on_startup(self, is_checked):
key = winreg.OpenKey(
winreg.HKEY_CURRENT_USER,
r'SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Run', 0,
winreg.KEY_SET_VALUE)
if (is_checked):
path = sys.executable
winreg.SetValueEx(key, 'Blender Version Manager', 0, winreg.REG_SZ,
path)
else:
try:
winreg.DeleteValue(key, 'Blender Version Manager')
except:
pass
key.Close()
self.settings.setValue('is_run_on_startup', is_checked)
def bring_to_front(self):
self.setWindowState(self.windowState() & ~Qt.WindowMinimized
| Qt.WindowActive)
self.activateWindow()
self.show()
class BVMQMainWindow(QMainWindow, main_window_design.Ui_MainWindow):
def __init__(self, app):
super().__init__()
self.app = app
self.platform = get_platform()
self.setWindowFlags(Qt.FramelessWindowHint)
self.setupUi(self)
# Read icons
self.icon_app = QIcon(":/icons/app.svg")
self.icon_star = QIcon(":/icons/star.png")
self.icon_trash = QIcon(":/icons/delete.png")
self.icon_quit = QIcon(":/icons/quit.png")
self.icon_fake = QIcon(":/icons/fake.svg")
# Read settings
self.settings = QSettings('b3d_version_manager', 'settings')
is_register_blend = self.settings.value('is_register_blend', type=bool)
self.is_run_minimized = self.settings.value('is_run_minimized',
type=bool)
is_run_on_startup = self.settings.value('is_run_on_startup', type=bool)
root_folder = self.settings.value('root_folder')
if (not root_folder) or (not os.path.isdir(root_folder)):
exe_path = os.path.dirname(sys.executable)
self.settings.setValue('root_folder', exe_path)
# Custom title bar
self.btnWiki.clicked.connect(lambda: webbrowser.open(
"https://github.com/DotBow/Blender-Version-Manager/wiki"))
self.title.setText("%s %s" % (QApplication.applicationName(),
QApplication.applicationVersion()))
self.btnClose.clicked.connect(self.hide)
self.btnMinimize.clicked.connect(self.showMinimized)
# Custom menu bar
self.actionToggleRegisterBlend.setChecked(is_register_blend)
self.actionToggleRegisterBlend.triggered.connect(
self.toggle_register_blend)
self.actionToggleRunMinimized.setChecked(self.is_run_minimized)
self.actionToggleRunMinimized.triggered.connect(
self.toggle_run_minimized)
self.actionToggleRunOnStartup.setChecked(is_run_on_startup)
self.actionToggleRunOnStartup.triggered.connect(
self.toggle_run_on_startup)
self.menubar.hide()
self.btnSettings.setMenu(self.menuFile)
self.menuFile.installEventFilter(self)
# Root folder layout
self.labelRootFolder.setText(self.settings.value('root_folder'))
self.btnSetRootFolder.clicked.connect(self.set_root_folder)
self.btnOpenRootFolder.clicked.connect(self.open_root_folder)
# Tray layout
self.blender_action = QAction(self.icon_star, "Blender ", self)
show_action = QAction("Show", self)
hide_action = QAction("Hide", self)
quit_action = QAction(self.icon_quit, "Quit", self)
self.blender_action.triggered.connect(self.open_latest_b3d)
show_action.triggered.connect(self.bring_to_front)
hide_action.triggered.connect(self.hide)
quit_action.triggered.connect(self.quit)
self.tray_menu = QMenu()
self.tray_menu.setStyleSheet(self.menuFile.styleSheet())
self.tray_menu.addAction(self.blender_action)
self.tray_menu.addAction(show_action)
self.tray_menu.addAction(hide_action)
self.tray_menu.addAction(quit_action)
self.tray_icon = QSystemTrayIcon(self.icon_app, self)
self.tray_icon.setToolTip("Blender Version Manager")
self.tray_icon.setContextMenu(self.tray_menu)
self.tray_icon.messageClicked.connect(self.bring_to_front)
self.tray_icon.activated.connect(self.onTrayIconActivated)
self.tray_icon.show()
# Version layout
self.btnUpdate.clicked.connect(self.update)
self.set_task_visible(False)
self.zeroBuildsWarning.hide()
self.layouts = []
self.collect_versions()
self.draw_list_versions()
# Custom drag behaviour
self.old_pos = self.pos()
self.pressed = False
# Update task
self.is_update_running = False
self.start_uptodate_thread()
self.taskbar_progress = None
self.left_click_timer = QTimer(self)
self.left_click_timer.setSingleShot(True)
self.left_click_timer.timeout.connect(self.bring_to_front)
def start_uptodate_thread(self):
self.uptodate_thread = CheckForUpdates(self)
self.uptodate_thread.new_version_obtained.connect(
self.show_new_version)
self.uptodate_thread.start()
def stop_uptodate_thread(self):
self.uptodate_thread.is_running = False
self.uptodate_thread.terminate()
self.uptodate_thread.wait()
def onTrayIconActivated(self, reason):
if reason == QSystemTrayIcon.Trigger:
self.left_click_timer.start(QApplication.doubleClickInterval())
elif reason == QSystemTrayIcon.DoubleClick:
self.left_click_timer.stop()
self.open_latest_b3d()
def showEvent(self, event):
# Setup taskbar
if self.platform == 'Windows':
if not self.taskbar_progress:
self.task_button = QWinTaskbarButton(self)
self.task_button.setWindow(self.windowHandle())
self.taskbar_progress = self.task_button.progress()
self.taskbar_progress.setVisible(True)
self.taskbar_progress.setValue(0)
def open_latest_b3d(self):
if self.layouts:
self.layouts[0].open()
def show_new_version(self, display_name):
self.set_task_visible(True)
self.set_progress_bar(0, 0, display_name)
if self.isHidden():
self.tray_icon.showMessage("Blender Version Manager",
"New build of Blender is available!",
QSystemTrayIcon.Information, 4000)
def set_task_visible(self, is_visible):
if is_visible:
self.progressBar.show()
self.btnUpdate.show()
self.btnCancel.hide()
else:
self.progressBar.hide()
self.btnUpdate.hide()
self.btnCancel.hide()
def is_running_task(self):
if self.is_update_running:
QMessageBox.information(self, "Warning",
"Update task in progress!", QMessageBox.Ok)
return True
else:
return False
def toggle_run_minimized(self, is_checked):
self.settings.setValue('is_run_minimized', is_checked)
def toggle_register_blend(self, is_checked):
self.settings.setValue('is_register_blend', is_checked)
def cleanup_layout(self, layout):
while layout.count():
item = layout.takeAt(0)
widget = item.widget()
if widget is not None:
widget.deleteLater()
else:
self.cleanup_layout(item.layout())
def collect_versions(self):
self.layouts.clear()
root_folder = self.settings.value('root_folder')
dirs = next(os.walk(root_folder))[1]
versions = []
if self.platform == 'Windows':
blender_exe = "blender.exe"
elif self.platform == 'Linux':
blender_exe = "blender"
for dir in dirs:
if os.path.isfile(os.path.join(root_folder, dir, blender_exe)):
versions.append(dir)
for ver in versions:
b3d_item_layout = B3dItemLayout(root_folder, ver, False, self)
self.layouts.append(b3d_item_layout)
def draw_list_versions(self):
if len(self.layouts) > 0:
self.zeroBuildsWarning.hide()
self.layouts.sort(key=lambda ver: ver.mtime, reverse=True)
self.blender_action.setVisible(True)
for b3d_item_layout in self.layouts:
self.layoutListVersions.removeItem(b3d_item_layout)
b3d_item_layout.set_is_latest(False)
self.layouts[0].set_is_latest(True)
for b3d_item_layout in self.layouts:
self.layoutListVersions.addLayout(b3d_item_layout)
else:
self.zeroBuildsWarning.show()
self.blender_action.setVisible(False)
def set_root_folder(self):
root_folder = self.settings.value('root_folder')
dir = QFileDialog.getExistingDirectory(self, "Choose Root Folder",
root_folder)
if dir and (dir != root_folder):
self.stop_uptodate_thread()
self.zeroBuildsWarning.hide()
self.set_task_visible(False)
self.settings.setValue('root_folder', dir)
self.labelRootFolder.setText(dir)
self.cleanup_layout(self.layoutListVersions)
self.collect_versions()
self.draw_list_versions()
self.start_uptodate_thread()
def open_root_folder(self):
root_folder = self.settings.value('root_folder')
if self.platform == 'Windows':
os.startfile(root_folder)
elif self.platform == 'Linux':
subprocess.call(["xdg-open", root_folder])
def update(self):
self.is_update_running = True
self.stop_uptodate_thread()
self.btnUpdate.hide()
self.btnCancel.show()
self.btnSetRootFolder.hide()
self.set_progress_bar(0, 0, "Downloading: %p%")
self.build_loader = BuildLoader(self,
self.uptodate_thread.download_url,
self.uptodate_thread.strptime)
self.build_loader.finished.connect(self.finished)
self.build_loader.progress_changed.connect(self.set_progress_bar)
self.build_loader.block_abortion.connect(lambda: self.btnCancel.hide())
self.btnCancel.clicked.connect(self.build_loader.stop)
self.build_loader.start()
def finished(self, version):
self.build_loader.terminate()
self.build_loader.wait()
self.btnSetRootFolder.show()
self.set_task_visible(False)
self.is_update_running = False
if version:
root_folder = self.settings.value('root_folder')
b3d_item_layout = B3dItemLayout(root_folder, version, True, self)
self.layouts.append(b3d_item_layout)
self.tray_icon.showMessage("Blender Version Manager",
"Download finished!",
QSystemTrayIcon.Information, 4000)
self.draw_list_versions()
self.set_progress_bar(0, 0, "")
self.start_uptodate_thread()
def set_progress_bar(self, progress_bar_val, taskbar_val, format):
self.progressBar.setFormat(format)
self.progressBar.setValue(progress_bar_val * 100)
if self.taskbar_progress:
self.taskbar_progress.setValue(taskbar_val * 100)
def quit(self):
if not self.is_running_task():
self.tray_icon.hide()
self.app.quit()
def closeEvent(self, event):
self.hide()
event.ignore()
def mousePressEvent(self, event):
self.old_pos = event.globalPos()
self.pressing = True
def mouseMoveEvent(self, event):
if self.pressing:
delta = QPoint(event.globalPos() - self.old_pos)
self.move(self.x() + delta.x(), self.y() + delta.y())
self.old_pos = event.globalPos()
def mouseReleaseEvent(self, QMouseEvent):
self.pressing = False
def toggle_run_on_startup(self, is_checked):
if (self.platform == 'Windows'):
key = winreg.OpenKey(
winreg.HKEY_CURRENT_USER,
r'SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Run', 0,
winreg.KEY_SET_VALUE)
if (is_checked):
path = sys.executable
winreg.SetValueEx(key, 'Blender Version Manager', 0,
winreg.REG_SZ, path)
else:
try:
winreg.DeleteValue(key, 'Blender Version Manager')
except:
pass
key.Close()
self.settings.setValue('is_run_on_startup', is_checked)
def bring_to_front(self):
self.setWindowState(self.windowState() & ~Qt.WindowMinimized
| Qt.WindowActive)
self.show()
self.activateWindow()
# Prevent QMenu from closing
def eventFilter(self, obj, event):
if event.type() in [QEvent.MouseButtonRelease]:
if isinstance(obj, QMenu):
if obj.activeAction():
if not obj.activeAction().menu():
obj.activeAction().trigger()
return True
return super(BVMQMainWindow, self).eventFilter(obj, event)
class B3dVersionMangerMainWindow(QMainWindow,
main_window_design.Ui_MainWindow):
def __init__(self, app):
super().__init__()
self.app = app
self.setupUi(self)
# Read Icons
self.app_icon = QIcon(":/icons/app.svg")
self.star_icon = QIcon(":/icons/star.svg")
self.star_inv_icon = QIcon(":/icons/star_inv.svg")
self.trash_icon = QIcon(":/icons/delete.svg")
self.quit_icon = QIcon(":/icons/quit_inv.svg")
self.fake_icon = QIcon(":/icons/fake.svg")
# Read Settings
self.settings = QSettings('b3d_version_manager', 'settings')
self.is_run_minimized = self.settings.value('is_run_minimized',
type=bool)
is_run_on_startup = self.settings.value('is_run_on_startup', type=bool)
root_folder = self.settings.value('root_folder')
if not os.path.isdir(root_folder):
exe_path = os.path.dirname(sys.executable)
self.settings.setValue('root_folder', exe_path)
# Custom Title Bar
self.btnClose.clicked.connect(self.close)
self.btnMinimize.clicked.connect(self.showMinimized)
# Custom Menu Bar
self.actionToggleRunMinimized.setChecked(self.is_run_minimized)
self.actionToggleRunMinimized.triggered.connect(
self.toggle_run_minimized)
self.actionToggleRunOnStartup.setChecked(is_run_on_startup)
self.actionToggleRunOnStartup.triggered.connect(
self.toggle_run_on_startup)
self.actionClearTempFolder.triggered.connect(self.clear_temp_folder)
self.actionQuit.triggered.connect(self.quit)
self.menubar.hide()
self.btnFile.setMenu(self.menuFile)
# Root Folder Layout
self.labelRootFolder.setText(self.settings.value('root_folder'))
self.btnSetRootFolder.clicked.connect(self.set_root_folder)
self.btnOpenRootFolder.clicked.connect(
lambda: os.startfile(self.settings.value('root_folder')))
# Tray Layout
self.blender_action = QAction(self.star_inv_icon, "Blender", self)
show_action = QAction("Show", self)
hide_action = QAction("Hide", self)
quit_action = QAction(self.quit_icon, "Quit", self)
self.blender_action.triggered.connect(self.exec_blender)
show_action.triggered.connect(self.bring_to_front)
hide_action.triggered.connect(self.hide)
quit_action.triggered.connect(self.quit)
self.tray_menu = QMenu()
self.tray_menu.addAction(self.blender_action)
self.tray_menu.addSeparator()
self.tray_menu.addAction(show_action)
self.tray_menu.addAction(hide_action)
self.tray_menu.addAction(quit_action)
self.tray_icon = QSystemTrayIcon(self.app_icon, self)
self.tray_icon.setContextMenu(self.tray_menu)
self.tray_icon.messageClicked.connect(self.bring_to_front)
self.tray_icon.activated.connect(lambda btn: self.bring_to_front()
if btn == 3 else False)
self.tray_icon.show()
# Version Layout
self.btnUpdate.clicked.connect(self.update)
self.set_task_visible(False)
self.draw_list_versions()
# Custom Drag Behaviour
self.old_pos = self.pos()
self.pressed = False
# Update Task
self.is_update_running = False
self.uptodate_thread = None
self.uptodate_silent = False
self.uptodate_task()
def uptodate_task(self):
if self.uptodate_thread:
self.uptodate_thread.finished.set()
try:
url = self.get_download_url()
version = self.get_download_url().split('-', )[-2]
new_version = True
if self.latest_local:
if version in self.latest_local:
new_version = False
if new_version:
info = urlopen(url).info()
ctime = info['last-modified']
size = str(int(info['content-length']) // 1048576) + " MB"
self.set_task_visible(True)
self.set_progress_bar(
0, "Git-" + version + " | " + ctime + " | " + size)
if self.isHidden() and not self.uptodate_silent:
self.tray_icon.showMessage(
"Blender Version Manager",
"New version of Blender 2.8 is avaliable!",
QSystemTrayIcon.Information, 2000)
self.uptodate_silent = True
else:
if self.progressBar.isVisible():
self.set_task_visible(False)
except urllib.error.URLError as e:
print(e)
self.uptodate_thread = threading.Timer(60.0, self.uptodate_task)
self.uptodate_thread.setDaemon(True)
self.uptodate_thread.start()
def set_task_visible(self, is_visible):
if is_visible:
self.progressBar.show()
self.btnUpdate.show()
self.btnCancel.hide()
else:
self.progressBar.hide()
self.btnUpdate.hide()
self.btnCancel.hide()
def is_running_task(self):
if self.is_update_running:
QMessageBox.information(self, "Warning",
"Update task in progress!", QMessageBox.Ok)
return True
else:
return False
def toggle_run_minimized(self, is_checked):
self.settings.setValue('is_run_minimized', is_checked)
def clear_temp_folder(self):
if not self.is_running_task():
temp_folder = os.path.join(self.settings.value('root_folder'),
"temp")
if os.path.isdir(temp_folder):
shutil.rmtree(temp_folder)
def cleanup_layout(self, layout):
while layout.count():
item = layout.takeAt(0)
widget = item.widget()
if widget is not None:
widget.deleteLater()
else:
self.cleanup_layout(item.layout())
def collect_versions(self):
root_folder = self.settings.value('root_folder')
dirs = next(os.walk(root_folder))[1]
prog = re.compile("blender-2.80")
versions = []
for dir in dirs:
if prog.match(dir):
versions.append(dir)
versions.sort(key=lambda ver: os.path.getctime(
os.path.join(root_folder, ver, "blender.exe")),
reverse=True)
return versions
def draw_list_versions(self):
self.cleanup_layout(self.layoutListVersions)
root_folder = self.settings.value('root_folder')
versions = self.collect_versions()
if versions:
self.blender_action.setVisible(True)
self.latest_local = versions[0]
for ver in versions:
is_latest = True if ver == self.latest_local else False
b3d_item_layout = B3dItemLayout(root_folder, ver, is_latest,
self)
self.layoutListVersions.addLayout(b3d_item_layout)
else:
self.latest_local = None
label = QLabel("No Local Versions Found!")
label.setStyleSheet("color: white; font-size: 10pt;")
label.setAlignment(Qt.AlignCenter)
label.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
self.layoutListVersions.addWidget(label)
self.blender_action.setVisible(False)
def exec_blender(self):
latest_ver = self.collect_versions()[0]
root_folder = self.settings.value('root_folder')
subprocess.Popen(os.path.join(root_folder, latest_ver, "blender.exe"))
def set_root_folder(self):
dir = QFileDialog.getExistingDirectory(
self, "Choose Root Folder", self.settings.value('root_folder'))
if dir:
if self.uptodate_thread:
self.uptodate_thread.cancel()
self.settings.setValue('root_folder', dir)
self.labelRootFolder.setText(dir)
self.draw_list_versions()
self.uptodate_task()
def update(self):
self.is_update_running = True
while self.uptodate_thread.finished.isSet():
pass
self.uptodate_thread.cancel()
self.btnUpdate.hide()
self.btnCancel.show()
self.btnSetRootFolder.setEnabled(False)
self.set_progress_bar(0, "Downloading: %p%")
self.build_loader = BuildLoader(self.settings.value('root_folder'),
self.get_download_url())
self.build_loader.finished.connect(self.finished)
self.build_loader.progress_changed.connect(self.set_progress_bar)
self.btnCancel.clicked.connect(self.build_loader.stop)
self.build_loader.start()
def finished(self, success):
self.build_loader.quit()
self.build_loader.terminate()
self.build_loader.wait()
self.btnSetRootFolder.setEnabled(True)
self.set_task_visible(False)
self.draw_list_versions()
self.is_update_running = False
if success:
self.tray_icon.showMessage("Blender Version Manager",
"Update finished!",
QSystemTrayIcon.Information, 2000)
self.uptodate_task()
def set_progress_bar(self, val, format):
self.progressBar.setFormat(format)
self.progressBar.setValue(val * 100)
def get_download_url(self):
builder_url = "https://builder.blender.org"
builder_content = urlopen(builder_url + "/download").read()
builder_soup = BeautifulSoup(builder_content, 'html.parser')
version_url = builder_soup.find(
href=re.compile("blender-2.80"))['href']
return builder_url + version_url
def quit(self):
if not self.is_running_task():
self.tray_icon.hide()
self.app.quit()
def closeEvent(self, event):
if self.actionToggleRunMinimized.isChecked():
event.ignore()
self.hide()
elif self.is_running_task():
event.ignore()
else:
self.tray_icon.hide()
event.accept()
def mousePressEvent(self, event):
self.old_pos = event.globalPos()
self.pressing = True
def mouseMoveEvent(self, event):
if self.pressing:
delta = QPoint(event.globalPos() - self.old_pos)
self.move(self.x() + delta.x(), self.y() + delta.y())
self.old_pos = event.globalPos()
def mouseReleaseEvent(self, QMouseEvent):
self.pressing = False
def toggle_run_on_startup(self, is_checked):
key = winreg.OpenKey(
winreg.HKEY_CURRENT_USER,
r'SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Run', 0,
winreg.KEY_SET_VALUE)
if (is_checked):
path = sys.executable
winreg.SetValueEx(key, 'B3DVersionManager', 0, winreg.REG_SZ, path)
else:
try:
winreg.DeleteValue(key, 'B3DVersionManager')
except:
pass
key.Close()
self.settings.setValue('is_run_on_startup', is_checked)
def bring_to_front(self):
self.setWindowState(self.windowState() & ~Qt.WindowMinimized
| Qt.WindowActive)
self.activateWindow()
self.show()