WxPyBrowser.py

import wx
import wx.aui
import wx.lib.platebtn
import wx.html2
import wx.lib.agw.aui.tabart


class HistoryPage(wx.Panel):
    def __init__(self, parent, history_var):
        wx.Panel.__init__(self, parent=parent)

        self.open = True
        self.parent = parent
        self.history_var = history_var
        self.frame = wx.GetTopLevelParent(self)
        
        pagesizer = wx.BoxSizer(wx.VERTICAL)
        self.listbox = listbox = wx.ListBox(self)
        top_bar_container = wx.BoxSizer(wx.HORIZONTAL)
        label = wx.StaticText(self, label='Double-click on an item to open it.', style=wx.ST_ELLIPSIZE_END)
        new = wx.Button(self, label='+', size=(30, 30), style=wx.BORDER_NONE)
        new_tip = wx.ToolTip('Open a new tab')
        new.SetToolTip(new_tip)
        top_bar_container.Add(label, 1, wx.LEFT | wx.RIGHT | wx.ALIGN_CENTER_VERTICAL, 5)
        top_bar_container.AddSpacer(30)
        top_bar_container.Add(new, 0, wx.BOTTOM | wx.RIGHT | wx.TOP, 5)
        pagesizer.Add(top_bar_container, proportion=False, flag=wx.EXPAND)
        pagesizer.Add(listbox, proportion=True, flag=wx.EXPAND)
        new.Bind(wx.EVT_BUTTON, self.tab_new)
        self.Bind(wx.EVT_LISTBOX_DCLICK, self.open_link)

        self.SetSizer(pagesizer)

        self.refresh()

    def refresh(self):
        if self.open:
            try:
                self.listbox.AppendItems(self.history_var[len(self.listbox.GetItems()):])
                wx.CallLater(3000, self.refresh)
            except:
                pass

    def open_link(self, event):
        page = WebPage(self.parent, self.history_var, url=event.GetString())
        self.parent.AddPage(page, caption="Loading")

    def tab_new(self, event):
        page = WebPage(self.parent, self.history_var)
        self.parent.AddPage(page, caption="Loading", select=True)

    def on_close(self):
        self.open = False
            
    def on_select(self):
        self.frame.SetTitle("WxPyBrowser History")


class SourceCode(wx.Panel):
    def __init__(self, parent, windowname, windowurl, history_var, *args, **kwargs):
        wx.Panel.__init__(self, parent)

        self.name = windowname
        self.parent = parent
        self.history_var = history_var
        self.frame = wx.GetTopLevelParent(self)

        pagesizer = wx.BoxSizer(wx.VERTICAL)
        self.source = source = wx.TextCtrl(self, *args, **kwargs)
        top_bar_container = wx.BoxSizer(wx.HORIZONTAL)
        label = wx.StaticText(self, label=('Source: '+windowurl), style=wx.ST_ELLIPSIZE_END)
        new = wx.Button(self, label='+', size=(30, 30), style=wx.BORDER_NONE)
        new_tip = wx.ToolTip('Open a new tab')
        new.SetToolTip(new_tip)
        top_bar_container.Add(label, 1, wx.LEFT | wx.RIGHT | wx.ALIGN_CENTER_VERTICAL, 5)
        top_bar_container.AddSpacer(30)
        top_bar_container.Add(new, 0, wx.BOTTOM | wx.RIGHT | wx.TOP, 5)
        pagesizer.Add(top_bar_container, proportion=False, flag=wx.EXPAND)
        pagesizer.Add(source, proportion=True, flag=wx.EXPAND)
        new.Bind(wx.EVT_BUTTON, self.tab_new)

        self.SetSizer(pagesizer)        

    def on_select(self):
        self.frame.SetTitle(self.name)

    def tab_new(self, event):
        page = WebPage(self.parent, self.history_var)
        self.parent.AddPage(page, caption="Loading", select=True)

    def on_close(self):
        pass
        
