From 3585fb8cc6e1e4cabb572076a29931b444dd634b Mon Sep 17 00:00:00 2001
From: Lonami Exo <totufals@hotmail.com>
Date: Tue, 4 Jul 2017 11:02:32 +0200
Subject: [PATCH] Fix setup.py for source distributions

---
 setup.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 05f2a9cd6..6a80d95b6 100755
--- a/setup.py
+++ b/setup.py
@@ -18,7 +18,6 @@
 # Always prefer setuptools over distutils
 from setuptools import find_packages, setup
 
-from telethon_generator.tl_generator import TLGenerator
 try:
     from telethon import TelegramClient
 except ImportError:
@@ -27,6 +26,7 @@
 
 if __name__ == '__main__':
     if len(argv) >= 2 and argv[1] == 'gen_tl':
+        from telethon_generator.tl_generator import TLGenerator
         generator = TLGenerator('telethon/tl')
         if generator.tlobjects_exist():
             print('Detected previous TLObjects. Cleaning...')
@@ -39,6 +39,7 @@
         print('Done.')
 
     elif len(argv) >= 2 and argv[1] == 'clean_tl':
+        from telethon_generator.tl_generator import TLGenerator
         print('Cleaning...')
         TLGenerator('telethon/tl').clean_tlobjects()
         print('Done.')