Remove underscore from module names

telethon_generator/generators/docs.py

@@ -5,7 +5,7 @@ import re
 import shutil
 from collections import defaultdict
 
-from ..docs_writer import DocsWriter
+from ..docswriter import DocsWriter
 from ..parsers import TLObject
 from ..utils import snake_to_camel_case
 

telethon_generator/generators/errors.py

@@ -1,6 +1,3 @@
-import itertools
-
-
 def generate_errors(errors, f):
     # Exact/regex match to create {CODE: ErrorClassName}
     exact_match = []
@@ -20,7 +17,7 @@ def generate_errors(errors, f):
             exact_match.append(error)
 
     # Imports and new subclass creation
-    f.write('from .rpc_base_errors import RPCError, {}\n'
+    f.write('from .rpcbaseerrors import RPCError, {}\n'
             .format(", ".join(sorted(import_base))))
 
     for cls, int_code in sorted(create_base.items(), key=lambda t: t[1]):

telethon_generator/generators/tlobject.py

@@ -6,7 +6,7 @@ import struct
 from collections import defaultdict
 from zlib import crc32
 
-from ..source_builder import SourceBuilder
+from ..sourcebuilder import SourceBuilder
 from ..utils import snake_to_camel_case
 
 AUTO_GEN_NOTICE = \
@@ -650,7 +650,7 @@ def generate_tlobjects(tlobjects, layer, import_depth, output_dir):
     _write_modules(get_file('types'), import_depth, 'TLObject',
                    namespace_types, type_constructors)
 
-    filename = os.path.join(get_file('all_tlobjects.py'))
+    filename = os.path.join(get_file('alltlobjects.py'))
     with open(filename, 'w', encoding='utf-8') as file:
         with SourceBuilder(file) as builder:
             _write_all_tlobjects(tlobjects, layer, builder)
@@ -663,6 +663,6 @@ def clean_tlobjects(output_dir):
         if os.path.isdir(d):
             shutil.rmtree(d)
 
-    tl = get_file('all_tlobjects.py')
+    tl = get_file('alltlobjects.py')
     if os.path.isfile(tl):
         os.remove(tl)