class WebPage(wx.Panel):
    def __init__(self, parent, history_var, url="https://duckduckgo.com/"):
        wx.Panel.__init__(self, parent=parent)
        
        self.parent = parent
        self.visited = history_var
        self.remember_history = True
        self.frame = wx.GetTopLevelParent(self)
        
        self.pagesizer = pagesizer = wx.BoxSizer(wx.VERTICAL)
        self.top_bar_container = top_bar_container = wx.FlexGridSizer(1, 8, 4, 4)
        self.back = back = wx.Button(self, label='<', size=(30, 30))
        back_tip = wx.ToolTip('Go back one page')
        back.SetToolTip(back_tip)
        self.forward = forward = wx.Button(self, label='>', size=(30, 30))
        forward_tip = wx.ToolTip('Go forward one page')
        forward.SetToolTip(forward_tip)
        self.reload = reload = wx.Button(self, label='⟳', size=(30, 30))
        reload_tip = wx.ToolTip('Reload current page')
        reload.SetToolTip(reload_tip)
        self.url_field = url_field = wx.TextCtrl(self, style = wx.TE_PROCESS_ENTER)
        new = wx.Button(self, label='+', size=(30, 30))
        new_tip = wx.ToolTip('Open a new tab')
        new.SetToolTip(new_tip)
        menu = wx.lib.platebtn.PlateButton(self, label='☰', size=(30, 30))
        menu_tip = wx.ToolTip('Show menu')
        menu.SetToolTip(menu_tip)
        self.html_window = html_window = wx.html2.WebView.New(self)
        top_bar_container.AddGrowableCol(4)
        top_bar_container.Add(back, 0, wx.LEFT | wx.BOTTOM | wx.TOP, 5)
        top_bar_container.Add(forward, 0, wx.LEFT | wx.BOTTOM | wx.TOP, 5)
        top_bar_container.Add(reload, 0, wx.LEFT | wx.BOTTOM | wx.TOP, 5)
        top_bar_container.AddSpacer(30)
        top_bar_container.Add(url_field, 0, wx.EXPAND | wx.BOTTOM | wx.TOP, 5)
        top_bar_container.Add(menu, 0, wx.RIGHT | wx.BOTTOM | wx.TOP, 5)
        top_bar_container.AddSpacer(30)
        top_bar_container.Add(new, 0, wx.ALIGN_RIGHT | wx.BOTTOM | wx.RIGHT | wx.TOP, 5)

        self.find_container = find_container = wx.BoxSizer(wx.HORIZONTAL)
        self.find_field = find_field = wx.TextCtrl(self, style = wx.TE_PROCESS_ENTER)
        find_next = wx.Button(self, label='>', size=(30, 30))
        find_next_tip = wx.ToolTip('Find next occurance')
        find_next.SetToolTip(find_next_tip)
        self.entire_word = entire_word = wx.CheckBox(self, label='Entire word')
        self.match_case = match_case = wx.CheckBox(self, label='Match case')
        self.highlight_results = highlight_results = wx.CheckBox(self, label='Highlight results')
        find_results_label = wx.StaticText(self)
        find_close = wx.Button(self, label='×', size=(30, 30))
        find_close_tip = wx.ToolTip('Close find bar')
        find_close.SetToolTip(find_close_tip)
        find_container.Add(find_field, 0, wx.EXPAND | wx.LEFT | wx.TOP | wx.BOTTOM, 5)
        find_container.Add(find_next, 0, wx.LEFT | wx.TOP | wx.BOTTOM | wx.ALIGN_CENTER_VERTICAL, 5)
        find_container.Add(entire_word, 0, wx.LEFT | wx.TOP | wx.BOTTOM | wx.ALIGN_CENTER_VERTICAL, 5)
        find_container.Add(match_case, 0, wx.LEFT | wx.TOP | wx.BOTTOM | wx.ALIGN_CENTER_VERTICAL, 5)
        find_container.Add(highlight_results, 0, wx.LEFT | wx.TOP | wx.BOTTOM | wx.ALIGN_CENTER_VERTICAL, 5)
        find_container.AddSpacer(30)
        find_container.Add(find_results_label, 1, wx.LEFT | wx.TOP | wx.BOTTOM | wx.ALIGN_CENTER_VERTICAL, 5)
        find_container.Add(find_close, 0, wx.BOTTOM | wx.RIGHT | wx.TOP, 5)

        self.zoom_container = zoom_container = wx.BoxSizer(wx.HORIZONTAL)
        self.zoom_slider = zoom_slider = wx.Slider(self, value=3, minValue=1, maxValue=5)
        zoom_close = wx.Button(self, label='×', size=(30, 30))
        zoom_close_tip = wx.ToolTip('Close zoom bar')
        zoom_close.SetToolTip(zoom_close_tip)
        zoom_container.Add(zoom_slider, 1, wx.EXPAND | wx.LEFT | wx.TOP | wx.BOTTOM, 5)
        zoom_container.AddSpacer(30)
        zoom_container.Add(zoom_close, 0, wx.BOTTOM | wx.RIGHT | wx.TOP, 5)
        
        pagesizer.Add(top_bar_container, proportion=False, flag=wx.EXPAND)
        pagesizer.Add(html_window, proportion=True, flag=wx.EXPAND)
        pagesizer.Add(find_container, proportion=False, flag=wx.EXPAND)
        pagesizer.Add(zoom_container, proportion=False, flag=wx.EXPAND)
        pagesizer.Hide(find_container)
        pagesizer.Hide(zoom_container)

        find_field.Bind(wx.EVT_TEXT_ENTER, self.continue_find)
        find_next.Bind(wx.EVT_BUTTON, lambda event, goprev=True: self.continue_find(event, goprev))
        entire_word.Bind(wx.EVT_CHECKBOX, self.continue_find)
        match_case.Bind(wx.EVT_CHECKBOX, self.continue_find)
        highlight_results.Bind(wx.EVT_CHECKBOX, self.continue_find)
        find_field.Bind(wx.EVT_TEXT, self.continue_find)
        find_close.Bind(wx.EVT_BUTTON, self.close_find)
        zoom_close.Bind(wx.EVT_BUTTON, self.close_zoom)
        zoom_slider.Bind(wx.EVT_COMMAND_SCROLL, self.zoom_slider_change)        
        back.Bind(wx.EVT_BUTTON, self.tab_back)
        forward.Bind(wx.EVT_BUTTON, self.tab_foward)
        reload.Bind(wx.EVT_BUTTON, self.tab_reload)
        new.Bind(wx.EVT_BUTTON, self.tab_new)
        url_field.Bind(wx.EVT_TEXT_ENTER, self.loadpage)
        url_field.Bind(wx.EVT_SET_FOCUS, self.click_on_url_field)
        html_window.Bind(wx.html2.EVT_WEBVIEW_LOADED, self.post_load_config)
        html_window.Bind(wx.html2.EVT_WEBVIEW_NEWWINDOW, self.open_in_new_tab)
        html_window.Bind(wx.html2.EVT_WEBVIEW_TITLE_CHANGED, self.change_title)

        self.load_url(url)
        url_field.SetValue(url)

        settings_menu = wx.Menu()
        page_menu = wx.Menu()
        menu_zoom = wx.MenuItem(settings_menu, 0, "Change zoom")
        self.menu_contextmenu = menu_contextmenu = wx.MenuItem(settings_menu, 2, "Enable context menu", "", wx.ITEM_CHECK)
        self.menu_historyenabled = menu_historyenabled = wx.MenuItem(settings_menu, 3, "Remember page history", "", wx.ITEM_CHECK)
        menu_source = wx.MenuItem(page_menu, 0, "Show source")
        menu_history = wx.MenuItem(page_menu, 1, "Show history")
        menu_print = wx.MenuItem(page_menu, 2, "Print this page")
        menu_find = wx.MenuItem(page_menu, 3, "Find in page")
        settings_menu.Append(menu_zoom)
        settings_menu.AppendSeparator()
        try:
            self.html_window.EnableAccessToDevTools()
            self.menu_devtools = menu_devtools = wx.MenuItem(settings_menu, 1, "Enable access to dev tools", "", wx.ITEM_CHECK)
            settings_menu.Append(menu_devtools)
            menu_devtools.Check()
            settings_menu.Bind(wx.EVT_MENU, self.enable_devtools, menu_devtools)
        except:
            pass
        settings_menu.Append(menu_contextmenu)
        settings_menu.Append(menu_historyenabled)     
        page_menu.Append(menu_source)
        page_menu.Append(menu_history)
        page_menu.AppendSeparator()
        page_menu.Append(menu_print)
        page_menu.Append(menu_find)
        page_menu.AppendSeparator()
        page_menu.AppendSubMenu(settings_menu, 'Page Settings')
        html_window.EnableContextMenu()
        html_window.EnableHistory()
        menu_contextmenu.Check()
        menu_historyenabled.Check()
        settings_menu.Bind(wx.EVT_MENU, self.enable_contextmenu, menu_contextmenu)
        settings_menu.Bind(wx.EVT_MENU, self.enable_historyenabled, menu_historyenabled)
        settings_menu.Bind(wx.EVT_MENU, self.adjust_zoom, menu_zoom)
        page_menu.Bind(wx.EVT_MENU, self.show_source, menu_source)
        page_menu.Bind(wx.EVT_MENU, self.show_history, menu_history)
        page_menu.Bind(wx.EVT_MENU, self.print_page, menu_print)
        page_menu.Bind(wx.EVT_MENU, self.find_in_page, menu_find)
        
        menu.SetMenu(page_menu)
        self.SetSizer(pagesizer)

    def show_source(self, event):
        title = self.html_window.GetCurrentTitle()
        if title == "":
            title = ("WxPyBrowser - Source for "+self.html_window.GetCurrentURL())
        else:
            title = ("WxPyBrowser - Source for "+self.html_window.GetCurrentTitle())
        page = SourceCode(self.parent, windowname=title, windowurl=self.html_window.GetCurrentURL(), history_var=self.visited, style=wx.TE_MULTILINE | wx.TE_READONLY | wx.HSCROLL)
        page.source.SetValue(self.html_window.GetPageSource())
        self.parent.AddPage(page, caption=title, select=False)

    def show_history(self, event):
        history_tab = HistoryPage(self.parent, self.visited)
        self.parent.AddPage(history_tab, caption="WxPyBrowser History", select=False)

    def adjust_zoom(self, event):
        self.pagesizer.Hide(self.find_container)
        self.pagesizer.Show(self.zoom_container)
        self.pagesizer.Layout()

    def close_zoom(self, event):
        self.pagesizer.Hide(self.zoom_container)
        self.pagesizer.Layout()

    def zoom_slider_change(self, event):
        scalenum = event.GetInt()
        scale_val = None
        if scalenum == 1:
            scale_val = wx.html2.WEBVIEW_ZOOM_TINY
        elif scalenum == 2:
            scale_val = wx.html2.WEBVIEW_ZOOM_SMALL
        elif scalenum == 3:
            scale_val = wx.html2.WEBVIEW_ZOOM_MEDIUM
        elif scalenum == 4:
            scale_val = wx.html2.WEBVIEW_ZOOM_LARGE
        elif scalenum == 5:
            scale_val = wx.html2.WEBVIEW_ZOOM_LARGEST
	
        self.html_window.SetZoom(scale_val)

    def print_page(self, event):
        self.html_window.Print()

    def find_in_page(self, event):
        self.pagesizer.Hide(self.zoom_container)
        self.pagesizer.Show(self.find_container)
        self.pagesizer.Layout()

    def continue_find(self, event, gonext=False):
        match_case = self.match_case.IsChecked()
        entire_word = self.entire_word.IsChecked()
        highlight_results = self.highlight_results.IsChecked()
        if not gonext:
            self.html_window.Find("")
        if match_case and entire_word and highlight_results:
            self.html_window.Find(self.find_field.GetValue(), flags=wx.html2.WEBVIEW_FIND_HIGHLIGHT_RESULT | wx.html2.WEBVIEW_FIND_ENTIRE_WORD | wx.html2.WEBVIEW_FIND_MATCH_CASE)
        elif match_case and highlight_results:
            self.html_window.Find(self.find_field.GetValue(), flags=wx.html2.WEBVIEW_FIND_HIGHLIGHT_RESULT | wx.html2.WEBVIEW_FIND_MATCH_CASE)
        elif entire_word and highlight_results:
            self.html_window.Find(self.find_field.GetValue(), flags=wx.html2.WEBVIEW_FIND_HIGHLIGHT_RESULT | wx.html2.WEBVIEW_FIND_ENTIRE_WORD)
        elif entire_word and match_case:
            self.html_window.Find(self.find_field.GetValue(), flags=wx.html2.WEBVIEW_FIND_ENTIRE_WORD | wx.html2.WEBVIEW_FIND_MATCH_CASE)
        elif match_case:
            self.html_window.Find(self.find_field.GetValue(), flags=wx.html2.WEBVIEW_FIND_MATCH_CASE)
        elif entire_word:
            self.html_window.Find(self.find_field.GetValue(), flags=wx.html2.WEBVIEW_FIND_ENTIRE_WORD)
        elif highlight_results:
            self.html_window.Find(self.find_field.GetValue(), flags=wx.html2.WEBVIEW_FIND_HIGHLIGHT_RESULT)
        else:
            self.html_window.Find(self.find_field.GetValue())

    def close_find(self, event):
        self.pagesizer.Hide(self.find_container)
        self.pagesizer.Layout()

    def click_on_url_field(self, event):
        self.url_field.SelectAll()

    def enable_devtools(self, event):
        self.html_window.EnableAccessToDevTools(self.menu_devtools.IsChecked())
    
    def enable_contextmenu(self, event):
        self.html_window.EnableContextMenu(self.menu_contextmenu.IsChecked())
    
    def enable_historyenabled(self, event):
        history_enabled = self.menu_historyenabled.IsChecked()
        self.html_window.EnableHistory(history_enabled)
        self.remember_history = history_enabled

        if self.html_window.CanGoBack():
            self.back.Enable()
        else:
            self.back.Disable()
        if self.html_window.CanGoForward():
            self.forward.Enable()
        else:
            self.forward.Disable()
    
    def load_url(self, url=None):
        self.back.Disable()
        self.forward.Disable()
        self.reload.SetLabel("×")
        self.url_field.Disable()
        
        if url:
            self.html_window.LoadURL(url)

    def tab_back(self, event):
        self.load_url()
        self.html_window.GoBack()
                
    def tab_foward(self, event):
        self.load_url()
        self.html_window.GoForward()
        
    def tab_reload(self, event):
        if self.reload.GetLabel() == "×":
            self.html_window.Stop()
            self.post_load_config()
        else:
            self.load_url()
            self.html_window.Reload()
        
    def tab_new(self, event):
        page = WebPage(self.parent,self.visited)
        self.parent.AddPage(page, caption="Loading", select=True)

    def post_load_config(self, event=None):
        url = self.html_window.GetCurrentURL()

        self.url_field.SetValue(url)
        if self.parent.GetSelection() == self.parent.GetPageIndex(self):
            self.on_select()
        
        if self.html_window.CanGoBack():
            self.back.Enable()
        if self.html_window.CanGoForward():
            self.forward.Enable()
        self.reload.SetLabel("⟳")
        self.url_field.Enable()
        if self.remember_history:
            self.visited.append(url)

    def change_title(self, event):
        title = self.html_window.GetCurrentTitle()
        current_page_index = self.parent.GetPageIndex(self)
        self.parent.SetPageText(current_page_index, title)    

    def on_select(self):
        title = self.html_window.GetCurrentTitle()
        divider = ""
        if title != "":
            divider = " - "
        self.frame.SetTitle("WxPyBrowser"+divider+title)

    def on_close(self):
        if self.html_window.IsBusy():
            self.html_window.Stop()

    def open_in_new_tab(self, event):
        page = WebPage(self.parent, self.visited, url=event.URL)
        self.parent.AddPage(page, caption="Loading")
    
    def loadpage(self, event):
        url = self.url_field.GetValue()
        if (not url.startswith("https://") and not url.startswith("http://")):
            url = "https://"+url
        self.load_url(url)


        
