pyworker fixes and ogimages fixes

pyworker/pyworker/scheduler.py

@@ -126,9 +126,9 @@ class TaskScheduler:
                     self.logger.info(f"Running task '{task_name}'")
                     # Use task instance as a context manager to ensure
                     # a single database connection is used for the entire task
-                    with task_info["instance"] as task_instance:
-                        # Execute the task instance's run method directly
-                        task_instance.run()
+                    with task_info["instance"]:
+                        # Execute the registered task function with its arguments
+                        task_info["func"](*task_info["args"], **task_info["kwargs"])
                     self.logger.info(f"Task '{task_name}' completed")
                 except Exception as e:
                     self.logger.exception(f"Error running task '{task_name}': {e}")

web/src/pages/ogimage.png.ts

@@ -3,21 +3,21 @@ import { urlParamsToObject } from '../lib/urls'
 
 import type { APIRoute } from 'astro'
 
-export const GET: APIRoute = ({ url }) => {
-  const { template, ...props } = urlParamsToObject(url.searchParams)
+export const GET: APIRoute = (context) => {
+  const { template, ...props } = urlParamsToObject(context.url.searchParams)
 
-  if (!template) return ogImageTemplates.default()
+  if (!template) return ogImageTemplates.default({}, context)
 
   if (!(template in ogImageTemplates)) {
     console.error(`Invalid template: "${template}"`)
-    return ogImageTemplates.default()
+    return ogImageTemplates.default({}, context)
   }
 
-  const response = ogImageTemplates[template as keyof typeof ogImageTemplates](props)
+  const response = ogImageTemplates[template as keyof typeof ogImageTemplates](props, context)
   // eslint-disable-next-line @typescript-eslint/no-unnecessary-condition
   if (!response) {
     console.error(`Cannot generate image for template: ${template} and props: ${JSON.stringify(props)}`)
-    return ogImageTemplates.default()
+    return ogImageTemplates.default({}, context)
   }
 
   return response