ext_database.py 529 B

1234567891011121314151617181920
  1. from flask_sqlalchemy import SQLAlchemy
  2. from sqlalchemy import MetaData
  3. from dify_app import DifyApp
  4. POSTGRES_INDEXES_NAMING_CONVENTION = {
  5. "ix": "%(column_0_label)s_idx",
  6. "uq": "%(table_name)s_%(column_0_name)s_key",
  7. "ck": "%(table_name)s_%(constraint_name)s_check",
  8. "fk": "%(table_name)s_%(column_0_name)s_fkey",
  9. "pk": "%(table_name)s_pkey",
  10. }
  11. metadata = MetaData(naming_convention=POSTGRES_INDEXES_NAMING_CONVENTION)
  12. db = SQLAlchemy(metadata=metadata)
  13. def init_app(app: DifyApp):
  14. db.init_app(app)