dbos 1.14.0a8__py3-none-any.whl → 1.15.0a1__py3-none-any.whl

dbos/_alembic_migrations/versions/83f3732ae8e7_workflow_timeout.py

@@ -1,44 +0,0 @@
-"""workflow_timeout
-
-Revision ID: 83f3732ae8e7
-Revises: f4b9b32ba814
-Create Date: 2025-04-16 17:05:36.642395
-
-"""
-
-from typing import Sequence, Union
-
-import sqlalchemy as sa
-from alembic import op
-
-# revision identifiers, used by Alembic.
-revision: str = "83f3732ae8e7"
-down_revision: Union[str, None] = "f4b9b32ba814"
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
-
-
-def upgrade() -> None:
-    op.add_column(
-        "workflow_status",
-        sa.Column(
-            "workflow_timeout_ms",
-            sa.BigInteger(),
-            nullable=True,
-        ),
-        schema="dbos",
-    )
-    op.add_column(
-        "workflow_status",
-        sa.Column(
-            "workflow_deadline_epoch_ms",
-            sa.BigInteger(),
-            nullable=True,
-        ),
-        schema="dbos",
-    )
-
-
-def downgrade() -> None:
-    op.drop_column("workflow_status", "workflow_deadline_epoch_ms", schema="dbos")
-    op.drop_column("workflow_status", "workflow_timeout_ms", schema="dbos")

dbos/_alembic_migrations/versions/933e86bdac6a_add_queue_priority.py

@@ -1,35 +0,0 @@
-"""add queue priority
-
-Revision ID: 933e86bdac6a
-Revises: 27ac6900c6ad
-Create Date: 2025-04-25 18:17:40.462737
-
-"""
-
-from typing import Sequence, Union
-
-import sqlalchemy as sa
-from alembic import op
-
-# revision identifiers, used by Alembic.
-revision: str = "933e86bdac6a"
-down_revision: Union[str, None] = "27ac6900c6ad"
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
-
-
-def upgrade() -> None:
-    op.add_column(
-        "workflow_queue",
-        sa.Column(
-            "priority",
-            sa.Integer(),
-            nullable=False,
-            server_default=sa.text("'0'::int"),
-        ),
-        schema="dbos",
-    )
-
-
-def downgrade() -> None:
-    op.drop_column("workflow_queue", "priority", schema="dbos")

dbos/_alembic_migrations/versions/a3b18ad34abe_added_triggers.py

@@ -1,72 +0,0 @@
-"""
-Add triggers.
-
-Revision ID: a3b18ad34abe
-Revises: 5c361fc04708
-Create Date: 2024-07-21 13:07:09.814989
-"""
-
-from typing import Sequence, Union
-
-from alembic import op
-
-# revision identifiers, used by Alembic.
-revision: str = "a3b18ad34abe"
-down_revision: Union[str, None] = "5c361fc04708"
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
-
-
-def upgrade() -> None:
-
-    op.execute(
-        """
-      CREATE OR REPLACE FUNCTION dbos.notifications_function() RETURNS TRIGGER AS $$
-      DECLARE
-          payload text := NEW.destination_uuid || '::' || NEW.topic;
-      BEGIN
-          PERFORM pg_notify('dbos_notifications_channel', payload);
-          RETURN NEW;
-      END;
-      $$ LANGUAGE plpgsql;
-               """
-    )
-    op.execute(
-        """
-      CREATE TRIGGER dbos_notifications_trigger
-      AFTER INSERT ON dbos.notifications
-      FOR EACH ROW EXECUTE FUNCTION dbos.notifications_function();
-               """
-    )
-    op.execute(
-        """
-      CREATE OR REPLACE FUNCTION dbos.workflow_events_function() RETURNS TRIGGER AS $$
-      DECLARE
-          payload text := NEW.workflow_uuid || '::' || NEW.key;
-      BEGIN
-          PERFORM pg_notify('dbos_workflow_events_channel', payload);
-          RETURN NEW;
-      END;
-      $$ LANGUAGE plpgsql;
-               """
-    )
-    op.execute(
-        """
-    CREATE TRIGGER dbos_workflow_events_trigger
-      AFTER INSERT ON dbos.workflow_events
-      FOR EACH ROW EXECUTE FUNCTION dbos.workflow_events_function();
-               """
-    )
-    # ### end Alembic commands ###
-
-
-def downgrade() -> None:
-    op.execute(
-        "DROP TRIGGER IF EXISTS dbos_notifications_trigger ON dbos.notifications;"
-    )
-    op.execute("DROP FUNCTION IF EXISTS dbos.notifications_function;")
-    op.execute(
-        "DROP TRIGGER IF EXISTS dbos_workflow_events_trigger ON dbos.workflow_events;"
-    )
-    op.execute("DROP FUNCTION IF EXISTS dbos.workflow_events_function;")
-    # ### end Alembic commands ###

