Support for SQLAlchemy 2.0

[Averylamp]

I've just started a new project with SQLAlchemy 2.0 and am now looking to integrate a migration tool into the mix and saw that Alembic was the SQLAlchemy supported tool. I wasn't able to find anything on whether Alembic has been updated to work with 2.0 yet or if it should work out of the box currently and am trying to track down which piece currently doesn't work (could be CockroachDB Dialect or something else specific). Is Alembic supposed to work with SQLAlchemy 2.0 out of the box?

INFO  [alembic.runtime.migration] Context impl CockroachDBImpl.
INFO  [alembic.runtime.migration] Will assume non-transactional DDL.
Traceback (most recent call last):
  File "/home/avery/.local/bin/alembic", line 8, in <module>
    sys.exit(main())
  File "/home/avery/.local/lib/python3.10/site-packages/alembic/config.py", line 590, in main
    CommandLine(prog=prog).main(argv=argv)
  File "/home/avery/.local/lib/python3.10/site-packages/alembic/config.py", line 584, in main
    self.run_cmd(cfg, options)
  File "/home/avery/.local/lib/python3.10/site-packages/alembic/config.py", line 561, in run_cmd
    fn(
  File "/home/avery/.local/lib/python3.10/site-packages/alembic/command.py", line 378, in upgrade
    script.run_env()
  File "/home/avery/.local/lib/python3.10/site-packages/alembic/script/base.py", line 569, in run_env
    util.load_python_file(self.dir, "env.py")
  File "/home/avery/.local/lib/python3.10/site-packages/alembic/util/pyfiles.py", line 94, in load_python_file
    module = load_module_py(module_id, path)
  File "/home/avery/.local/lib/python3.10/site-packages/alembic/util/pyfiles.py", line 110, in load_module_py
    spec.loader.exec_module(module)  # type: ignore
  File "<frozen importlib._bootstrap_external>", line 883, in exec_module
  File "<frozen importlib._bootstrap>", line 241, in _call_with_frames_removed
  File "/home/avery/adept/megatron/backend/database/alembic/env.py", line 75, in <module>
    run_migrations_online()
  File "/home/avery/adept/megatron/backend/database/alembic/env.py", line 69, in run_migrations_online
    context.run_migrations()
  File "<string>", line 8, in run_migrations
  File "/home/avery/.local/lib/python3.10/site-packages/alembic/runtime/environment.py", line 867, in run_migrations
    self.get_context().run_migrations(**kw)
  File "/home/avery/.local/lib/python3.10/site-packages/alembic/runtime/migration.py", line 601, in run_migrations
    heads = self.get_current_heads()
  File "/home/avery/.local/lib/python3.10/site-packages/alembic/runtime/migration.py", line 534, in get_current_heads
    if not self._has_version_table():
  File "/home/avery/.local/lib/python3.10/site-packages/alembic/runtime/migration.py", line 551, in _has_version_table
    return sqla_compat._connectable_has_table(
  File "/home/avery/.local/lib/python3.10/site-packages/alembic/util/sqla_compat.py", line 207, in _connectable_has_table
    return inspect(connectable).has_table(tablename, schemaname)
  File "/home/avery/.local/lib/python3.10/site-packages/sqlalchemy/engine/reflection.py", line 430, in has_table
    return self.dialect.has_table(
TypeError: CockroachDBDialect.has_table() got an unexpected keyword argument 'info_cache'

Thanks!

ps: Hope this is the right place for this discussion. I'm just trying to narrow down possibilities while debugging

[CaselIT]

Hi,

Yes alembic supports sqlalchemy v2.
The error you are seeing comes from the external dialect that you are using. I suggest raising the issue with their maintainers

[Averylamp]

Thanks!