engineering/unrhodecode
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

vcs: refactor get_lexer for nodes so it can be used in external code.

Browse source
This commit is contained in:
Marcin Kuzminski 2017-01-25 16:09:47 +01:00
parent c1855fb60c
commit 67dfb2f772
1 changed files with 14 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

22
rhodecode/lib/vcs/nodes.py
View file

@ -446,25 +446,23 @@ class FileNode(Node):
def mimetype_main(self):
return self.mimetype.split('/')[0]
@LazyProperty
def lexer(self):
"""
Returns pygment's lexer class. Would try to guess lexer taking file's
content, name and mimetype.
"""
@classmethod
def get_lexer(cls, filename, content=None):
from pygments import lexers
extension = filename.split('.')[-1]
lexer = None
try:
lexer = lexers.guess_lexer_for_filename(
self.name, self.content, stripnl=False)
filename, content, stripnl=False)
except lexers.ClassNotFound:
lexer = None
# try our EXTENSION_MAP
if not lexer:
try:
lexer_class = LANGUAGES_EXTENSIONS_MAP.get(self.extension)
lexer_class = LANGUAGES_EXTENSIONS_MAP.get(extension)
if lexer_class:
lexer = lexers.get_lexer_by_name(lexer_class[0])
except lexers.ClassNotFound:
@ -475,6 +473,14 @@ class FileNode(Node):
return lexer
@LazyProperty
def lexer(self):
"""
Returns pygment's lexer class. Would try to guess lexer taking file's
content, name and mimetype.
"""
return self.get_lexer(self.name, self.content)
@LazyProperty
def lexer_alias(self):
"""