dbos/_alembic_migrations/versions/d76646551a6b_job_queue_limiter.py

@@ -1,43 +0,0 @@
-"""
-Adjust workflow queue to add columns for rate limiter.
-
-Revision ID: d76646551a6b
-Revises: 50f3227f0b4b
-Create Date: 2024-09-25 14:48:10.218015
-
-"""
-
-from typing import Sequence, Union
-
-import sqlalchemy as sa
-from alembic import op
-
-# revision identifiers, used by Alembic.
-revision: str = "d76646551a6b"
-down_revision: Union[str, None] = "50f3227f0b4b"
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
-
-
-def upgrade() -> None:
-    op.add_column(
-        "job_queue",
-        sa.Column(
-            "started_at_epoch_ms",
-            sa.BigInteger(),
-        ),
-        schema="dbos",
-    )
-    op.add_column(
-        "job_queue",
-        sa.Column(
-            "completed_at_epoch_ms",
-            sa.BigInteger(),
-        ),
-        schema="dbos",
-    )
-
-
-def downgrade() -> None:
-    op.drop_column("job_queue", "started_at_epoch_ms", schema="dbos")
-    op.drop_column("job_queue", "completed_at_epoch_ms", schema="dbos")

dbos/_alembic_migrations/versions/d76646551a6c_workflow_queue.py

@@ -1,28 +0,0 @@
-"""workflow_queue
-
-Revision ID: d76646551a6c
-Revises: d76646551a6b
-Create Date: 2024-09-27 12:00:00.0
-
-"""
-
-from typing import Sequence, Union
-
-import sqlalchemy as sa
-from alembic import op
-
-# revision identifiers, used by Alembic.
-revision: str = "d76646551a6c"
-down_revision: Union[str, None] = "d76646551a6b"
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
-
-
-def upgrade() -> None:
-    op.rename_table("job_queue", "workflow_queue", schema="dbos")
-    op.execute("CREATE VIEW dbos.job_queue AS SELECT * FROM dbos.workflow_queue;")
-
-
-def downgrade() -> None:
-    op.execute("DROP VIEW dbos.job_queue;")
-    op.rename_table("workflow_queue", "job_queue", schema="dbos")

dbos/_alembic_migrations/versions/d994145b47b6_consolidate_inputs.py

@@ -1,30 +0,0 @@
-"""consolidate_inputs
-
-Revision ID: d994145b47b6
-Revises: 66478e1b95e5
-Create Date: 2025-05-23 08:09:15.515009
-
-"""
-
-from typing import Sequence, Union
-
-import sqlalchemy as sa
-from alembic import op
-
-# revision identifiers, used by Alembic.
-revision: str = "d994145b47b6"
-down_revision: Union[str, None] = "66478e1b95e5"
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
-
-
-def upgrade() -> None:
-    op.add_column(
-        "workflow_status",
-        sa.Column("inputs", sa.Text(), nullable=True),
-        schema="dbos",
-    )
-
-
-def downgrade() -> None:
-    op.drop_column("workflow_status", "inputs", schema="dbos")

