mirror of
https://github.com/langgenius/dify.git
synced 2026-03-10 01:46:14 +08:00
fcc71124b1
* phase1:uuid stringdefault timestamp date * phase2:json adapt * phase3:BinaryData,ARRAY,Text, * fix sa.String() * fix defalut valud * models adapt * models adapt supplement * fix pg unique dialect * supplement * migration fix * migration fix * date sql adapt * model adapt fix * config adapt * migration fix * fix * fix anotation in .env.example * config feat * config fix * .env.example adapt fix
65 lines
3.1 KiB
Python
65 lines
3.1 KiB
Python
from datetime import datetime
|
|
|
|
import sqlalchemy as sa
|
|
from sqlalchemy import String
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
|
|
from libs.uuid_utils import uuidv7
|
|
|
|
from .base import Base
|
|
from .engine import db
|
|
from .types import LongText, StringUUID
|
|
|
|
|
|
class DatasourceOauthParamConfig(Base): # type: ignore[name-defined]
|
|
__tablename__ = "datasource_oauth_params"
|
|
__table_args__ = (
|
|
db.PrimaryKeyConstraint("id", name="datasource_oauth_config_pkey"),
|
|
db.UniqueConstraint("plugin_id", "provider", name="datasource_oauth_config_datasource_id_provider_idx"),
|
|
)
|
|
|
|
id = mapped_column(StringUUID, default=lambda: str(uuidv7()))
|
|
plugin_id: Mapped[str] = mapped_column(String(255), nullable=False)
|
|
provider: Mapped[str] = mapped_column(String(255), nullable=False)
|
|
system_credentials: Mapped[dict] = mapped_column(sa.JSON, nullable=False)
|
|
|
|
|
|
class DatasourceProvider(Base):
|
|
__tablename__ = "datasource_providers"
|
|
__table_args__ = (
|
|
db.PrimaryKeyConstraint("id", name="datasource_provider_pkey"),
|
|
db.UniqueConstraint("tenant_id", "plugin_id", "provider", "name", name="datasource_provider_unique_name"),
|
|
db.Index("datasource_provider_auth_type_provider_idx", "tenant_id", "plugin_id", "provider"),
|
|
)
|
|
id = mapped_column(StringUUID, default=lambda: str(uuidv7()))
|
|
tenant_id = mapped_column(StringUUID, nullable=False)
|
|
name: Mapped[str] = mapped_column(String(255), nullable=False)
|
|
provider: Mapped[str] = mapped_column(String(128), nullable=False)
|
|
plugin_id: Mapped[str] = mapped_column(String(255), nullable=False)
|
|
auth_type: Mapped[str] = mapped_column(String(255), nullable=False)
|
|
encrypted_credentials: Mapped[dict] = mapped_column(sa.JSON, nullable=False)
|
|
avatar_url: Mapped[str] = mapped_column(LongText, nullable=True, default="default")
|
|
is_default: Mapped[bool] = mapped_column(sa.Boolean, nullable=False, server_default=db.text("false"))
|
|
expires_at: Mapped[int] = mapped_column(sa.Integer, nullable=False, server_default="-1")
|
|
|
|
created_at: Mapped[datetime] = mapped_column(sa.DateTime, nullable=False, default=datetime.now)
|
|
updated_at: Mapped[datetime] = mapped_column(sa.DateTime, nullable=False, default=datetime.now)
|
|
|
|
|
|
class DatasourceOauthTenantParamConfig(Base):
|
|
__tablename__ = "datasource_oauth_tenant_params"
|
|
__table_args__ = (
|
|
db.PrimaryKeyConstraint("id", name="datasource_oauth_tenant_config_pkey"),
|
|
db.UniqueConstraint("tenant_id", "plugin_id", "provider", name="datasource_oauth_tenant_config_unique"),
|
|
)
|
|
|
|
id = mapped_column(StringUUID, default=lambda: str(uuidv7()))
|
|
tenant_id = mapped_column(StringUUID, nullable=False)
|
|
provider: Mapped[str] = mapped_column(String(255), nullable=False)
|
|
plugin_id: Mapped[str] = mapped_column(String(255), nullable=False)
|
|
client_params: Mapped[dict] = mapped_column(sa.JSON, nullable=False, default={})
|
|
enabled: Mapped[bool] = mapped_column(sa.Boolean, nullable=False, default=False)
|
|
|
|
created_at: Mapped[datetime] = mapped_column(sa.DateTime, nullable=False, default=datetime.now)
|
|
updated_at: Mapped[datetime] = mapped_column(sa.DateTime, nullable=False, default=datetime.now)
|