Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-07-09 03:04:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

Styling cleanups

Browse Source
This commit is contained in:
Kovid Goyal 2010-09-21 21:05:23 -06:00
parent fd9164f768
commit 7f472f742e
1 changed files with 18 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
src/calibre/utils/pyconsole/formatter.py
View File

@ -8,18 +8,20 @@ __docformat__ = 'restructuredtext en'
from PyQt4.Qt import QTextCharFormat, QFont, QBrush, QColor from PyQt4.Qt import QTextCharFormat, QFont, QBrush, QColor
from pygments.formatter import Formatter as PF from pygments.formatter import Formatter as PF
from pygments.token import Token, Generic from pygments.token import Token, Generic, string_to_tokentype
class Formatter(object): class Formatter(object):
def __init__(self, prompt, continuation, **options): def __init__(self, prompt, continuation, style='default'):
if len(prompt) != len(continuation): if len(prompt) != len(continuation):
raise ValueError('%r does not have the same length as %r' % raise ValueError('%r does not have the same length as %r' %
(prompt, continuation)) (prompt, continuation))
self.prompt, self.continuation = prompt, continuation self.prompt, self.continuation = prompt, continuation
self.set_style(style)
pf = PF(**options) def set_style(self, style):
pf = PF(style=style)
self.styles = {} self.styles = {}
self.normal = self.base_fmt() self.normal = self.base_fmt()
self.background_color = pf.style.background_color self.background_color = pf.style.background_color
@ -27,6 +29,7 @@ class Formatter(object):
for ttype, ndef in pf.style: for ttype, ndef in pf.style:
fmt = self.base_fmt() fmt = self.base_fmt()
fmt.setProperty(fmt.UserProperty, str(ttype))
if ndef['color']: if ndef['color']:
fmt.setForeground(QBrush(QColor('#%s'%ndef['color']))) fmt.setForeground(QBrush(QColor('#%s'%ndef['color'])))
fmt.setUnderlineColor(QColor('#%s'%ndef['color'])) fmt.setUnderlineColor(QColor('#%s'%ndef['color']))
@ -49,6 +52,14 @@ class Formatter(object):
QTextEdit { color: %s; background-color: %s } QTextEdit { color: %s; background-color: %s }
'''%(self.color, self.background_color) '''%(self.color, self.background_color)
def get_fmt(self, token):
if type(token) != type(Token.Generic):
token = string_to_tokentype(token)
fmt = self.styles.get(token, None)
if fmt is None:
fmt = self.base_fmt()
fmt.setProperty(fmt.UserProperty, str(token))
return fmt
def base_fmt(self): def base_fmt(self):
fmt = QTextCharFormat() fmt = QTextCharFormat()
@ -59,7 +70,7 @@ class Formatter(object):
cursor.insertText(raw, self.normal) cursor.insertText(raw, self.normal)
def render_syntax_error(self, tb, cursor): def render_syntax_error(self, tb, cursor):
fmt = self.styles[Token.Error] fmt = self.get_fmt(Token.Error)
cursor.insertText(tb, fmt) cursor.insertText(tb, fmt)
def render(self, tokens, cursor): def render(self, tokens, cursor):
@ -84,7 +95,9 @@ class Formatter(object):
def render_prompt(self, is_continuation, cursor): def render_prompt(self, is_continuation, cursor):
pr = self.continuation if is_continuation else self.prompt pr = self.continuation if is_continuation else self.prompt
fmt = self.styles[Generic.Prompt] fmt = self.get_fmt(Generic.Prompt)
if fmt is None:
fmt = self.base_fmt()
cursor.insertText(pr, fmt) cursor.insertText(pr, fmt)