Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
code_view.py
# coding: utf-8
import ui
from pygments import highlight
from pygments.formatters import HtmlFormatter
from pygments.styles import get_style_by_name
from pygments.lexers import get_lexer_by_name
from pygments.styles import get_all_styles
class CodeDisplay(ui.View):
def __init__(self, style = 'colorful', use_linenos = True,
code = '', *args, **kwargs):
ui.View.__init__(self, *args, **kwargs)
self.code = code
self.web_view = None
self.style = style
self.use_linenos = use_linenos
self._create_view_items()
self.load_code_str()
def _create_view_items(self):
self.web_view = ui.WebView()
self.web_view.size_to_fit()
self.web_view.scales_page_to_fit = True
self.add_subview(self.web_view)
def layout(self):
self.web_view.frame = self.bounds
def get_style_names(self):
#return [s for s in get_all_styles()]
# i stopped with list comp. i think get_all_styles(), was causing
# Pythonista to crash. tried many times. its was during editing, so
# i think some code executing on the import. creating the problem.
# but just guessing
lst = []
for style in get_all_styles():
lst.append(style)
return sorted(lst)
def load_code_str(self):
# load python code as a string into the webview
self.web_view.load_html(self.get_html(self.code))
def get_html(self, code, use_linenos = None,
style = None, hl_lines = None):
# returns the html from pygments
if not use_linenos: use_linenos = self.use_linenos
if not style: style = self.style
lexer = get_lexer_by_name("python", stripall=False)
formatter = HtmlFormatter(linenos=use_linenos , style=style, full=True)
return highlight(code, lexer, formatter)
if __name__ == '__main__':
# example usage
import editor
f = (0,0, ui.get_screen_size()[0] * .8, ui.get_screen_size()[1] * .8)
path = editor.get_path()
with open(path) as file:
code = file.read()
cd = CodeDisplay(frame = f, name = 'Snippet x', code = code)
cd.present('sheet', animated = False)
# left this in just so easy to see the different style names
print cd.get_style_names()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.