mirror of
https://github.com/kovidgoyal/calibre.git
synced 2026-03-10 03:43:40 -04:00
91 lines
2.7 KiB
Python
91 lines
2.7 KiB
Python
#!/usr/bin/env python
|
|
# License: GPL v3 Copyright: 2021, Kovid Goyal <kovid at kovidgoyal.net>
|
|
|
|
import importlib
|
|
import os
|
|
from pprint import pprint
|
|
|
|
QT_WRAPPER = 'PyQt6'
|
|
|
|
base = os.path.dirname(os.path.abspath(__file__))
|
|
module_lists = {
|
|
'core': (
|
|
'QtCore',
|
|
'QtGui',
|
|
'QtWidgets',
|
|
'QtNetwork',
|
|
'QtSvg',
|
|
'QtPrintSupport',
|
|
'QtOpenGL',
|
|
'QtOpenGLWidgets',
|
|
'QtQuick',
|
|
'QtMultimedia',
|
|
'QtMultimediaWidgets',
|
|
'QtTextToSpeech',
|
|
),
|
|
'webengine': (
|
|
'QtWebEngineCore',
|
|
'QtWebEngineWidgets',
|
|
),
|
|
'dbus': (
|
|
'QtDBus',
|
|
),
|
|
}
|
|
|
|
|
|
def scan(name):
|
|
module_names = module_lists[name]
|
|
name_map = {}
|
|
types = []
|
|
for mod_name in module_names:
|
|
mod = importlib.import_module(f'{QT_WRAPPER}.{mod_name}')
|
|
full_name = name_map[mod_name] = mod.__name__
|
|
types.append(f'import {full_name}')
|
|
for obj_name in sorted(dir(mod)):
|
|
if not obj_name.startswith('_') and obj_name not in name_map:
|
|
name_map[obj_name] = full_name
|
|
types.append(f'{obj_name} = {full_name}.{obj_name}')
|
|
with open(f'{base}/{name}.py', 'w') as f:
|
|
print('# autogenerated by __main__.py do not edit', file=f)
|
|
print(f'from .{name}_name_map import module_names, name_map', file=f)
|
|
print('''from .loader import dynamic_load
|
|
|
|
already_imported = {}
|
|
qt_modules = {}
|
|
|
|
def __getattr__(name):
|
|
return dynamic_load(name, name_map, already_imported, qt_modules, module_names)
|
|
''', end='', file=f)
|
|
with open(f'{base}/{name}.pyi', 'w') as f:
|
|
print('# autogenerated by __main__.py do not edit', file=f)
|
|
print('\n'.join(types), file=f)
|
|
if name == 'core':
|
|
module_names += ('sip',)
|
|
mod = importlib.import_module(f'{QT_WRAPPER}.sip')
|
|
name_map['sip'] = mod.__name__
|
|
with open(f'{base}/{name}_name_map.py', 'w') as f:
|
|
print('# autogenerated by __main__.py do not edit', file=f)
|
|
print('name_map =', end=' ', file=f)
|
|
pprint(name_map, stream=f)
|
|
print('module_names = frozenset(', end='', file=f)
|
|
pprint(module_names, stream=f)
|
|
print(')', file=f)
|
|
|
|
|
|
top_level_module_names = ()
|
|
for name, module_name in module_lists.items():
|
|
top_level_module_names += module_name
|
|
scan(name)
|
|
with open(f'{base}/__init__.py', 'w') as f:
|
|
print('# autogenerated by __main__.py do not edit', file=f)
|
|
print('top_level_module_names = ', end='', file=f)
|
|
pprint(top_level_module_names, stream=f)
|
|
print(f'''
|
|
|
|
def __getattr__(name):
|
|
if name in top_level_module_names:
|
|
import importlib
|
|
return importlib.import_module('{QT_WRAPPER}.' + name)
|
|
raise AttributeError(name)
|
|
''', end='', file=f)
|