diff --git a/src/constants.py b/src/constants.py index ee4983d..5ad458c 100644 --- a/src/constants.py +++ b/src/constants.py @@ -1,4 +1,5 @@ import os token = os.environ['TG_TOKEN'] +font = os.getenv('FONT', None) dbPath = 'database.json' diff --git a/src/utils.py b/src/utils.py index a2be935..1f2ce2d 100644 --- a/src/utils.py +++ b/src/utils.py @@ -9,6 +9,8 @@ from pygments.formatters.img import ImageFormatter from pygments.lexers import * from telegram.ext import CommandHandler +from src.constants import font + def toJson(obj): return json.dumps(obj, default=lambda o: o.__dict__, sort_keys=True, indent=4) @@ -89,7 +91,7 @@ def render(message): message = unquote(message) lexer = get_lexer_by_name("diff", stripall=True) # 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) return result @@ -108,3 +110,29 @@ def sendRequest(req): return text except requests.exceptions.ConnectionError as e: return "Error: ConnectionError!\n" + str(e) + + +if __name__ == '__main__': + before = wrap(""" +
  • Etterna-0.70.3-Darwin.dmg
  • +
  • favicon-dark.svg
  • +
  • Kant- short form (no Kingdom of Ends).pdf
  • +
  • kdeconnect-kde-master-985-macos-64-clang (1).dmg
  • +
  • Keep/
  • +
  • Media/
  • +
  • osu!macOS Agent中文.app/
  • """) + after = wrap(""" +
  • Etterna-0.70.3-Darwin.dmg
  • +
  • favicon-dark.svg
  • +
  • Kant- short form (no Kingdom of Ends).pdf
  • +
  • kdeconnect-kde-master-985-macos-64-clang (111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111).dmg
  • +
  • Keep/
  • +
  • Media/
  • +
  • osu!macOS Agent中文.app/
  • """) + + 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()