From 1b9d6aac06ec6dce2a6ef915c9ba516fdfd1a9da Mon Sep 17 00:00:00 2001
From: Lonami Exo <totufals@hotmail.com>
Date: Thu, 4 Oct 2018 16:15:51 +0200
Subject: [PATCH] Gzip only content related data

---
 telethon/network/mtprotostate.py | 4 ++--
 telethon/tl/core/gzippacked.py   | 5 ++---
 2 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/telethon/network/mtprotostate.py b/telethon/network/mtprotostate.py
index 4b3dbefa..3495ae8a 100644
--- a/telethon/network/mtprotostate.py
+++ b/telethon/network/mtprotostate.py
@@ -79,9 +79,9 @@ class MTProtoState:
         msg_id = self._get_new_msg_id()
         seq_no = self._get_seq_no(content_related)
         if after_id is None:
-            body = GzipPacked.gzip_if_smaller(data)
+            body = GzipPacked.gzip_if_smaller(content_related, data)
         else:
-            body = GzipPacked.gzip_if_smaller(
+            body = GzipPacked.gzip_if_smaller(content_related,
                 bytes(InvokeAfterMsgRequest(after_id, data)))
 
         buffer.write(struct.pack('<qii', msg_id, seq_no, len(body)))
diff --git a/telethon/tl/core/gzippacked.py b/telethon/tl/core/gzippacked.py
index 3f887022..906c2c67 100644
--- a/telethon/tl/core/gzippacked.py
+++ b/telethon/tl/core/gzippacked.py
@@ -11,15 +11,14 @@ class GzipPacked(TLObject):
         self.data = data
 
     @staticmethod
-    def gzip_if_smaller(data):
+    def gzip_if_smaller(content_related, data):
         """Calls bytes(request), and based on a certain threshold,
            optionally gzips the resulting data. If the gzipped data is
            smaller than the original byte array, this is returned instead.
 
            Note that this only applies to content related requests.
         """
-        # TODO Only content-related requests should be gzipped
-        if len(data) > 512:
+        if content_related and len(data) > 512:
             gzipped = bytes(GzipPacked(data))
             return gzipped if len(gzipped) < len(data) else data
         else: