Make relative imports work in gen scripts even when directly executed

gen/apc_parsers.py

@@ -7,6 +7,11 @@ import sys
 from collections import defaultdict
 from typing import Any, DefaultDict, Dict, FrozenSet, List, Tuple, Union
 
+if __name__ == '__main__' and not __package__:
+    import __main__
+    __main__.__package__ = 'gen'
+    sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
 KeymapType = Dict[str, Tuple[str, Union[FrozenSet[str], str]]]
 
 

gen/config.py

@@ -10,6 +10,11 @@ from typing import List
 
 from kitty.conf.generate import write_output
 
+if __name__ == '__main__' and not __package__:
+    import __main__
+    __main__.__package__ = 'gen'
+    sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
 
 def patch_color_list(path: str, colors: List[str], name: str, spc: str = '    ') -> None:
     with open(path, 'r+') as f:

gen/cursors.py

@@ -6,6 +6,11 @@ import subprocess
 import sys
 from typing import List
 
+if __name__ == '__main__' and not __package__:
+    import __main__
+    __main__.__package__ = 'gen'
+    sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
 from .key_constants import patch_file
 
 # References for these names:

gen/go_code.py

@@ -49,6 +49,12 @@ from kitty.rc.base import RemoteCommand, all_command_names, command_for_name
 from kitty.remote_control import global_options_spec
 from kitty.rgb import color_names
 
+if __name__ == '__main__' and not __package__:
+    import __main__
+    __main__.__package__ = 'gen'
+    sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
+
 changed: List[str] = []
 
 

gen/key_constants.py

@@ -7,6 +7,11 @@ import sys
 from pprint import pformat
 from typing import Any, Dict, List, Union
 
+if __name__ == '__main__' and not __package__:
+    import __main__
+    __main__.__package__ = 'gen'
+    sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
 functional_key_defs = '''# {{{
 # kitty                     XKB                         macVK  macU
 escape                      Escape                      0x35   -

gen/srgb_lut.py

@@ -6,6 +6,11 @@ import sys
 from functools import lru_cache
 from typing import List
 
+if __name__ == '__main__' and not __package__:
+    import __main__
+    __main__.__package__ = 'gen'
+    sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
 
 def to_linear(a: float) -> float:
     if a <= 0.04045:

gen/wcwidth.py

@@ -26,6 +26,12 @@ from typing import (
 )
 from urllib.request import urlopen
 
+if __name__ == '__main__' and not __package__:
+    import __main__
+    __main__.__package__ = 'gen'
+    sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
+
 non_characters = frozenset(range(0xfffe, 0x10ffff, 0x10000))
 non_characters |= frozenset(range(0xffff, 0x10ffff + 1, 0x10000))
 non_characters |= frozenset(range(0xfdd0, 0xfdf0))