b3-prosaic-harness-postgres 0.2.1__py3-none-any.whl

b3_prosaic_harness_postgres-0.2.1.dist-info/METADATA

@@ -0,0 +1,222 @@
+Metadata-Version: 2.4
+Name: b3-prosaic-harness-postgres
+Version: 0.2.1
+Summary: Explicit PostgreSQL run storage for Prosaic Harness
+License-Expression: Apache-2.0
+Requires-Python: >=3.11
+Description-Content-Type: text/markdown
+License-File: LICENSE
+License-File: NOTICE
+Requires-Dist: b3-prosaic-harness<0.8,>=0.7
+Requires-Dist: psycopg[binary]<4,>=3.2.3
+Requires-Dist: psycopg-pool<4,>=3.2
+Provides-Extra: test
+Requires-Dist: pytest>=8; extra == "test"
+Requires-Dist: build>=1; extra == "test"
+Requires-Dist: setuptools>=77; extra == "test"
+Requires-Dist: wheel; extra == "test"
+Requires-Dist: packaging; extra == "test"
+Dynamic: license-file
+
+# PostgreSQL persistence for Prosaic Harness
+
+Run on worker A, inspect/resume on worker B, without a shared writable run
+directory. Core remains database-free; install this separate adapter only when
+you need shared durable state. The API is synchronous; a private I/O thread
+bounds network waits. Model execution holds no database connection/transaction.
+The current candidate is `b3-prosaic-harness-postgres` **0.2.0**, selecting
+`b3-prosaic-harness>=0.7,<0.8`. The adapter is maintained in this repository but
+installed and versioned separately, keeping PostgreSQL drivers optional for
+Harness users. Imports and CLI names remain unchanged. Final qualification and
+publication of the b3 release train are pending. The historical
+[Harness v0.6.1 release](https://github.com/B3Cognition/prosaic-harness/releases/tag/v0.6.1)
+retains its original 0.6.1/0.1.1 package names and receipts.
+
+## 1. Install and supply an existing database
+
+Python 3.11+, PostgreSQL 16 or 18 on a writable primary, and libpq 17+ are
+required. `psycopg[binary]` supplies libpq; older libpq is rejected.
+From the exact candidate Harness checkout, use a new environment and a curated
+wheelhouse containing only the manifest-selected b3 candidate wheels:
+
+```sh
+python3 -m venv .venv-postgres
+.venv-postgres/bin/python -m pip install --find-links /absolute/path/to/curated-wheelhouse . ./adapters/postgres
+export PATH="$PWD/.venv-postgres/bin:$PATH"
+prosaic --version
+prosaic-harness-postgres --help
+```
+
+ALWAYS migrate using a fresh environment or image; NEVER install legacy and b3
+distributions together because they own the same import paths. See
+[package migration](../../docs/package-migration.md). The Runtime 0.8 candidate
+must be available before this installation can resolve.
+
+Harness does **not** provision PostgreSQL, create databases/users, start a
+service, or fall back to files. Your operator supplies two DSNs through a secret
+manager: `HARNESS_MIGRATION_DSN` (schema owner with database CREATE permission)
+and `HARNESS_RUNTIME_DSN` (restricted runtime role). Never put DSNs in Markdown,
+YAML, argv, logs or Git. For a local terminal:
+
+```sh
+export HARNESS_MIGRATION_DSN="$(python -c 'import getpass; print(getpass.getpass("Migration DSN: "))')"
+export HARNESS_RUNTIME_DSN="$(python -c 'import getpass; print(getpass.getpass("Runtime DSN: "))')"
+```
+
+Preserve operator TLS/authentication settings, e.g. `sslmode=verify-full` and
+certificate roots. The adapter adds `target_session_attrs=read-write` for primary
+routing; it never weakens TLS.
+
+## 2. Initialize once, then grant runtime access
+
+```sh
+prosaic-harness-postgres init --dsn-env HARNESS_MIGRATION_DSN
+```
+
+Initialization is explicit, transactional, repeatable and concurrency-safe.
+It creates schema/version 1, not your database. It refuses unknown versions,
+foreign-owned, partial or damaged schemas; it does not repair or downgrade them.
+
+As schema owner, apply these grants, replacing `harness_runtime` with your
+existing runtime role (role/password creation belongs to your operator):
+
+```sql
+GRANT USAGE ON SCHEMA prosaic_harness TO harness_runtime;
+GRANT SELECT ON prosaic_harness.schema_version, prosaic_harness.leases,
+    prosaic_harness.runs, prosaic_harness.receipts TO harness_runtime;
+GRANT INSERT, UPDATE ON prosaic_harness.leases, prosaic_harness.runs TO harness_runtime;
+GRANT INSERT ON prosaic_harness.receipts TO harness_runtime;
+```
+
+Do **not** grant runtime schema CREATE, schema-version UPDATE, receipt UPDATE,
+or table DELETE. Check using the actual application's credentials:
+
+```sh
+prosaic-harness-postgres doctor --dsn-env HARNESS_RUNTIME_DSN
+```
+
+Default doctor reads only catalogs/settings/privileges: no rows, model calls or
+migrations. For an explicit write-and-cleanup test, use migration or dedicated
+diagnostic credentials with DELETE on the three data tables:
+
+```sh
+prosaic-harness-postgres doctor --dsn-env HARNESS_MIGRATION_DSN --check-write
+```
+
+This exercises guarded create/CAS/receipt/read/release operations in a fresh
+`prosaic-diagnostic-<uuid>` namespace. Reserve this prefix; applications must not
+use it. Cleanup deletes only its exact receipt/run/lease identities, in that
+order, after release. It never deletes production lease generations. Cleanup,
+interruption and commit-ack failures fail the command. Its error includes the
+diagnostic namespace for reconciliation: inspect exact rows before manual
+cleanup/retry, never blindly delete by prefix.
+
+## 3. Embed the single Harness engine
+
+```python
+from prosaic_harness import Harness, Workflow
+from prosaic_harness_postgres import PostgresRunStore
+
+workflow = Workflow.load('/srv/workflows/approval/pause.yml')
+# Construct after fork; keep one long-lived store per bounded trusted namespace.
+with PostgresRunStore.from_env('HARNESS_RUNTIME_DSN', namespace='my-app') as store:
+    store.check_ready()  # startup gate; no implicit init
+    h = Harness(workflow, store=store, run_id='0123456789abcdef0123456789abcdef')
+    h.run({'task': 'example'})  # use a NEW UUID for each new run
+    snapshot = h.status()     # read-only .state and opaque .revision
+    h.resume(choice='approve', expected_revision=snapshot.revision)
+```
+
+Worker B reconstructs the same Workflow/namespace/run ID and calls status/resume,
+not run. Human choices, structured responses and explicit interrupted-call retry
+require the current revision. A stale approval cannot approve a later same-named
+pause. Never silently fetch a fresh revision to bypass rejection: show current
+state and require a new decision. See the [runnable example](../../examples/postgres/README.md).
+
+`load_run` returns an independent RunSnapshot; create/save return revisions.
+Receipt absence is None, corruption is an error. Writes require this store/run's
+active lease. Tokens are run/namespace scoped, not lease owner/fencing tokens.
+Receipts are immutable; identical repeats need live ownership. Sealed finite
+JSON is limited to 8 MiB per document including UTF-8 bytes. Public APIs return
+no driver objects. The legacy `Harness(workflow, run_dir)` and filesystem CLI
+remain supported; the adapter adds no new workflow interpreter.
+
+## Deployment and safety
+
+- All reads/writes use the authoritative writable primary, never replicas/caches.
+  Tables must be permanent/logged; fsync/full_page_writes must be on, effective
+  synchronous_commit on or remote_apply. Existing remote_apply is preserved.
+  Primary/WAL guards run on every operation/reconnection.
+- Mount immutable workflow, prose, schema, evidence and reviewed tool assets at
+  identical absolute paths across workers. Keep executables, validator versions
+  and Runtime configuration identical. Drift fails before dispatch. PostgreSQL
+  moves run state, not assets or private tool scratch.
+- Synchronize worker/database clocks and avoid abrupt clock steps. Expiry uses
+  fresh clock_timestamp() after the lease-row lock; local ownership uses a
+  conservative monotonic deadline starting before the request, not at reply.
+- Default lease/renewal 60s/15s. Duration >=15s; renewal >=1s and <=duration/3;
+  renewal + operation budget + 1s must be less than duration. Default total I/O
+  budget 10s, pool/connect 5s, row lock 1s, statement 5s. Deadlines cover pool,
+  network and COMMIT waits. Stalled connections are quarantined; unresolved
+  operations retain capacity and prevent unsafe close.
+- Default lazy pool bounds per store/process: 8 ordinary + 4 renewal connections,
+  maximum 4 active leases. Renewal has reserved capacity. Multiply capacity
+  across replicas/namespaces; do not create a pool per HTTP request. Applications
+  still own scheduling, model concurrency and quotas.
+- Construct after fork; do not share across processes. Drain executions before
+  close/context exit. Threads can share a store; use a separate Harness per run.
+- Namespaces are trusted isolation keys, **not authorization or DB row-level
+  security**. A shared DB role can directly access other namespaces. Derive
+  namespace/run ownership from authenticated server state, never browser/model
+  input. Applications own tenant authorization and admission policies.
+- Raw status state contains private inputs, evidence, responses and history.
+  Products must expose a filtered authorized view, not the operator demo JSON.
+- Prose frontmatter requests tools; Runtime/workflow permissions authorize them.
+  Neither grants database permissions. No HTTP service, queue, PostgreSQL server
+  or historical file importer is included.
+- Fencing stops stale DB mutations, not external tool/model effects. Use
+  application-owned idempotency keys. Pending calls without receipts require
+  explicit fresh retry consent, not automatic redispatch. Unknown token usage
+  may still prevent a token-budgeted retry; abandoned calls retain their budget.
+- Acknowledged WAL durability is not zero-loss asynchronous failover. A lagging
+  promoted primary or older backup can lose receipts/generations. Require
+  operator replication guarantees where needed; do not promise exactly-once
+  external effects from this adapter.
+
+## Errors and recovery
+
+| Code | Action |
+| --- | --- |
+| store_uninitialized | Operator runs init, applies grants, then doctor. |
+| store_incompatible | Check role/version/primary/WAL settings and intact logged schema; no automatic repair. |
+| store_unavailable | Restore primary connectivity and reload status. outcome_unknown means a commit may have succeeded; never blindly repeat writes/effects. |
+| run_busy | Another worker owns execution; back off without launching another call. |
+| lease_lost | Stop dispatch/admission/writes, discard late output; acquire fresh execution and reconcile durable state. |
+| revision_conflict | Show the user's current pause and require a new explicit decision. |
+| store_busy | Drain active/queued/renewal work; construct a new store after fork. |
+| receipt_conflict / store_corrupt | Stop; investigate evidence without overwriting it. |
+
+## Development verification
+
+Core tests need no database/driver. Adapter tests use disposable databases and
+synthetic inference, never TokenProxy. Required mode fails rather than skips
+missing infrastructure. Select Podman/Docker explicitly for crash/standby gates:
+
+```sh
+python -m pip install -e '.[test]' -e './adapters/postgres[test]'
+python adapters/postgres/tests/support/local_cluster.py start --engine podman --version 16
+# Retain the printed ownership.json path as FIXTURE_MANIFEST.
+python adapters/postgres/tests/support/local_cluster.py test --manifest "$FIXTURE_MANIFEST"
+python adapters/postgres/tests/support/local_cluster.py stop --manifest "$FIXTURE_MANIFEST"
+```
+
+Repeat with version 18. Stop validates exact IDs/names/ownership labels before
+removing only owned containers/volumes/network. Generated fixture networks and
+loopback ports use synthetic trust authentication, never production defaults.
+# Lease registration candidate
+
+The adapter transfers acquisition capacity into registered
+session capacity atomically under the existing store mutex. A registering lease
+can no longer be counted twice while maintenance is scheduled. The database
+storage/fencing contract is unchanged. A unit scheduling-boundary regression and
+an owned PostgreSQL 16/18 race regression cover the transition.

b3_prosaic_harness_postgres-0.2.1.dist-info/RECORD

@@ -0,0 +1,13 @@
+b3_prosaic_harness_postgres-0.2.1.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+b3_prosaic_harness_postgres-0.2.1.dist-info/licenses/NOTICE,sha256=ElUIgB9aqeWDj68Un0FliVam9tItd9kKPum77UrCD8s,258
+prosaic_harness_postgres/__init__.py,sha256=8v_irnkIUIB7ulxCVK5ilZ-BsxLTLzcXVe6Am5YznTM,146
+prosaic_harness_postgres/cli.py,sha256=ZGjAKIRVwDEwrTKe_XZjAb4FCbge4XaoSMU91V1yB2A,4103
+prosaic_harness_postgres/leases.py,sha256=Fv_yMKfpEoMBK_eR2otkpxoPN4a4JOdF_zzgTLF035E,2402
+prosaic_harness_postgres/schema.py,sha256=bgA09ds5R2g6axOAhIvEmVyl4ieqDEBsRAk4l6Wt6r0,7686
+prosaic_harness_postgres/store.py,sha256=nDXE8mD8RgSJzAePIHjFJg59K1xHayeLmSjasGt3Cwk,13496
+prosaic_harness_postgres/transport.py,sha256=OGw4LNMq4QKys2QglzSzQf_j0zvEXlJ8PpsH_zesmq0,14178
+b3_prosaic_harness_postgres-0.2.1.dist-info/METADATA,sha256=EZYesowXCIRwAONDv4baAAKezeJDAvO358R4QoWok9A,11879
+b3_prosaic_harness_postgres-0.2.1.dist-info/WHEEL,sha256=YVMoNqKzERt-wjUZwJ33xBGAwnFl-4cqbYkTtWa4itE,91
+b3_prosaic_harness_postgres-0.2.1.dist-info/entry_points.txt,sha256=2j44SxhtLoMVgQnq-wmY5SVnf-LjUcm4-s2yGJmcAWY,79
+b3_prosaic_harness_postgres-0.2.1.dist-info/top_level.txt,sha256=wdHvfuHjA1dMXVsvCq0tjLD76i9noP5PYs1PKHpxcWs,25
+b3_prosaic_harness_postgres-0.2.1.dist-info/RECORD,,

b3_prosaic_harness_postgres-0.2.1.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (84.0.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

b3_prosaic_harness_postgres-0.2.1.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+prosaic-harness-postgres = prosaic_harness_postgres.cli:main

b3_prosaic_harness_postgres-0.2.1.dist-info/licenses/LICENSE

@@ -0,0 +1,202 @@
+
+                                 Apache License
+                           Version 2.0, January 2004
+                        http://www.apache.org/licenses/
+
+   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+   1. Definitions.
+
+      "License" shall mean the terms and conditions for use, reproduction,
+      and distribution as defined by Sections 1 through 9 of this document.
+
+      "Licensor" shall mean the copyright owner or entity authorized by
+      the copyright owner that is granting the License.
+
+      "Legal Entity" shall mean the union of the acting entity and all
+      other entities that control, are controlled by, or are under common
+      control with that entity. For the purposes of this definition,
+      "control" means (i) the power, direct or indirect, to cause the
+      direction or management of such entity, whether by contract or
+      otherwise, or (ii) ownership of fifty percent (50%) or more of the
+      outstanding shares, or (iii) beneficial ownership of such entity.
+
+      "You" (or "Your") shall mean an individual or Legal Entity
+      exercising permissions granted by this License.
+
+      "Source" form shall mean the preferred form for making modifications,
+      including but not limited to software source code, documentation
+      source, and configuration files.
+
+      "Object" form shall mean any form resulting from mechanical
+      transformation or translation of a Source form, including but
+      not limited to compiled object code, generated documentation,
+      and conversions to other media types.
+
+      "Work" shall mean the work of authorship, whether in Source or
+      Object form, made available under the License, as indicated by a
+      copyright notice that is included in or attached to the work
+      (an example is provided in the Appendix below).
+
+      "Derivative Works" shall mean any work, whether in Source or Object
+      form, that is based on (or derived from) the Work and for which the
+      editorial revisions, annotations, elaborations, or other modifications
+      represent, as a whole, an original work of authorship. For the purposes
+      of this License, Derivative Works shall not include works that remain
+      separable from, or merely link (or bind by name) to the interfaces of,
+      the Work and Derivative Works thereof.
+
+      "Contribution" shall mean any work of authorship, including
+      the original version of the Work and any modifications or additions
+      to that Work or Derivative Works thereof, that is intentionally
+      submitted to Licensor for inclusion in the Work by the copyright owner
+      or by an individual or Legal Entity authorized to submit on behalf of
+      the copyright owner. For the purposes of this definition, "submitted"
+      means any form of electronic, verbal, or written communication sent
+      to the Licensor or its representatives, including but not limited to
+      communication on electronic mailing lists, source code control systems,
+      and issue tracking systems that are managed by, or on behalf of, the
+      Licensor for the purpose of discussing and improving the Work, but
+      excluding communication that is conspicuously marked or otherwise
+      designated in writing by the copyright owner as "Not a Contribution."
+
+      "Contributor" shall mean Licensor and any individual or Legal Entity
+      on behalf of whom a Contribution has been received by Licensor and
+      subsequently incorporated within the Work.
+
+   2. Grant of Copyright License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      copyright license to reproduce, prepare Derivative Works of,
+      publicly display, publicly perform, sublicense, and distribute the
+      Work and such Derivative Works in Source or Object form.
+
+   3. Grant of Patent License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      (except as stated in this section) patent license to make, have made,
+      use, offer to sell, sell, import, and otherwise transfer the Work,
+      where such license applies only to those patent claims licensable
+      by such Contributor that are necessarily infringed by their
+      Contribution(s) alone or by combination of their Contribution(s)
+      with the Work to which such Contribution(s) was submitted. If You
+      institute patent litigation against any entity (including a
+      cross-claim or counterclaim in a lawsuit) alleging that the Work
+      or a Contribution incorporated within the Work constitutes direct
+      or contributory patent infringement, then any patent licenses
+      granted to You under this License for that Work shall terminate
+      as of the date such litigation is filed.
+
+   4. Redistribution. You may reproduce and distribute copies of the
+      Work or Derivative Works thereof in any medium, with or without
+      modifications, and in Source or Object form, provided that You
+      meet the following conditions:
+
+      (a) You must give any other recipients of the Work or
+          Derivative Works a copy of this License; and
+
+      (b) You must cause any modified files to carry prominent notices
+          stating that You changed the files; and
+
+      (c) You must retain, in the Source form of any Derivative Works
+          that You distribute, all copyright, patent, trademark, and
+          attribution notices from the Source form of the Work,
+          excluding those notices that do not pertain to any part of
+          the Derivative Works; and
+
+      (d) If the Work includes a "NOTICE" text file as part of its
+          distribution, then any Derivative Works that You distribute must
+          include a readable copy of the attribution notices contained
+          within such NOTICE file, excluding those notices that do not
+          pertain to any part of the Derivative Works, in at least one
+          of the following places: within a NOTICE text file distributed
+          as part of the Derivative Works; within the Source form or
+          documentation, if provided along with the Derivative Works; or,
+          within a display generated by the Derivative Works, if and
+          wherever such third-party notices normally appear. The contents
+          of the NOTICE file are for informational purposes only and
+          do not modify the License. You may add Your own attribution
+          notices within Derivative Works that You distribute, alongside
+          or as an addendum to the NOTICE text from the Work, provided
+          that such additional attribution notices cannot be construed
+          as modifying the License.
+
+      You may add Your own copyright statement to Your modifications and
+      may provide additional or different license terms and conditions
+      for use, reproduction, or distribution of Your modifications, or
+      for any such Derivative Works as a whole, provided Your use,
+      reproduction, and distribution of the Work otherwise complies with
+      the conditions stated in this License.
+
+   5. Submission of Contributions. Unless You explicitly state otherwise,
+      any Contribution intentionally submitted for inclusion in the Work
+      by You to the Licensor shall be under the terms and conditions of
+      this License, without any additional terms or conditions.
+      Notwithstanding the above, nothing herein shall supersede or modify
+      the terms of any separate license agreement you may have executed
+      with Licensor regarding such Contributions.
+
+   6. Trademarks. This License does not grant permission to use the trade
+      names, trademarks, service marks, or product names of the Licensor,
+      except as required for reasonable and customary use in describing the
+      origin of the Work and reproducing the content of the NOTICE file.
+
+   7. Disclaimer of Warranty. Unless required by applicable law or
+      agreed to in writing, Licensor provides the Work (and each
+      Contributor provides its Contributions) on an "AS IS" BASIS,
+      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+      implied, including, without limitation, any warranties or conditions
+      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
+      PARTICULAR PURPOSE. You are solely responsible for determining the
+      appropriateness of using or redistributing the Work and assume any
+      risks associated with Your exercise of permissions under this License.
+
+   8. Limitation of Liability. In no event and under no legal theory,
+      whether in tort (including negligence), contract, or otherwise,
+      unless required by applicable law (such as deliberate and grossly
+      negligent acts) or agreed to in writing, shall any Contributor be
+      liable to You for damages, including any direct, indirect, special,
+      incidental, or consequential damages of any character arising as a
+      result of this License or out of the use or inability to use the
+      Work (including but not limited to damages for loss of goodwill,
+      work stoppage, computer failure or malfunction, or any and all
+      other commercial damages or losses), even if such Contributor
+      has been advised of the possibility of such damages.
+
+   9. Accepting Warranty or Additional Liability. While redistributing
+      the Work or Derivative Works thereof, You may choose to offer,
+      and charge a fee for, acceptance of support, warranty, indemnity,
+      or other liability obligations and/or rights consistent with this
+      License. However, in accepting such obligations, You may act only
+      on Your own behalf and on Your sole responsibility, not on behalf
+      of any other Contributor, and only if You agree to indemnify,
+      defend, and hold each Contributor harmless for any liability
+      incurred by, or claims asserted against, such Contributor by reason
+      of your accepting any such warranty or additional liability.
+
+   END OF TERMS AND CONDITIONS
+
+   APPENDIX: How to apply the Apache License to your work.
+
+      To apply the Apache License to your work, attach the following
+      boilerplate notice, with the fields enclosed by brackets "[]"
+      replaced with your own identifying information. (Don't include
+      the brackets!)  The text should be enclosed in the appropriate
+      comment syntax for the file format. We also recommend that a
+      file or class name and description of purpose be included on the
+      same "printed page" as the copyright notice for easier
+      identification within third-party archives.
+
+   Copyright [yyyy] [name of copyright owner]
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.

b3_prosaic_harness_postgres-0.2.1.dist-info/licenses/NOTICE

@@ -0,0 +1,5 @@
+Prosaic Harness
+
+This project is distributed under the Apache License, Version 2.0; see LICENSE.
+Third-party dependencies retain their own licenses and attribution notices.
+Previously published versions and tags retain the license notices shipped with them.

b3_prosaic_harness_postgres-0.2.1.dist-info/top_level.txt

@@ -0,0 +1 @@
+prosaic_harness_postgres

prosaic_harness_postgres/__init__.py

@@ -0,0 +1,6 @@
+"""Separate PostgreSQL adapter for Prosaic Harness."""
+__version__ = '0.2.1'
+
+from .store import PostgresRunStore
+
+__all__ = ['PostgresRunStore']

prosaic_harness_postgres/cli.py

@@ -0,0 +1,81 @@
+"""Explicit operator setup; no provisioning, inference or secret arguments."""
+import argparse
+import json
+import sys
+import uuid
+
+from prosaic_harness import StoreError, StoreIncompatible, StoreBusy
+from prosaic_harness.contracts import seal
+from .store import PostgresRunStore
+
+
+def check_write(store):
+    async def permissions(conn):
+        for table in ('receipts', 'runs', 'leases'):
+            allowed = await (await conn.execute('SELECT pg_catalog.has_table_privilege(%s,%s)',
+                ('prosaic_harness.' + table, 'DELETE'))).fetchone()
+            if not allowed[0]:
+                raise StoreIncompatible('write diagnostic requires separate DELETE-capable credentials')
+    store._transport.call(permissions)
+    run_id, invocation_id = uuid.uuid4().hex, uuid.uuid4().hex
+    state = seal({'run_id': run_id, 'inputs': {'diagnostic': True}})
+    receipt = seal({'run_id': run_id, 'id': invocation_id, 'events': []})
+    async def cleanup(conn):
+        row = await (await conn.execute('SELECT owner FROM prosaic_harness.leases '
+            'WHERE namespace=%s AND run_id=%s FOR UPDATE', (store.namespace, run_id))).fetchone()
+        if row and row[0] is not None:
+            raise StoreBusy('diagnostic cleanup incomplete; reconcile the reserved diagnostic namespace')
+        await conn.execute('DELETE FROM prosaic_harness.receipts WHERE namespace=%s AND run_id=%s AND invocation_id=%s',
+                           (store.namespace, run_id, invocation_id))
+        await conn.execute('DELETE FROM prosaic_harness.runs WHERE namespace=%s AND run_id=%s', (store.namespace, run_id))
+        await conn.execute('DELETE FROM prosaic_harness.leases WHERE namespace=%s AND run_id=%s', (store.namespace, run_id))
+    try:
+        with store.lease(run_id) as lease:
+            first = store.create_run(run_id, state, lease)
+            second = store.save_run(run_id, state, first, lease)
+            store.save_receipt(run_id, invocation_id, receipt, lease)
+            if store.load_run(run_id).revision != second or store.load_receipt(run_id, invocation_id) != receipt:
+                raise StoreIncompatible('diagnostic persistence verification failed')
+    finally:
+        # Cleanup must acknowledge success before the command can pass.
+        store._transport.call(cleanup)
+
+
+def main(argv=None):
+    parser = argparse.ArgumentParser(description='Initialize/check an existing Harness database')
+    commands = parser.add_subparsers(dest='command', required=True)
+    for name in ('init', 'doctor'):
+        command = commands.add_parser(name)
+        command.add_argument('--dsn-env', required=True, help='name of the DSN environment variable, never its value')
+        if name == 'doctor':
+            command.add_argument('--check-write', action='store_true', help='explicit disposable write test; DELETE credentials required')
+    args = parser.parse_args(argv)
+    try:
+        namespace = 'prosaic-diagnostic-' + uuid.uuid4().hex
+        with PostgresRunStore.from_env(args.dsn_env, namespace=namespace) as store:
+            if args.command == 'init':
+                store.initialize()
+                result = {'initialized': True, 'schema_version': 1}
+            else:
+                store.check_ready()
+                if args.check_write:
+                    check_write(store)
+                result = {'ready': True, 'write_checked': args.check_write}
+        print(json.dumps(result))
+        return 0
+    except StoreError as exc:
+        result = {'error': exc.code, 'message': str(exc)}
+        if args.command == 'doctor' and args.check_write:
+            result['diagnostic_namespace'] = namespace
+        print(json.dumps(result), file=sys.stderr)
+        return 1
+    except KeyboardInterrupt:
+        print('{"error":"interrupted","message":"operation interrupted; reconcile before retrying"}', file=sys.stderr)
+        return 130
+    except Exception:
+        print('{"error":"invalid_configuration","message":"check database environment and configuration"}', file=sys.stderr)
+        return 1
+
+
+if __name__ == '__main__':
+    raise SystemExit(main())

prosaic_harness_postgres/leases.py

@@ -0,0 +1,64 @@
+"""Private renewable execution sessions; database generations fence writes."""
+import asyncio
+import os
+import threading
+import time
+import uuid
+
+from prosaic_harness import LeaseLost, StoreError, StoreBusy
+
+
+class PostgresLease:
+    def __init__(self, store, run_id, started):
+        self.store, self.run_id = store, run_id
+        self.pid = os.getpid()
+        self.owner = uuid.uuid4().hex
+        self.generation = None
+        self._deadline = started + store.lease_s
+        self._lost = False
+        self._active = True
+        self._stopping = threading.Event()
+        self._stopped = threading.Event()
+        self._task = None
+        self._renewal = None
+
+    def check_valid(self):
+        if (self.pid != os.getpid() or not self._active or self._lost
+                or time.monotonic() >= self._deadline):
+            self._lost = True
+            raise LeaseLost('database execution session is no longer owned')
+
+    async def maintain(self):
+        self._task = asyncio.current_task()
+        try:
+            while not self._stopping.is_set():
+                await asyncio.sleep(self.store.renew_s)
+                if self._stopping.is_set():
+                    break
+                started = time.monotonic()
+                await self.store._transport.execute_async(
+                    lambda conn: self.store._renew(conn, self), renewal=True)
+                self.check_valid()
+                self._deadline = started + self.store.lease_s
+        except (StoreError, asyncio.CancelledError):
+            self._lost = True
+        except Exception:
+            self._lost = True
+        finally:
+            self._stopped.set()
+
+    def stop(self):
+        self._stopping.set()
+        def cancel():
+            if self._task is not None:
+                self._task.cancel()
+            elif self._renewal is not None:
+                # Submission may have raised before scheduling, or while the
+                # runner was still queued. Only the I/O loop can close it
+                # without racing its first execution.
+                self._renewal.close()
+                self._stopped.set()
+        self.store._transport._loop.call_soon_threadsafe(cancel)
+        if not self._stopped.wait(self.store._transport.operation_timeout_s + 2):
+            self._lost = True
+            raise StoreBusy('renewal task has not drained; retain store until cleanup completes')