fml_doc_gen

Documentation

Documentation Status

Testing

codecov

Package

PyPI Latest Release

Downloads

PyPI Downloads

Supported Versions

Python Version

Authors

  • Farhan Bin Faisal

  • Michael Suriawan

  • Lukman Lateef

Summary

fml_doc_gen is a Python package designed to simplify the process of writing docstrings for user-defined functions. It offers an automated way to generate docstring templates or fully detailed docstrings through integration with the OpenAI API. By streamlining documentation, it helps developers save time, improve code readability, and maintain consistent documentation standards across projects.

Installation

$ pip install fml_doc_gen

Usage

generate_docstring_template can be used to generate docstring template for your function shown below:

import fml_doc_gen
from fml_doc_gen.fml_doc_gen import generate_docstring_template

def square(base: int, pow: int) -> int:
    return base ** pow

print(generate_docstring_template(square, output_file=None))

OpenAI Dependency:

fml_doc_gen requires an OpenAI API key to generate docstrings. You can set your API key using the OPENAI_API_KEY environment variable. Check our our documentation here for more information.

Functions Included

  • generate_docstring_template(func: Callable, output_file: str, auto_generate: bool = False) -> str: The main end-user function that generates either a docstring template with placeholders or a fully detailed docstring if auto_generate is set to True.

  • read_user_function(func: Callable) -> FunctionDTO: Reads and extracts the signature and existing docstring (if any) of a given user-defined function.

  • generate_template(func_signature: str) -> str: Creates a docstring template with placeholders for parameters, return values, and a brief description based on the function signature.

  • write_docstring_to_file(docstring: str, output_file: str) -> None: Writes the generated docstring to a specified file.

Python Ecosystem

fml_doc_gen fits into the Python ecosystem as a specialized tool for automating function documentation. While other Python packages like sphinx and pydoc exist for generating documentation from docstrings, there are no widely known packages that automatically generate docstrings themselves using AI. This makes fml_doc_gen unique in its approach by leveraging OpenAI’s capabilities to produce high-quality docstrings with minimal effort from the developer.

If you are aware of a similar tool, feel free to contribute to our documentation by suggesting it!

Discussion and Improvements

Found a bug? or have an idea on how to improve fml_doc_gen further? Feel free to open an issue with the appropriate label (bug or features) under the issue section here.

Running Tests

To run tests and check code coverage, use the following command in the project root:

poetry run pytest --cov=fml_doc_gen --cov-report=term-missing tests/

Contributing

Interested in contributing? Check out the contributing guidelines. Please note that this project is released with a Code of Conduct. By contributing to this project, you agree to abide by its terms.

License

fml_doc_gen was created by Farhan Faisal, Lukman Lateef, and Michael Suriawan. It is licensed under the terms of the MIT license.

Credits

fml_doc_gen was created with cookiecutter and the py-pkgs-cookiecutter template.