dbos/_alembic_migrations/versions/eab0cc1d9a14_job_queue.py

@@ -1,56 +0,0 @@
-"""
-Add workflow queue table.
-
-Revision ID: eab0cc1d9a14
-Revises: a3b18ad34abe
-Create Date: 2024-09-13 14:50:00.531294
-
-"""
-
-from typing import Sequence, Union
-
-import sqlalchemy as sa
-from alembic import op
-
-# revision identifiers, used by Alembic.
-revision: str = "eab0cc1d9a14"
-down_revision: Union[str, None] = "a3b18ad34abe"
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
-
-
-def upgrade() -> None:
-    op.create_table(
-        "job_queue",
-        sa.Column("workflow_uuid", sa.Text(), nullable=False),
-        sa.Column("queue_name", sa.Text(), nullable=False),
-        sa.Column(
-            "created_at_epoch_ms",
-            sa.BigInteger(),
-            server_default=sa.text(
-                "(EXTRACT(epoch FROM now()) * 1000::numeric)::bigint"
-            ),
-            nullable=False,
-            primary_key=True,
-        ),
-        sa.ForeignKeyConstraint(
-            ["workflow_uuid"],
-            ["dbos.workflow_status.workflow_uuid"],
-            onupdate="CASCADE",
-            ondelete="CASCADE",
-        ),
-        schema="dbos",
-    )
-    op.add_column(
-        "workflow_status",
-        sa.Column(
-            "queue_name",
-            sa.Text(),
-        ),
-        schema="dbos",
-    )
-
-
-def downgrade() -> None:
-    op.drop_table("job_queue", schema="dbos")
-    op.drop_column("workflow_status", "queue_name", schema="dbos")

dbos/_alembic_migrations/versions/f4b9b32ba814_functionname_childid_op_outputs.py

@@ -1,46 +0,0 @@
-"""functionname_childid_op_outputs
-
-Revision ID: f4b9b32ba814
-Revises: 04ca4f231047
-Create Date: 2025-03-21 14:32:43.091074
-
-"""
-
-from typing import Sequence, Union
-
-import sqlalchemy as sa
-from alembic import op
-
-# revision identifiers, used by Alembic.
-revision: str = "f4b9b32ba814"
-down_revision: Union[str, None] = "04ca4f231047"
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
-
-
-def upgrade() -> None:
-    op.add_column(
-        "operation_outputs",
-        sa.Column(
-            "function_name",
-            sa.Text(),
-            nullable=False,
-            server_default="",
-        ),
-        schema="dbos",
-    )
-
-    op.add_column(
-        "operation_outputs",
-        sa.Column(
-            "child_workflow_id",
-            sa.Text(),
-            nullable=True,
-        ),
-        schema="dbos",
-    )
-
-
-def downgrade() -> None:
-    op.drop_column("operation_outputs", "function_name", schema="dbos")
-    op.drop_column("operation_outputs", "child_workflow_id", schema="dbos")

dbos/_templates/dbos-db-starter/alembic.ini

