From 61e1b74434bbfe881459b8862b2f2ac181440b17 Mon Sep 17 00:00:00 2001
From: Kovid Goyal <kovid@kovidgoyal.net>
Date: Tue, 8 May 2018 22:32:34 +0530
Subject: [PATCH] Highlight .pyj files as python

---
 kittens/diff/highlight.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/kittens/diff/highlight.py b/kittens/diff/highlight.py
index fe47b7a7c..ffc0bedd5 100644
--- a/kittens/diff/highlight.py
+++ b/kittens/diff/highlight.py
@@ -74,6 +74,9 @@ def initialize_highlighter(style='default'):
 
 
 def highlight_data(code, filename):
+    base, ext = os.path.splitext(filename)
+    if ext.lower() == '.pyj':
+        filename = base + '.py'
     try:
         lexer = get_lexer_for_filename(filename, stripnl=False)
     except ClassNotFound: