Skip to content

ostr00000/freecad-stubs

Repository files navigation

freecad-stubs

PyPI version GitHub license PyPI pyversions pre-commit Code style: black Checked with pyright Checked with mypy

Python stubs generated from FreeCAD source code.

Install stubs

You can install all stubs for freeCAD packages by running:

python -m pip install freecad-stubs

If you installed this package, then you probably develop some FreeCAD components/scripts. You should remember to set up your IDE by adding path to the real FreeCAD libraries, especially for modules written mainly in python (ex. Draft), because this package does not provide stubs for that module 😕 (yet - you may contribute).

Also note that App or Gui are only aliases available in FreeCAD. To fully use advantage of stubs you should always import a module you are referencing (ex. import FreeCADGui as Gui).

Generating stubs manually

Package on pypi always has stubs generated for the newest freecad (master branch). If you need older version you may try to install specific version from pypi or manually run a tool from this repository to generate stubs.

Stub source

Stubs are generated based on several available info:

  • *Py.xml files - mainly docstrings, argument names, properties.
  • corresponding *PyImp.cpp file - argument types are extracted from C code.
  • other *.cpp files - functions or modules.

Unfortunately not all typing information may be generated. For example some objects are added dynamically. There are also many special cases therefore not all object are correctly mapped. Moreover, some of C function has errors - invalid types, missing arguments (you can see more these errors if you change logger flag in configuration file freecad-stubs/lib/freecad_stub_gen/config.py)

Stub Generation

  1. Clone freecad repository.

    git clone https://github.com/FreeCAD/FreeCAD.git

    Warning: FreeCAD repository has over 1.29 GB

    You may download only these required folders:

    • /src/App
    • /src/Base
    • /src/Gui
    • /src/Main
    • /src/Mod
  2. Clone this repository

    git clone https://github.com/ostr00000/freecad-stubs
  3. Configure paths In the file freecad-stubs/lib/freecad_stub_gen/config.py set desired configuration:

    • SOURCE_DIR - src folder from FreeCAD repository,
    • TARGET_DIR - target folder where stubs should be generated.

      Warning: TARGET_DIR folder and its content may be removed when generating stubs.

  4. Run the main file from this project in Python

    python freecad-stubs/lib/freecad_stub_gen/__main__.py

    Required python version: >=3.11.

Adding stubs to python path

At this point stubs must be already generated. There are a lot of possible methods. This is only example:

  1. Copy stubs from TARGET_DIR to location of your choice. For example:.

    cp -r ./freecad_stubs "$HOME/.local/lib/python3.9/freecad_stubs"
  2. Add location with stubs to python search path. For example:

    echo "$HOME/.local/lib/python3.9/freecad_stubs" > "$HOME/.local/lib/python3.9/site-packages/freecad_stubs.pth"

Implementation progress

Stub source

  • generate class stub (xml files):
    • property in xml,
    • dynamic property added in cpp,
    • method (+ static/class method),
    • rich comparison (ex. __eq__ method),
    • number protocol (ex. __add__ method),
  • generate class stub (cpp files):
    • dynamically added in init_type(),
    • qt slots,
    • qt signals,
  • generate functions stub (cpp files):
    • declared in PyMethodDef array,
    • dynamically added in module constructor (subclass Py::ExtensionModule),

Stub quality

  • found class/function/method name,
  • copy docstring (+ generate property docstring),
  • guess argument names from C code or from docstrings,
  • found function/method argument types based on c-api,
  • argument default values,
  • function/method return type,
  • raised exception in docstrings,
  • property getter type,
  • property setter type,
  • add comment "This class can be imported" for importable classes,
  • dynamically generated exceptions,

About

Python stubs generated from FreeCAD source code

Topics

Resources

License

Stars

Watchers

Forks

Packages

No packages published