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

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

  • 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.0.0.tar.gz (39.2 kB view details)

Uploaded Source

Built Distribution

sqlacodegen-3.0.0-py3-none-any.whl (21.7 kB view details)

Uploaded Python 3

File details

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

File metadata

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

File hashes

Hashes for sqlacodegen-3.0.0.tar.gz
Algorithm Hash digest
SHA256 785bc9307b787a9ae7ffc1460abf0a718862b6fa47f304259503612298e523f2
MD5 d8728a2b99f3491f74f4a6e8c7ef2878
BLAKE2b-256 93ed633a47383bc1640ae94625b9df1e6f7f461644c0c539c7e69c7654ab1239

See more details on using hashes here.

Provenance

The following attestation bundles were made for sqlacodegen-3.0.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.0.0-py3-none-any.whl.

File metadata

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

File hashes

Hashes for sqlacodegen-3.0.0-py3-none-any.whl
Algorithm Hash digest
SHA256 ce3042fff03ceb0426cdfb7237a4a13781166718ae22b63156e5f4b3232137e4
MD5 560002aac60cc28b60585431fcada78a
BLAKE2b-256 926a0398fa0ab13142d8836258814dc6069c57f459afa5adfd60859e8657fd6a

See more details on using hashes here.

Provenance

The following attestation bundles were made for sqlacodegen-3.0.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 Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page
pFad - Phonifier reborn

Pfad - The Proxy pFad of © 2024 Garber Painting. All rights reserved.

Note: This service is not intended for secure transactions such as banking, social media, email, or purchasing. Use at your own risk. We assume no liability whatsoever for broken pages.


Alternative Proxies:

Alternative Proxy

pFad Proxy

pFad v3 Proxy

pFad v4 Proxy