diff options
Diffstat (limited to '.local/share/Anki2/addons21/anki_reworked/__init__.py')
| -rw-r--r-- | .local/share/Anki2/addons21/anki_reworked/__init__.py | 705 |
1 files changed, 0 insertions, 705 deletions
diff --git a/.local/share/Anki2/addons21/anki_reworked/__init__.py b/.local/share/Anki2/addons21/anki_reworked/__init__.py deleted file mode 100644 index 81b2627..0000000 --- a/.local/share/Anki2/addons21/anki_reworked/__init__.py +++ /dev/null @@ -1,705 +0,0 @@ -import json -### Custom util functions -from .utils.modules import * -### Logger for debuging -from .utils.logger import logger - -from typing import Any, List, Optional -from PyQt5.QtWidgets import QWidget -from platform import system, version, release -from ctypes import * -# import the main window object (mw) from aqt -from aqt import AnkiQt, DialogManager, mw -from aqt.theme import theme_manager, colors -from aqt import gui_hooks -## QT dialog windows -# Browser import legacy check (2.1.22) -if module_exists("aqt.browser.browser"): - from aqt.browser.browser import Browser -else: - from aqt.browser import Browser -# DeckStats import legacy check -if module_has_attribute("aqt.stats", "NewDeckStats"): - from aqt.stats import DeckStats, NewDeckStats -else: - from aqt.stats import DeckStats -from aqt.addcards import AddCards -from aqt.editcurrent import EditCurrent -from aqt.about import ClosableQDialog -from aqt.preferences import Preferences -from aqt.addons import AddonsDialog -# FilteredDeckConfigDialog import non-legacy check -if module_exists("aqt.filtered_deck"): - from aqt.filtered_deck import FilteredDeckConfigDialog -# QT page views -from aqt.toolbar import Toolbar, TopToolbar -from aqt.deckbrowser import DeckBrowser, DeckBrowserBottomBar -from aqt.overview import Overview, OverviewBottomBar -from aqt.editor import Editor -from aqt.reviewer import Reviewer, ReviewerBottomBar -from aqt.webview import AnkiWebView, WebContent - -### Load config data here -from .config import config, write_config, get_config - -## Addon compatibility fixes -# More Overview Stats 2.1 addon compatibility fix -addon_more_overview_stats_fix = config['addon_more_overview_stats'] -addon_advanced_review_bottom_bar = config['addon_advanced_review_bottom_bar'] - -## Customization -theme = config['theme'] -# Init script/file path -from .utils.css_files import css_files_dir -from .utils.themes import themes, write_theme, get_theme -logger.debug(css_files_dir) -logger.debug(themes) -themes_parsed = get_theme(theme) -color_mode = 2 if theme_manager.get_night_mode() else 1 # 1 = light and 2 = dark - -dwmapi = None -## Darkmode windows titlebar thanks to miere43 -def set_dark_titlebar(window, dwmapi) -> None: - if dwmapi: - handler_window = c_void_p(int(window.winId())) - DWMWA_USE_IMMERSIVE_DARK_MODE_BEFORE_20H1 = c_int(19) - DWMWA_USE_IMMERSIVE_DARK_MODE = c_int(20) - windows_version = int(version().split('.')[2]) - attribute = DWMWA_USE_IMMERSIVE_DARK_MODE if windows_version >= 18985 else DWMWA_USE_IMMERSIVE_DARK_MODE_BEFORE_20H1 - if windows_version >= 17763 and int(release()) >= 10: - dwmapi.DwmSetWindowAttribute(handler_window, attribute, byref(c_int(1)), c_size_t(4)) -def set_dark_titlebar_qt(obj, dwmapi, fix=True) -> None: - if dwmapi and theme_manager.get_night_mode(): - set_dark_titlebar(obj, dwmapi) - # Trick to refresh the titlebar after dark titlebar is set - if fix: - obj.showFullScreen() - obj.showNormal() -if system() == "Windows" and theme_manager.get_night_mode(): - dwmapi = WinDLL("dwmapi") - if dwmapi: - dwmapi.DwmSetWindowAttribute.argtypes = [c_void_p, c_int, c_void_p, c_size_t] - dwmapi.DwmSetWindowAttribute.restype = c_int - set_dark_titlebar(mw, dwmapi) -logger.debug(dwmapi) - -### CSS injections -def load_custom_style(): - theme_colors = "" - for color_name in themes_parsed.get("colors"): - color = themes_parsed.get("colors").get(color_name) - if color[3]: - theme_colors += f"{color[3]}: {color[color_mode]};\n " - else: - theme_colors += f"--{color_name.lower().replace('_','-')}: {color[color_mode]};\n " - custom_style = """ -<style> - :root, - :root .isMac, - :root .isWin, - :root .isLin { - %s - } - html { - font-family: %s; - font-size: %spx; - } -</style> - """ % (theme_colors, config["font"], config["font_size"]) - return custom_style - -## Adds styling on the different webview contents, before the content is set -def on_webview_will_set_content(web_content: WebContent, context: Optional[Any]) -> None: - logger.debug(context) # Logs content being loaded, find out the instance - web_content.css.append(css_files_dir['global']) # Global css - web_content.head += load_custom_style() # Custom styling - # Deckbrowser - if isinstance(context, DeckBrowser): - web_content.css.append(css_files_dir['DeckBrowser']) - # TopToolbar - elif isinstance(context, TopToolbar): - web_content.css.append(css_files_dir['TopToolbar']) - # BottomToolbar (Buttons) - elif isinstance(context, DeckBrowserBottomBar) or isinstance(context, OverviewBottomBar): - web_content.css.append(css_files_dir['BottomBar']) - # Overview - elif isinstance(context, Overview): - if addon_more_overview_stats_fix: - web_content.head += "<style>center > table tr:first-of-type {display: table-row; flex-direction: unset;}</style>" - web_content.css.append(css_files_dir['Overview']) - # Editor - elif isinstance(context, Editor): - web_content.css.append(css_files_dir['Editor']) - # Reviewer - elif isinstance(context, Reviewer): - web_content.css.append(css_files_dir['Reviewer']) - elif isinstance(context, ReviewerBottomBar): - if addon_advanced_review_bottom_bar: - #web_content.head += "<style>td.stat[align='left']:nth-of-type(2) {position: absolute; z-index: 1;}</style>" - web_content.body += "<script>const center = document.getElementById('outer');center.classList.add('arbb');</script>" - else: - web_content.css.append(css_files_dir['BottomBar']) - web_content.css.append(css_files_dir['ReviewerBottomBar']) - # Button padding bottom - web_content.body += "<div style='height: 9px; opacity: 0; pointer-events: none;'></div>" - web_content.body += "<div id='padFix' style='height: 30px; opacity: 0; pointer-events: none;'><script>const e = document.getElementById('padFix');e.parentElement.removeChild(e);</script></div>" - mw.bottomWeb.adjustHeightToFit() - # CardLayout - elif context_name_includes(context, "aqt.clayout.CardLayout"): - web_content.css.append(css_files_dir['CardLayout']) - ## Legacy webviews - # ResetRequired on card edit (legacy) - elif context_name_includes(context, "aqt.main.ResetRequired"): - web_content.css.append(css_files_dir['legacy']) -gui_hooks.webview_will_set_content.append(on_webview_will_set_content) - -# TopToolbar styling fix through height change by adding <br> tag -def redraw_toolbar() -> None: - # Reload the webview content with added <br/> tag, making the bar larger in height - mw.toolbar.web.setFixedHeight(60) - mw.toolbar.web.eval(""" - var br = document.createElement("br"); - document.body.appendChild(br); - """) - - if 'Qt6' in QPalette.ColorRole.__module__: - mw.toolbar.web.eval(""" - var div = document.createElement("div"); - div.style.width = "5px"; - div.style.height = "10px"; - document.body.appendChild(div); - """) - # Auto adjust the height, then redraw the toolbar - mw.toolbar.web.adjustHeightToFit() - mw.toolbar.redraw() - -def redraw_toolbar_legacy(links: List[str], _: Toolbar) -> None: - # Utilizing the link hook, we inject <br/> tag through javascript - inject_br = """ - <script> - while (document.body.querySelectorAll("br").length > 1) - document.body.querySelectorAll("br")[0].remove(); - var br = document.createElement("br"); - document.body.appendChild(br); - </script> - """ - mw.toolbar.web.setFixedHeight(60) - links.append(inject_br) - mw.toolbar.web.adjustHeightToFit() - -if attribute_exists(gui_hooks, "main_window_did_init"): - pass - #gui_hooks.main_window_did_init.append(redraw_toolbar) -elif attribute_exists(gui_hooks, "top_toolbar_did_init_links"): - gui_hooks.top_toolbar_did_init_links.append(redraw_toolbar_legacy) - pass - -# Dialog window styling -def on_dialog_manager_did_open_dialog(dialog_manager: DialogManager, dialog_name: str, dialog_instance: QWidget) -> None: - logger.debug(dialog_name) - dialog: AnkiQt = dialog_manager._dialogs[dialog_name][1] - # If dwmapi found and nightmode is enabled, set dark titlebar to dialog window - set_dark_titlebar_qt(dialog, dwmapi) - # AddCards - if dialog_name == "AddCards": - context: AddCards = dialog_manager._dialogs[dialog_name][1] - context.setStyleSheet(open(css_files_dir['QAddCards'], encoding='utf-8').read()) - # Addons popup - elif dialog_name == "AddonsDialog": - context: AddonsDialog = dialog_manager._dialogs[dialog_name][1] - context.setStyleSheet(open(css_files_dir['QAddonsDialog'], encoding='utf-8').read()) - # Browser - elif dialog_name == "Browser": - context: Browser = dialog_manager._dialogs[dialog_name][1] - context.setStyleSheet(open(css_files_dir['QBrowser'], encoding='utf-8').read()) - pass - # EditCurrent - elif dialog_name == "EditCurrent": - context: EditCurrent = dialog_manager._dialogs[dialog_name][1] - context.setStyleSheet(open(css_files_dir['QEditCurrent'], encoding='utf-8').read()) - # FilteredDeckConfigDialog - elif module_exists("aqt.filtered_deck") and dialog_name == "FilteredDeckConfigDialog": - context: FilteredDeckConfigDialog = dialog_manager._dialogs[dialog_name][1] - context.setStyleSheet(open(css_files_dir['QFilteredDeckConfigDialog'], encoding='utf-8').read()) - # Statistics / NewDeckStats - elif dialog_name == "NewDeckStats": - context: NewDeckStats = dialog_manager._dialogs[dialog_name][1] - context.setStyleSheet(open(css_files_dir['QNewDeckStats'], encoding='utf-8').read()) - # About - elif dialog_name == "About": - context: ClosableQDialog = dialog_manager._dialogs[dialog_name][1] - context.setStyleSheet(open(css_files_dir['QAbout'], encoding='utf-8').read()) - # Preferences - elif dialog_name == "Preferences": - context: Preferences = dialog_manager._dialogs[dialog_name][1] - context.setStyleSheet(open(css_files_dir['QPreferences'], encoding='utf-8').read()) - # sync_log - kore ha nani desu??? - elif dialog_name == "sync_log": - pass - -if attribute_exists(gui_hooks, "dialog_manager_did_open_dialog"): - gui_hooks.dialog_manager_did_open_dialog.append(on_dialog_manager_did_open_dialog) -else: - ## Legacy dialog window styling - # Sad monkey patch, instead of hooks :c - # setupDialogGC is being called on almost all dialog windows, utilizing this, the - # function is used as a type of hook to inject CSS styling on the QT instances - def monkey_setup_dialog_gc(obj: Any) -> None: - obj.finished.connect(lambda: mw.gcWindow(obj)) - logger.debug(obj) - set_dark_titlebar_qt(obj, dwmapi) - # AddCards - if isinstance(obj, AddCards): - obj.setStyleSheet(open(css_files_dir['QAddCards'], encoding='utf-8').read()) - # EditCurrent - elif isinstance(obj, EditCurrent): - obj.setStyleSheet(open(css_files_dir['QEditCurrent'], encoding='utf-8').read()) - # Statistics / DeckStats - elif isinstance(obj, DeckStats): - obj.setStyleSheet(open(css_files_dir['QNewDeckStats'], encoding='utf-8').read()) - # About - elif isinstance(obj, ClosableQDialog): - obj.setStyleSheet(open(css_files_dir['QAbout'], encoding='utf-8').read()) - # Preferences - ## Haven't found a solution for preferences yet :c - mw.setupDialogGC = monkey_setup_dialog_gc # Should be rare enough for other addons to also patch this I hope. - - # Addons popup - if attribute_exists(gui_hooks, "addons_dialog_will_show"): - def on_addons_dialog_will_show(dialog: AddonsDialog) -> None: - logger.debug(dialog) - set_dark_titlebar_qt(dialog, dwmapi) - dialog.setStyleSheet(open(css_files_dir['QAddonsDialog'], encoding='utf-8').read()) - gui_hooks.addons_dialog_will_show.append(on_addons_dialog_will_show) - # Browser - if attribute_exists(gui_hooks, "browser_will_show"): - def on_browser_will_show(browser: Browser) -> None: - logger.debug(browser) - set_dark_titlebar_qt(browser, dwmapi) - browser.setStyleSheet(open(css_files_dir['QBrowser'], encoding='utf-8').read()) - gui_hooks.browser_will_show.append(on_browser_will_show) - -## CONFIG DIALOG -from aqt import mw, gui_hooks -from aqt.qt import * -from .config import config -from aqt.utils import showInfo -from aqt.webview import AnkiWebView - -class ThemeEditor(QDialog): - def __init__(self, parent, *args, **kwargs): - super().__init__(parent=parent or mw, *args, **kwargs) - self.config_editor = parent - self.setWindowModality(Qt.ApplicationModal) - self.setWindowTitle(f'Anki-redesign Advanced Editor') - self.setSizePolicy(self.make_size_policy()) - self.setMinimumSize(420, 420) - self.setAttribute(Qt.WidgetAttribute.WA_DeleteOnClose) - set_dark_titlebar_qt(self, dwmapi, fix=False) - # Root layout - self.root_layout = QVBoxLayout(self) - # Main layout - self.layout = QVBoxLayout() - self.textedit = QTextEdit() - themes_plaintext = open(themes[theme], encoding='utf-8').read() - self.textedit.setPlainText(themes_plaintext) - self.layout.addWidget(self.textedit) - self.root_layout.addLayout(self.layout) - self.root_layout.addLayout(self.make_button_box()) - - def save_edit(self) -> None: - themes_parsed = json.loads(self.textedit.toPlainText()) - write_theme(themes[theme], themes_parsed) - self.config_editor.update() - self.accept() - - def make_button_box(self) -> QWidget: - def cancel(): - button = QPushButton('Cancel') - button.setCursor(QCursor(Qt.CursorShape.PointingHandCursor)) - qconnect(button.clicked, self.accept) - return button - def save(): - button = QPushButton('Save') - button.setCursor(QCursor(Qt.CursorShape.PointingHandCursor)) - button.setDefault(True) - button.setShortcut("Ctrl+Return") - button.clicked.connect(lambda _: self.save_edit()) - return button - - button_box = QHBoxLayout() - button_box.addStretch() - button_box.addWidget(cancel()) - button_box.addWidget(save()) - return button_box - - def make_size_policy(self) -> QSizePolicy: - size_policy = QSizePolicy(QSizePolicy.Preferred, QSizePolicy.Minimum) - size_policy.setHorizontalStretch(0) - size_policy.setVerticalStretch(0) - size_policy.setHeightForWidth(self.sizePolicy().hasHeightForWidth()) - return size_policy - -class ConfigDialog(QDialog): - def __init__(self, parent: QWidget, *args, **kwargs): - super().__init__(parent=parent or mw, *args, **kwargs) - self.setWindowModality(Qt.ApplicationModal) - self.setWindowTitle(f'Anki-redesign Configuration') - self.setSizePolicy(self.make_size_policy()) - self.setMinimumSize(420, 580) - self.setAttribute(Qt.WidgetAttribute.WA_DeleteOnClose) - set_dark_titlebar_qt(self, dwmapi, fix=False) - - # Color/theme - ## Loads theme color - self.theme_colors = themes_parsed.get("colors") - self.updates = [] - self.theme_general = ["TEXT_FG", "WINDOW_BG", "FRAME_BG", "BUTTON_BG", "TOOLTIP_BG", "BORDER", "MEDIUM_BORDER", "FAINT_BORDER", "HIGHLIGHT_BG", "HIGHLIGHT_FG" , "LINK", "DISABLED", "SLIGHTLY_GREY_TEXT", "PRIMARY_COLOR"] - self.theme_decks = ["CURRENT_DECK", "NEW_COUNT", "LEARN_COUNT", "REVIEW_COUNT", "ZERO_COUNT"] - self.theme_browse = ["BURIED_FG", "SUSPENDED_FG", "MARKED_BG", "FLAG1_BG", "FLAG1_FG", "FLAG2_BG", "FLAG2_FG", "FLAG3_BG", "FLAG3_FG", "FLAG4_BG", "FLAG4_FG", "FLAG5_BG", "FLAG5_FG", "FLAG6_BG", "FLAG6_FG", "FLAG7_BG", "FLAG7_FG"] - self.theme_extra = [] - - # Root layout - self.root_layout = QVBoxLayout(self) - # Main layout - self.layout = QVBoxLayout() - ## Initialize tab screen - self.tabs = QTabWidget(objectName="tabs") - self.tabs.setFocusPolicy(Qt.FocusPolicy.StrongFocus) - self.tab_general = QWidget(objectName="general") - self.tab_general.setLayout(self.create_color_picker_layout(self.theme_general)) - self.tab_decks = QWidget(objectName="decks") - self.tab_decks.setLayout(self.create_color_picker_layout(self.theme_decks)) - self.tab_browse = QWidget(objectName="browse") - self.tab_browse.setLayout(self.create_color_picker_layout(self.theme_browse)) - self.tab_extra = QWidget(objectName="extra") - self.tab_extra.setLayout(self.create_color_picker_layout(self.theme_extra)) - - self.tab_settings = QWidget(objectName="settings") - self.settings_layout = QFormLayout() - self.theme_label = QLabel("Theme:") - self.theme_label.setStyleSheet('QLabel { font-size: 14px; font-weight: bold }') - self.settings_layout.addRow(self.theme_label) - for key in themes: - self.radio = self.radio_button(key) - self.settings_layout.addRow(key, self.radio) - self.settings_layout.addRow(QLabel()) - - self.font_label = QLabel("Font:") - self.font_label.setStyleSheet('QLabel { font-size: 14px; font-weight: bold }') - self.settings_layout.addRow(self.font_label) - self.interface_font = QFontComboBox() - self.interface_font.setFixedWidth(200) - self.interface_font.setCurrentFont(QFont(config["font"])) - self.settings_layout.addRow(self.interface_font) - - self.font_size = QSpinBox() - self.font_size.setFixedWidth(200) - self.font_size.setValue(config["font_size"]) - self.font_size.setSuffix("px") - self.settings_layout.addRow(self.font_size) - - self.settings_layout.addRow(QLabel()) - - - self.fix_label = QLabel("Addon-Compatibility Fixes: ") - self.fix_label.setStyleSheet('QLabel { font-size: 14px; font-weight: bold }') - self.settings_layout.addRow(self.fix_label) - self.reload_theme = self.checkbox("theme_reload") - self.settings_layout.addRow("QT6 theme on-start reload fix", self.reload_theme) - self.addon_more_overview_stats_check = self.checkbox("addon_more_overview_stats") - self.settings_layout.addRow("More Overview Stats 21", self.addon_more_overview_stats_check) - self.addon_advanced_review_bottom_bar_check = self.checkbox("addon_advanced_review_bottom_bar") - self.settings_layout.addRow("Advanced Review Bottom Bar", self.addon_advanced_review_bottom_bar_check) - - self.tab_settings.setLayout(self.settings_layout) - - ## Add tabs - self.tabs.resize(300,200) - self.tabs.addTab(self.tab_settings,"Settings") - self.tabs.addTab(self.tab_general,"General") - self.tabs.addTab(self.tab_decks,"Decks") - self.tabs.addTab(self.tab_browse,"Browse") - #self.tabs.addTab(self.tab_extra,"Extra") - - ## Add tabs to widget - self.layout.addWidget(self.tabs) - - self.root_layout.addLayout(self.layout) - self.root_layout.addLayout(self.make_button_box()) - self.setLayout(self.root_layout) - self.show() - - def update(self) -> None: - global themes_parsed - themes_parsed = get_theme(theme) - self.theme_colors = themes_parsed.get("colors") - for update in self.updates: - update() - - def checkbox(self, key: str) -> QCheckBox: - checkbox = QCheckBox() - - def update() -> None: - value = config[key] - checkbox.setChecked(value) - - self.updates.append(update) - update() - return checkbox - - def radio_button(self, key: str) -> QRadioButton: - radio = QRadioButton() - - def update() -> None: - if theme == key: - radio.setChecked(True) - elif radio.isChecked(): - radio.setChecked(False) - - def toggle(checked) -> None: - global theme - if checked: - theme = key - self.update() - - self.updates.append(update) - radio.toggled.connect(lambda checked: toggle(checked)) - update() - return radio - - def color_input(self, key: str) -> QPushButton: - button = QPushButton() - button.setCursor(QCursor(Qt.CursorShape.PointingHandCursor)) - button.setFixedWidth(25) - button.setFixedHeight(25) - - color_dialog = QColorDialog(self) - - def set_color(rgb: str) -> None: - # Check for valid color - color = QColor() - color.setNamedColor(rgb) - if not color.isValid(): - return - # Update color - color_dialog.setCurrentColor(color) - button.setStyleSheet('QPushButton{ background-color: "%s"; border: none; border-radius: 2px}' % rgb) - - def update() -> None: - rgb = self.theme_colors.get(key)[color_mode] - set_color(rgb) - - def save(color: QColor) -> None: - rgb = color.name(QColor.NameFormat.HexRgb) - self.theme_colors[key][color_mode] = rgb - set_color(rgb) - - self.updates.append(update) - color_dialog.colorSelected.connect(lambda color: save(color)) - button.clicked.connect(lambda _: color_dialog.exec()) - return button - - def create_color_picker_layout(self, colors) -> None: - layout = QFormLayout() - for key in colors: - self.test = self.color_input(key) - layout.addRow(self.theme_colors.get(key)[0], self.test) - return layout - - def theme_file_editor(self) -> None: - diag = ThemeEditor(self) - diag.show() - - def make_button_box(self) -> QWidget: - def advanced(): - button = QPushButton('Advanced') - button.setCursor(QCursor(Qt.CursorShape.PointingHandCursor)) - qconnect(button.clicked, self.theme_file_editor) - return button - def cancel(): - button = QPushButton('Cancel') - button.setCursor(QCursor(Qt.CursorShape.PointingHandCursor)) - qconnect(button.clicked, self.accept) - return button - def save(): - button = QPushButton('Save') - button.setCursor(QCursor(Qt.CursorShape.PointingHandCursor)) - button.setDefault(True) - button.setShortcut("Ctrl+Return") - button.clicked.connect(lambda _: self.save()) - return button - - button_box = QHBoxLayout() - button_box.addWidget(advanced()) - button_box.addStretch() - button_box.addWidget(cancel()) - button_box.addWidget(save()) - return button_box - - def make_size_policy(self) -> QSizePolicy: - size_policy = QSizePolicy(QSizePolicy.Preferred, QSizePolicy.Minimum) - size_policy.setHorizontalStretch(0) - size_policy.setVerticalStretch(0) - size_policy.setHeightForWidth(self.sizePolicy().hasHeightForWidth()) - return size_policy - - def save(self) -> None: - # Save settings and update config - global config, color_mode - config["font"] = self.interface_font.currentFont().family() - config["font_size"] = self.font_size.value() - config['addon_more_overview_stats'] = self.addon_more_overview_stats_check.isChecked() - config['addon_advanced_review_bottom_bar'] = self.addon_advanced_review_bottom_bar_check.isChecked() - config['theme_reload'] = self.reload_theme.isChecked() - config["theme"] = theme - write_config(config) - config = get_config() - - # Write and update theme - color_mode = 2 if theme_manager.get_night_mode() else 1 # 1 = light and 2 = dark - themes_parsed["colors"] = self.theme_colors - write_theme(themes[theme], themes_parsed) - update_theme() - - # mw.reset() - # ShowInfo for both new and legacy support - showInfo(_("Changes will take effect when you restart Anki.")) - #showInfo(tr.preferences_changes_will_take_effect_when_you()) - self.accept() - -def check_legacy_colors() -> None: - try: - _ = colors.items() - except: - return False - return True - -def refresh_all_windows() -> None: - # Redraw top toolbar - mw.toolbar.draw() - if attribute_exists(gui_hooks, "top_toolbar_did_init_links"): - gui_hooks.top_toolbar_did_init_links.append(lambda a,b: [redraw_toolbar_legacy(a,b), gui_hooks.top_toolbar_did_init_links.remove(print)]) - - # Redraw main body - if mw.state == "review": - mw.reviewer._initWeb() - if mw.reviewer.state == "question": - mw.reviewer._showQuestion() - else: - mw.reviewer._showAnswer() - elif mw.state == "overview": - mw.overview.refresh() - elif mw.state == "deckBrowser": - mw.deckBrowser.show() - - if attribute_exists(gui_hooks, "top_toolbar_did_init_links"): - gui_hooks.top_toolbar_did_init_links.remove(redraw_toolbar) - -def update_theme() -> None: - themes_parsed = get_theme(theme) - theme_colors = themes_parsed.get("colors") - # Apply theme on colors - ncolors = {} - # Legacy color check - logger.debug(dir(colors)) - legacy = check_legacy_colors() - for color_name in theme_colors: - c = theme_colors.get(color_name) - ncolors[color_name] = c[color_mode] - if legacy: - colors[f"day{c[3].replace('--','-')}"] = c[1] - colors[f"night{c[3].replace('--','-')}"] = c[2] - # Potentially add fusion fixes too? - else: - logger.debug(getattr(colors, color_name, False)) - if getattr(colors, color_name, False): - setattr(colors, color_name, (c[1], c[2])) - # Apply theme on palette - apply_theme(ncolors) - refresh_all_windows() - -def apply_theme(colors) -> None: - # Load palette - palette = QPalette() - # QT mappings - #logger.debug(QPalette.ColorRole.__dict__) - color_map = { - QPalette.ColorRole.Window: "WINDOW_BG", - QPalette.ColorRole.WindowText: "TEXT_FG", - QPalette.ColorRole.Base: "FRAME_BG", - QPalette.ColorRole.AlternateBase: "WINDOW_BG", - QPalette.ColorRole.ToolTipBase: "TOOLTIP_BG", - QPalette.ColorRole.ToolTipText: "TEXT_FG", - QPalette.ColorRole.Text: "TEXT_FG", - QPalette.ColorRole.Button: "BUTTON_BG", - QPalette.ColorRole.ButtonText: "TEXT_FG", - QPalette.ColorRole.BrightText: "HIGHLIGHT_FG", - QPalette.ColorRole.HighlightedText: "HIGHLIGHT_FG", - QPalette.ColorRole.Link: "LINK", - QPalette.ColorRole.NoRole: "WINDOW_BG", - } - for color_role in color_map: - palette.setColor(color_role, QColor(colors[color_map[color_role]])) - - highlight_bg = QColor(colors["HIGHLIGHT_BG"]) - highlight_bg.setAlpha(64) - palette.setColor(QPalette.ColorRole.Highlight, highlight_bg) - - disabled_color = QColor(colors["DISABLED"]) - palette.setColor(QPalette.ColorRole.PlaceholderText, disabled_color) - palette.setColor(QPalette.ColorGroup.Disabled, QPalette.ColorRole.Text, disabled_color) - palette.setColor(QPalette.ColorGroup.Disabled, QPalette.ColorRole.ButtonText, disabled_color) - palette.setColor(QPalette.ColorGroup.Disabled, QPalette.ColorRole.HighlightedText, disabled_color) - - # Update palette - mw.app.setPalette(palette) - theme_manager.default_palette = palette - theme_manager._apply_style(mw.app) - - # Update webview bg - AnkiWebView._getWindowColor = lambda *args: QColor(colors["WINDOW_BG"]) - AnkiWebView.get_window_bg_color = lambda *args: QColor(colors["WINDOW_BG"]) - -# Create menu actions -def create_menu_action(parent: QWidget, dialog_class: QDialog, dialog_name: str) -> QAction: - def open_dialog(): - dialog = dialog_class(mw) - return dialog.exec() - - action = QAction(dialog_name, parent) - #qconnect(action.triggered, open_dialog) - action.triggered.connect(open_dialog) - return action - -# Load in the Anki-redesign menu -if not hasattr(mw, 'anki_redesign'): - mw.anki_redesign = QMenu("&Anki-redesign", mw) - mw.form.menubar.insertMenu(mw.form.menuHelp.menuAction(), mw.anki_redesign) - - mw.anki_redesign.addAction(create_menu_action(mw.anki_redesign, ConfigDialog, "&Config")) - mw.anki_redesign.addSeparator() - mw.reset() - update_theme() - mw.reset() - - # Rereload view to fix theme change on startup - if 'Qt6' in QPalette.ColorRole.__module__: - logger.debug('QT6 DETECTED....') - config = get_config() - if config["theme_reload"]: - update_theme() - -def on_theme_did_change() -> None: - global color_mode - color_mode = 2 if theme_manager.get_night_mode() else 1 # 1 = light and 2 = dark - update_theme() - logger.debug("THEME CHANGEEEED") - refresh_all_windows() - mw.reset() - update_theme() - - -if attribute_exists(gui_hooks, "theme_did_change"): - gui_hooks.theme_did_change.append(on_theme_did_change) - logger.debug("YEP")
\ No newline at end of file |