mirror of
https://github.com/pallets-eco/flask-debugtoolbar.git
synced 2026-01-01 19:19:33 -06:00
1c594b9ba3
- Rename the package to flask_debugtoolbar - Fix importing of the panels so that the views in the panels are correctly registered in the blueprint There is still one major problem in the sqlalchemy panel where we are not able to get the SQLAlchemy reference
57 lines
1.6 KiB
Python
57 lines
1.6 KiB
Python
import os.path
|
|
import sys
|
|
|
|
try:
|
|
from pygments import highlight
|
|
from pygments.formatters import HtmlFormatter
|
|
from pygments.lexers import SqlLexer
|
|
from pygments.styles import get_style_by_name
|
|
PYGMENT_STYLE = get_style_by_name('colorful')
|
|
HAVE_PYGMENTS = True
|
|
except ImportError:
|
|
HAVE_PYGMENTS = False
|
|
|
|
|
|
from flask import current_app
|
|
|
|
def format_fname(value):
|
|
# If the value is not an absolute path, the it is a builtin or
|
|
# a relative file (thus a project file).
|
|
if not os.path.isabs(value):
|
|
if value.startswith(('{', '<')):
|
|
return value
|
|
if value.startswith('.' + os.path.sep):
|
|
return value
|
|
return '.' + os.path.sep + value
|
|
|
|
# If the file is absolute and within the project root handle it as
|
|
# a project file
|
|
if value.startswith(current_app.root_path):
|
|
return "." + value[len(current_app.root_path):]
|
|
|
|
# Loop through sys.path to find the longest match and return
|
|
# the relative path from there.
|
|
paths = sys.path
|
|
prefix = None
|
|
prefix_len = 0
|
|
for path in sys.path:
|
|
new_prefix = os.path.commonprefix([path, value])
|
|
if len(new_prefix) > prefix_len:
|
|
prefix = new_prefix
|
|
prefix_len = len(prefix)
|
|
|
|
if not prefix.endswith(os.path.sep):
|
|
prefix_len -= 1
|
|
path = value[prefix_len:]
|
|
return '<%s>' % path
|
|
|
|
def format_sql(query, args):
|
|
if not HAVE_PYGMENTS:
|
|
return query
|
|
|
|
return highlight(
|
|
query,
|
|
SqlLexer(encoding='utf-8'),
|
|
HtmlFormatter(encoding='utf-8', noclasses=True, style=PYGMENT_STYLE))
|
|
|