class Browser(wx.Frame):
    def __init__(self, parent, title):
        wx.Frame.__init__(self, parent, title=title)

        self.SetMinSize((900, 500))
        
        self.history_closed = []
        self.load_notebook()
        
    def load_notebook(self):
        self.panel = panel = wx.Panel(self)
        box = wx.BoxSizer(wx.HORIZONTAL)
        self.notebook = notebook = wx.aui.AuiNotebook(panel, style=wx.aui.AUI_NB_DEFAULT_STYLE)
        box.Add(notebook, proportion=True, flag=wx.EXPAND)
        panel.SetSizer(box)
        
        notebook.AddPage(WebPage(self.notebook, self.history_closed), caption="Loading", select=True)

        notebook.Bind(wx.aui.EVT_AUINOTEBOOK_PAGE_CLOSE, self.on_page_close)
        notebook.Bind(wx.aui.EVT_AUINOTEBOOK_PAGE_CHANGED, self.on_page_select)
        

    def on_page_close(self, event):
        event.Skip()
        try:
            page = self.notebook.GetCurrentPage()
            page.on_close()
        except:
            pass
        if self.notebook.GetPageCount() <= 1:
            self.Close()
                

    def on_page_select(self, event):
        self.notebook.GetCurrentPage().on_select()
        

def main():
    app = wx.App()
    browser = Browser(None, title='WxPyBrowser')
    browser.Show()
    app.MainLoop()


if __name__ == '__main__':
    main()