|
|
@ -1,7 +1,7 @@
|
|
|
|
import inspect
|
|
|
|
import inspect
|
|
|
|
import os.path
|
|
|
|
import os.path
|
|
|
|
import tempfile
|
|
|
|
import tempfile
|
|
|
|
import mypy.stubgen as sg
|
|
|
|
import pybind11_stubgen as sg
|
|
|
|
|
|
|
|
|
|
|
|
package = 'quicktex'
|
|
|
|
package = 'quicktex'
|
|
|
|
prefix = '_'
|
|
|
|
prefix = '_'
|
|
|
@ -20,10 +20,15 @@ def find_submodules(pkg):
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
if __name__ == "__main__":
|
|
|
|
find_submodules(__import__(prefix + package))
|
|
|
|
find_submodules(__import__(prefix + package))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
pkgdir = os.path.abspath(os.curdir)
|
|
|
|
|
|
|
|
|
|
|
|
with tempfile.TemporaryDirectory() as out:
|
|
|
|
with tempfile.TemporaryDirectory() as out:
|
|
|
|
|
|
|
|
|
|
|
|
# generate stubs using mypy Stubgen
|
|
|
|
# generate stubs using mypy Stubgen
|
|
|
|
sg.generate_stubs(sg.parse_options(['-o', out, '-p', prefix + package]))
|
|
|
|
sg.main(['-o', out, '--root-module-suffix', "", prefix + package])
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
os.curdir = pkgdir
|
|
|
|
|
|
|
|
|
|
|
|
# walk resulting stubs and move them to their new location
|
|
|
|
# walk resulting stubs and move them to their new location
|
|
|
|
for root, dirs, files in os.walk(out):
|
|
|
|
for root, dirs, files in os.walk(out):
|
|
|
@ -31,6 +36,9 @@ if __name__ == "__main__":
|
|
|
|
# location of the extension module's stub file
|
|
|
|
# location of the extension module's stub file
|
|
|
|
ext_module = os.path.relpath(root, out)
|
|
|
|
ext_module = os.path.relpath(root, out)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if stub_name.split('.')[-1] != 'pyi':
|
|
|
|
|
|
|
|
continue
|
|
|
|
|
|
|
|
|
|
|
|
if stub_name != '__init__.pyi':
|
|
|
|
if stub_name != '__init__.pyi':
|
|
|
|
ext_module = os.path.join(ext_module, os.path.splitext(stub_name)[0])
|
|
|
|
ext_module = os.path.join(ext_module, os.path.splitext(stub_name)[0])
|
|
|
|
|
|
|
|
|
|
|
|