US02-03: Execute Jobs with Leases, Locks, and Recovery (#56)

photo_pipeline/__main__.py

@@ -14,6 +14,8 @@ def main(argv: Sequence[str] | None = None) -> int:
     commands = parser.add_subparsers(dest="command", required=True)
     commands.add_parser("serve", help="Run the API server")
     commands.add_parser("migrate", help="Upgrade the database to the latest revision")
+    worker_cmd = commands.add_parser("worker", help="Run a durable-job worker")
+    worker_cmd.add_argument("--id", default="worker-1", help="Worker id (lease owner)")
     args = parser.parse_args(argv)
 
     config = Config.from_env()
@@ -23,6 +25,15 @@ def main(argv: Sequence[str] | None = None) -> int:
         run_migrations(config.database_url)
         return 0
 
+    if args.command == "worker":
+        from photo_pipeline.db import create_db_engine, create_session_factory
+        from photo_pipeline.jobs.worker import Worker
+
+        run_migrations(config.database_url)
+        engine = create_db_engine(config.database_url)
+        Worker(create_session_factory(engine), worker_id=args.id).run_forever()
+        return 0
+
     import uvicorn
 
     from photo_pipeline.api.app import create_app