@@ -1,116 +0,0 @@
-# A generic, single database configuration.
-
-[alembic]
-# path to migration scripts
-# Use forward slashes (/) also on windows to provide an os agnostic path
-script_location = migrations
-
-# template used to generate migration file names; The default value is %%(rev)s_%%(slug)s
-# Uncomment the line below if you want the files to be prepended with date and time
-# see https://alembic.sqlalchemy.org/en/latest/tutorial.html#editing-the-ini-file
-# for all available tokens
-file_template = %%(year)d_%%(month).2d_%%(day).2d_%%(hour).2d%%(minute).2d%%(second).2d_%%(slug)s
-
-# sys.path path, will be prepended to sys.path if present.
-# defaults to the current working directory.
-prepend_sys_path = .
-
-# timezone to use when rendering the date within the migration file
-# as well as the filename.
-# If specified, requires the python>=3.9 or backports.zoneinfo library.
-# Any required deps can installed by adding `alembic[tz]` to the pip requirements
-# string value is passed to ZoneInfo()
-# leave blank for localtime
-# timezone =
-
-# max length of characters to apply to the "slug" field
-# truncate_slug_length = 40
-
-# set to 'true' to run the environment during
-# the 'revision' command, regardless of autogenerate
-# revision_environment = false
-
-# set to 'true' to allow .pyc and .pyo files without
-# a source .py file to be detected as revisions in the
-# versions/ directory
-# sourceless = false
-
-# version location specification; This defaults
-# to migrations/versions.  When using multiple version
-# directories, initial revisions must be specified with --version-path.
-# The path separator used here should be the separator specified by "version_path_separator" below.
-# version_locations = %(here)s/bar:%(here)s/bat:migrations/versions
-
-# version path separator; As mentioned above, this is the character used to split
-# version_locations. The default within new alembic.ini files is "os", which uses os.pathsep.
-# If this key is omitted entirely, it falls back to the legacy behavior of splitting on spaces and/or commas.
-# Valid values for version_path_separator are:
-#
-# version_path_separator = :
-# version_path_separator = ;
-# version_path_separator = space
-version_path_separator = os  # Use os.pathsep. Default configuration used for new projects.
-
-# set to 'true' to search source files recursively
-# in each "version_locations" directory
-# new in Alembic version 1.10
-# recursive_version_locations = false
-
-# the output encoding used when revision files
-# are written from script.py.mako
-# output_encoding = utf-8
-
-# sqlalchemy.url is set in env.py from the dbos-config.yaml file
-
-
-[post_write_hooks]
-# post_write_hooks defines scripts or Python functions that are run
-# on newly generated revision scripts.  See the documentation for further
-# detail and examples
-
-# format using "black" - use the console_scripts runner, against the "black" entrypoint
-# hooks = black
-# black.type = console_scripts
-# black.entrypoint = black
-# black.options = -l 79 REVISION_SCRIPT_FILENAME
-
-# lint with attempts to fix using "ruff" - use the exec runner, execute a binary
-# hooks = ruff
-# ruff.type = exec
-# ruff.executable = %(here)s/.venv/bin/ruff
-# ruff.options = --fix REVISION_SCRIPT_FILENAME
-
-# Logging configuration
-[loggers]
-keys = root,sqlalchemy,alembic
-
-[handlers]
-keys = console
-
-[formatters]
-keys = generic
-
-[logger_root]
-level = WARN
-handlers = console
-qualname =
-
-[logger_sqlalchemy]
-level = WARN
-handlers =
-qualname = sqlalchemy.engine
-
-[logger_alembic]
-level = INFO
-handlers =
-qualname = alembic
-
-[handler_console]
-class = StreamHandler
-args = (sys.stderr,)
-level = NOTSET
-formatter = generic
-
-[formatter_generic]
-format = %(levelname)-5.5s [%(name)s] %(message)s
-datefmt = %H:%M:%S

dbos/_templates/dbos-db-starter/migrations/env.py.dbos

