env.py 2.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980
  1. from __future__ import with_statement
  2. from logging.config import fileConfig
  3. from sqlalchemy import engine_from_config
  4. from sqlalchemy import pool
  5. from alembic import context
  6. # this is the Alembic Config object, which provides
  7. # access to the values within the .ini file in use.
  8. config = context.config
  9. # Interpret the config file for Python logging.
  10. # This line sets up loggers basically.
  11. fileConfig(config.config_file_name)
  12. # add your model's MetaData object here
  13. # for 'autogenerate' support
  14. # from myapp import mymodel
  15. # target_metadata = mymodel.Base.metadata
  16. import os
  17. import sys
  18. sys.path.append(os.getcwd())
  19. from app import models
  20. target_metadata = models.Base.metadata
  21. # other values from the config, defined by the needs of env.py,
  22. # can be acquired:
  23. # my_important_option = config.get_main_option("my_important_option")
  24. # ... etc.
  25. def run_migrations_offline():
  26. """Run migrations in 'offline' mode.
  27. This configures the context with just a URL
  28. and not an Engine, though an Engine is acceptable
  29. here as well. By skipping the Engine creation
  30. we don't even need a DBAPI to be available.
  31. Calls to context.execute() here emit the given string to the
  32. script output.
  33. """
  34. url = config.get_main_option("sqlalchemy.url")
  35. context.configure(
  36. url=url, target_metadata=target_metadata, literal_binds=True
  37. )
  38. with context.begin_transaction():
  39. context.run_migrations()
  40. def run_migrations_online():
  41. """Run migrations in 'online' mode.
  42. In this scenario we need to create an Engine
  43. and associate a connection with the context.
  44. """
  45. connectable = engine_from_config(
  46. config.get_section(config.config_ini_section),
  47. prefix="sqlalchemy.",
  48. poolclass=pool.NullPool,
  49. )
  50. with connectable.connect() as connection:
  51. context.configure(
  52. connection=connection, target_metadata=target_metadata
  53. )
  54. with context.begin_transaction():
  55. context.run_migrations()
  56. if context.is_offline_mode():
  57. run_migrations_offline()
  58. else:
  59. run_migrations_online()