Add the ability to disable Discord emojis

Author: jay3332

pilmoji/__init__.py

@@ -5,5 +5,5 @@
 from .aio import AsyncPilmoji, AsyncRequester
 
 
-__version__ = '1.3.1'
+__version__ = '1.3.2'
 __author__ = 'jay3332'

pilmoji/aio/core.py

@@ -17,10 +17,19 @@ class AsyncPilmoji(BasePilmoji):
     """
     The synchronous emoji renderer.
     """
-    def __init__(self, image: Image.Image, *, session: typing.Optional[ClientSession] = None, loop: typing.Optional[asyncio.AbstractEventLoop] = None, use_microsoft_emoji: bool = False):
+    def __init__(
+            self,
+            image: Image.Image,
+            *,
+            session: typing.Optional[ClientSession] = None,
+            loop: typing.Optional[asyncio.AbstractEventLoop] = None,
+            use_microsoft_emoji: bool = False,
+            render_discord_emoji: bool = True
+    ):
         if not isinstance(image, Image.Image):
             raise TypeError(f'Image must be of type Image, got {type(image).__name__!r} instead.')
 
+        self.render_discord_emoji: bool = render_discord_emoji
         self.http: AsyncRequester = AsyncRequester(session=session, loop=loop, _microsoft=use_microsoft_emoji)
         self.image: Image.Image = image
         self.draw = ImageDraw.Draw(image)
@@ -84,7 +93,10 @@ async def text(self,
                     if node['type'] == 'twemoji':
                         stream = await self.http.get_twemoji(content)
                     else:
-                        stream = await self.http.get_discord_emoji(content)
+                        stream = (
+                            await self.http.get_discord_emoji(content)
+                            if self.render_discord_emoji else None
+                        )
 
                     if not stream:
                         self.draw.text((x, y), content, *args, **kwargs)

pilmoji/sync/core.py

@@ -16,10 +16,18 @@ class Pilmoji(BasePilmoji):
     """
     The synchronous emoji renderer.
     """
-    def __init__(self, image: Image.Image, *, session: typing.Optional[Session] = None, use_microsoft_emoji: bool = False):
+    def __init__(
+            self,
+            image: Image.Image,
+            *,
+            session: typing.Optional[Session] = None,
+            use_microsoft_emoji: bool = False,
+            render_discord_emoji: bool = True
+    ):
         if not isinstance(image, Image.Image):
             raise TypeError(f'Image must be of type Image, got {type(image).__name__!r} instead.')
 
+        self.render_discord_emoji: bool = render_discord_emoji
         self.http: Requester = Requester(session=session, _microsoft=use_microsoft_emoji)
         self.image: Image.Image = image
         self.draw = ImageDraw.Draw(image)
@@ -83,7 +91,10 @@ def text(self,
                     if node['type'] == 'twemoji':
                         stream = self.http.get_twemoji(content)
                     else:
-                        stream = self.http.get_discord_emoji(content)
+                        stream = (
+                            self.http.get_discord_emoji(content)
+                            if self.render_discord_emoji else None
+                        )
 
                     if not stream:
                         self.draw.text((x, y), content, *args, **kwargs)

tests/tests.py

@@ -0,0 +1,15 @@
+from pilmoji import Pilmoji
+from PIL import Image, ImageFont
+
+
+my_string = '😎😎😎😎😎😎'
+
+with Image.new('RGB', (550, 80), (255, 255, 255)) as image:
+    font = ImageFont.truetype('arial.ttf', 24)
+
+    with Pilmoji(image, use_microsoft_emoji=True) as pilmoji:
+        w, h = Pilmoji.getsize(my_string, font)
+        pilmoji.text((10, 10), my_string, (0, 0, 0), font)
+        pilmoji.text((10+w, 10), 'hi', (0, 0, 0), font)
+
+    image.show()