agronholm / sqlacodegen / 19762862247

97%

push

github

Add support for rendering dialect kwargs and info, and introduce keep-dialect-types option (#438)

* [Add support for rendering dialect kwargs and info; introduce keep-dialect-types option

### Summary
- Render `Table`/`Column` dialect-specific kwargs and `info` in
generated code.
- Add `keep-dialect-types` generator option to preserve dialect-specific
column types instead of adapting to generic SQLAlchemy types.

### Motivation
- Some dialect features (e.g., engine/storage params, partitioning,
materialized view metadata) are carried in SQLAlchemy `dialect_kwargs`
or `info` but weren’t emitted by sqlacodegen.
- For custom dialects with their own Column types, the current type
“adaptation” step collapses them into generic SQLAlchemy types, losing
fidelity.

### What’s changed
- Render dialect kwargs and info
- Table (Core): include table-level `dialect_kwargs` and `info` in
`Table(...)` kwargs.
- Column (Core/ORM): include column-level `dialect_kwargs` and `info`
in `Column(...)` / `mapped_column(...)`.
- Declarative: include table-level `dialect_kwargs` and `info` in
`__table_args__` dict.
- New option: keep-dialect-types
  - Generator option: `keep_dialect_types`.
  - CLI flag: `--options=keep_dialect_types`.
- Behavior: gates only the type adaptation step in
`fix_column_types()`; Boolean/Enum inference and PostgreSQL sequence
handling remain intact.

### Usage examples

- Core (Table)

```python
t_orders = Table(
    'orders',
    metadata,
    Column('id', INTEGER, primary_key=True, starrocks_aggr_type='SUM'),
    schema='public',
    comment='orders table',

    info={'table_kind': 'view'}
)
```

- Declarative (`__table_args__`)

```python
class Orders(Base):
    __tablename__ = 'orders'
    __table_args__ = (
        {
            'schema': 'public',
            'comment': 'orders table',
            'info': {'table_kind': 'view'},
	    'starrocks_properties': {'replication_num': '1'},
        }
    )
id: Mapped[INTEGER] = mapped_column(primary_... (continued)

71 of 77 new or added lines in 3 files covered. (92.21%)

1512 of 1553 relevant lines covered (97.36%)

4.86 hits per line