azalea/NetworkDiffBot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[+] Make font configurable and add test case

Browse Source
This commit is contained in:
Hykilpikonna
2020-12-01 11:19:13 -05:00
parent 141abdf85f
commit 7a03a182d2
2 changed files with 30 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/constants.py
+1
View File
@@ -1,4 +1,5 @@
import os import os
token = os.environ['TG_TOKEN'] token = os.environ['TG_TOKEN']
font = os.getenv('FONT', None)
dbPath = 'database.json' dbPath = 'database.json'
src/utils.py
+29 -1
View File
@@ -9,6 +9,8 @@ from pygments.formatters.img import ImageFormatter
from pygments.lexers import * from pygments.lexers import *
from telegram.ext import CommandHandler from telegram.ext import CommandHandler
from src.constants import font
def toJson(obj): def toJson(obj):
return json.dumps(obj, default=lambda o: o.__dict__, sort_keys=True, indent=4) return json.dumps(obj, default=lambda o: o.__dict__, sort_keys=True, indent=4)
@@ -89,7 +91,7 @@ def render(message):
message = unquote(message) message = unquote(message)
lexer = get_lexer_by_name("diff", stripall=True) lexer = get_lexer_by_name("diff", stripall=True)
# lexer = guess_lexer(message) # lexer = guess_lexer(message)
formatter = ImageFormatter(font_name="sarasa-mono-cl-regular", style="colorful", font_size=12, line_pad=4) formatter = ImageFormatter(font_name=font, style="colorful", font_size=12, line_pad=4)
result = syntax_highlight(message, lexer, formatter, outfile=None) result = syntax_highlight(message, lexer, formatter, outfile=None)
return result return result
@@ -108,3 +110,29 @@ def sendRequest(req):
return text return text
except requests.exceptions.ConnectionError as e: except requests.exceptions.ConnectionError as e:
return "Error: ConnectionError!\n" + str(e) return "Error: ConnectionError!\n" + str(e)
if __name__ == '__main__':
before = wrap("""
<li><a href="Etterna-0.70.3-Darwin.dmg">Etterna-0.70.3-Darwin.dmg</a></li>
<li><a href="favicon-dark.svg">favicon-dark.svg</a></li>
<li><a href="Kant-%20short%20form%20%28no%20Kingdom%20of%20Ends%29.pdf">Kant- short form (no Kingdom of Ends).pdf</a></li>
<li><a href="kdeconnect-kde-master-985-macos-64-clang%20%281%29.dmg">kdeconnect-kde-master-985-macos-64-clang (1).dmg</a></li>
<li><a href="Keep/">Keep/</a></li>
<li><a href="Media/">Media/</a></li>
<li><a href="osu%21macOS%20Agent中文.app/">osu!macOS Agent中文.app/</a></li>""")
after = wrap("""
<li><a href="Etterna-0.70.3-Darwin.dmg">Etterna-0.70.3-Darwin.dmg</a></li>
<li><a href="favicon-dark.svg">favicon-dark.svg</a></li>
<li><a href="Kant-z%20short%20form%20%28no%20Kingdom%20of%20Ends%29.pdf">Kant- short form (no Kingdom of Ends).pdf</a></li>
<li><a href="kdeconnect-kde-master-985-macos-64-clang%20%281%29.dmg">kdeconnect-kde-master-985-macos-64-clang (111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111).dmg</a></li>
<li><a href="Keep/">Keep/</a></li>
<li><a href="Media/">Media/</a></li>
<li><a href="osu%21macOS%20Agent中文.app/">osu!macOS Agent中文.app/</a></li>""")
diffRaw = difflib.unified_diff(before.splitlines(1), after.splitlines(1))
diff = ''.join(diffRaw)
r = render(diff)
f = open('test.png', 'wb')
f.write(r)
f.close()