Skip to main content

Automatic model code generator for SQLAlchemy

Project description

Build Status Code Coverage

This is a tool that reads the structure of an existing database and generates the appropriate SQLAlchemy model code, using the declarative style if possible.

This tool was written as a replacement for sqlautocode, which was suffering from several issues (including, but not limited to, incompatibility with Python 3 and the latest SQLAlchemy version).

Features

  • Supports SQLAlchemy 2.x

  • Produces declarative code that almost looks like it was hand written

  • Produces PEP 8 compliant code

  • Accurately determines relationships, including many-to-many, one-to-one

  • Automatically detects joined table inheritance

  • Excellent test coverage

Installation

To install, do:

pip install sqlacodegen

To include support for the PostgreSQL CITEXT extension type (which should be considered as tested only under a few environments) specify the citext extra:

pip install sqlacodegen[citext]

To include support for the PostgreSQL GEOMETRY, GEOGRAPHY, and RASTER types (which should be considered as tested only under a few environments) specify the geoalchemy2 extra:

To include support for the PostgreSQL PGVECTOR extension type, specify the pgvector extra:

pip install sqlacodegen[pgvector]
pip install sqlacodegen[geoalchemy2]

Quickstart

At the minimum, you have to give sqlacodegen a database URL. The URL is passed directly to SQLAlchemy’s create_engine() method so please refer to SQLAlchemy’s documentation for instructions on how to construct a proper URL.

Examples:

sqlacodegen postgresql:///some_local_db
sqlacodegen --generator tables mysql+pymysql://user:password@localhost/dbname
sqlacodegen --generator dataclasses sqlite:///database.db
# --engine-arg values are parsed with ast.literal_eval
sqlacodegen oracle+oracledb://user:pass@127.0.0.1:1521/XE --engine-arg thick_mode=True
sqlacodegen oracle+oracledb://user:pass@127.0.0.1:1521/XE --engine-arg thick_mode=True --engine-arg connect_args='{"user": "user", "dsn": "..."}'

To see the list of generic options:

sqlacodegen --help

Available generators

The selection of a generator determines the

The following built-in generators are available:

  • tables (only generates Table objects, for those who don’t want to use the ORM)

  • declarative (the default; generates classes inheriting from declarative_base()

  • dataclasses (generates dataclass-based models; v1.4+ only)

  • sqlmodels (generates model classes for SQLModel)

Generator-specific options

The following options can be turned on by passing them using --options (multiple values must be delimited by commas, e.g. --options noconstraints,nobidi):

  • tables

    • noconstraints: ignore constraints (foreign key, unique etc.)

    • nocomments: ignore table/column comments

    • noindexes: ignore indexes

    • noidsuffix: prevent the special naming logic for single column many-to-one and one-to-one relationships (see Relationship naming logic for details)

    • include_dialect_options: render a table’ dialect options, such as starrocks_partition for StarRocks’ specific options.

    • keep_dialect_types: preserve dialect-specific column types instead of adapting to generic SQLAlchemy types.

  • declarative

    • all the options from tables

    • use_inflect: use the inflect library when naming classes and relationships (turning plural names into singular; see below for details)

    • nojoined: don’t try to detect joined-class inheritance (see below for details)

    • nobidi: generate relationships in a unidirectional fashion, so only the many-to-one or first side of many-to-many relationships gets a relationship attribute, as on v2.X

  • dataclasses

    • all the options from declarative

  • sqlmodels

    • all the options from declarative

Model class generators

The code generators that generate classes try to generate model classes whenever possible. There are two circumstances in which a Table is generated instead:

  • the table has no primary key constraint (which is required by SQLAlchemy for every model class)

  • the table is an association table between two other tables (see below for the specifics)

Model class naming logic

By default, table names are converted to valid PEP 8 compliant class names by replacing all characters unsuitable for Python identifiers with _. Then, each valid parts (separated by underscores) are title cased and then joined together, eliminating the underscores. So, example_name becomes ExampleName.

If the use_inflect option is used, the table name (which is assumed to be in English) is converted to singular form using the “inflect” library. For example, sales_invoices becomes SalesInvoice. Since table names are not always in English, and the inflection process is far from perfect, inflection is disabled by default.

Relationship detection logic

Relationships are detected based on existing foreign key constraints as follows:

  • many-to-one: a foreign key constraint exists on the table

  • one-to-one: same as many-to-one, but a unique constraint exists on the column(s) involved

  • many-to-many: (not implemented on the sqlmodel generator) an association table is found to exist between two tables

A table is considered an association table if it satisfies all of the following conditions:

  1. has exactly two foreign key constraints

  2. all its columns are involved in said constraints

Relationship naming logic

Relationships are typically named based on the table name of the opposite class. For example, if a class has a relationship to another class with the table named companies, the relationship would be named companies (unless the use_inflect option was enabled, in which case it would be named company in the case of a many-to-one or one-to-one relationship).

A special case for single column many-to-one and one-to-one relationships, however, is if the column is named like employer_id. Then the relationship is named employer due to that _id suffix.

For self referential relationships, the reverse side of the relationship will be named with the _reverse suffix appended to it.

Customizing code generation logic

If the built-in generators with all their options don’t quite do what you want, you can customize the logic by subclassing one of the existing code generator classes. Override whichever methods you need, and then add an entry point in the sqlacodegen.generators namespace that points to your new class. Once the entry point is in place (you typically have to install the project with pip install), you can use --generator <yourentrypoint> to invoke your custom code generator.

For examples, you can look at sqlacodegen’s own entry points in its pyproject.toml.

Getting help

If you have problems or other questions, you should start a discussion on the sqlacodegen discussion forum. As an alternative, you could also try your luck on the sqlalchemy room on Gitter.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

sqlacodegen-3.2.0.tar.gz (45.4 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

sqlacodegen-3.2.0-py3-none-any.whl (23.8 kB view details)

Uploaded Python 3

File details

Details for the file sqlacodegen-3.2.0.tar.gz.

File metadata

  • Download URL: sqlacodegen-3.2.0.tar.gz
  • Upload date:
  • Size: 45.4 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.7

File hashes

Hashes for sqlacodegen-3.2.0.tar.gz
Algorithm Hash digest
SHA256 ce2a8a3c422bee40a58e9469befd4aec87a160f37158acbcc1c05e9d1c2e786d
MD5 7f48d1b81bbb8907d09ebaf31f92a4bf
BLAKE2b-256 4e4a6f8be0ac3e1fd498fa1d62c7855da8cb5127ce5ade1b6510694f08ed26b5

See more details on using hashes here.

Provenance

The following attestation bundles were made for sqlacodegen-3.2.0.tar.gz:

Publisher: publish.yml on agronholm/sqlacodegen

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file sqlacodegen-3.2.0-py3-none-any.whl.

File metadata

  • Download URL: sqlacodegen-3.2.0-py3-none-any.whl
  • Upload date:
  • Size: 23.8 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.7

File hashes

Hashes for sqlacodegen-3.2.0-py3-none-any.whl
Algorithm Hash digest
SHA256 bec3da027832a1513487ca9169224b64b5578c6b54f1bed31b4821584de010cf
MD5 a423d4ca37f9db2a6816f828b96042f5
BLAKE2b-256 4fa4c79e6c5514a21cb26abc028bf343853e588eab9ff68a03ba4bb47c751a8e

See more details on using hashes here.

Provenance

The following attestation bundles were made for sqlacodegen-3.2.0-py3-none-any.whl:

Publisher: publish.yml on agronholm/sqlacodegen

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page