@@ -1,85 +0,0 @@
-import re
-import os
-from logging.config import fileConfig
-
-from alembic import context
-from sqlalchemy import engine_from_config, pool
-
-# this is the Alembic Config object, which provides
-# access to the values within the .ini file in use.
-config = context.config
-
-# Interpret the config file for Python logging.
-# This line sets up loggers basically.
-if config.config_file_name is not None:
-    fileConfig(config.config_file_name)
-
-# Programmatically set the sqlalchemy.url field to the DBOS application database URL
-conn_string = os.environ.get("DBOS_DATABASE_URL", "postgresql+psycopg://postgres:dbos@localhost:5432/${default_db_name}?connect_timeout=5")
-# Alembic requires the % in URL-escaped parameters to itself be escaped to %%
-escaped_conn_string = re.sub(
-    r"%(?=[0-9A-Fa-f]{2})",
-    "%%",
-    conn_string,
-)
-config.set_main_option("sqlalchemy.url", escaped_conn_string)
-
-# add your model's MetaData object here
-# for 'autogenerate' support
-from ${package_name}.schema import metadata
-target_metadata = metadata
-
-# other values from the config, defined by the needs of env.py,
-# can be acquired:
-# my_important_option = config.get_main_option("my_important_option")
-# ... etc.
-
-
-def run_migrations_offline() -> None:
-    """Run migrations in 'offline' mode.
-
-    This configures the context with just a URL
-    and not an Engine, though an Engine is acceptable
-    here as well.  By skipping the Engine creation
-    we don't even need a DBAPI to be available.
-
-    Calls to context.execute() here emit the given string to the
-    script output.
-
-    """
-    url = config.get_main_option("sqlalchemy.url")
-    context.configure(
-        url=url,
-        target_metadata=target_metadata,
-        literal_binds=True,
-        dialect_opts={"paramstyle": "named"},
-    )
-
-    with context.begin_transaction():
-        context.run_migrations()
-
-
-def run_migrations_online() -> None:
-    """Run migrations in 'online' mode.
-
-    In this scenario we need to create an Engine
-    and associate a connection with the context.
-
-    """
-    connectable = engine_from_config(
-        config.get_section(config.config_ini_section, {}),
-        prefix="sqlalchemy.",
-        poolclass=pool.NullPool,
-    )
-
-    with connectable.connect() as connection:
-        context.configure(connection=connection, target_metadata=target_metadata)
-
-        with context.begin_transaction():
-            context.run_migrations()
-
-
-if context.is_offline_mode():
-    run_migrations_offline()
-else:
-    run_migrations_online()

dbos/_templates/dbos-db-starter/migrations/script.py.mako

@@ -1,26 +0,0 @@
-"""${message}
-
-Revision ID: ${up_revision}
-Revises: ${down_revision | comma,n}
-Create Date: ${create_date}
-
-"""
-from typing import Sequence, Union
-
-from alembic import op
-import sqlalchemy as sa
-${imports if imports else ""}
-
-# revision identifiers, used by Alembic.
-revision: str = ${repr(up_revision)}
-down_revision: Union[str, None] = ${repr(down_revision)}
-branch_labels: Union[str, Sequence[str], None] = ${repr(branch_labels)}
-depends_on: Union[str, Sequence[str], None] = ${repr(depends_on)}
-
-
-def upgrade() -> None:
-    ${upgrades if upgrades else "pass"}
-
-
-def downgrade() -> None:
-    ${downgrades if downgrades else "pass"}

dbos/_templates/dbos-db-starter/migrations/versions/2024_07_31_180642_init.py

@@ -1,35 +0,0 @@
-"""
-Initialize application database.
-
-Revision ID: c6b516e182b2
-Revises:
-Create Date: 2024-07-31 18:06:42.500040
-"""
-
-from typing import Sequence, Union
-
-import sqlalchemy as sa
-from alembic import op
-
-# revision identifiers, used by Alembic.
-revision: str = "c6b516e182b2"
-down_revision: Union[str, None] = None
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
-
-
-def upgrade() -> None:
-    # ### commands auto generated by Alembic - please adjust! ###
-    op.create_table(
-        "dbos_hello",
-        sa.Column("greet_count", sa.Integer(), autoincrement=True, nullable=False),
-        sa.Column("name", sa.String(), nullable=False),
-        sa.PrimaryKeyConstraint("greet_count"),
-    )
-    # ### end Alembic commands ###
-
-
-def downgrade() -> None:
-    # ### commands auto generated by Alembic - please adjust! ###
-    op.drop_table("dbos_hello")
-    # ### end Alembic commands ###