mirror of
https://github.com/hatchet-dev/hatchet.git
synced 2026-02-11 10:38:37 -06:00
cfa4b5c8f4
* fix: python sdk graceful shutdown and retry errors * chore: address changes from review
38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
from .client import new_client
|
|
from typing import List
|
|
from .workflow import WorkflowMeta
|
|
from .worker import Worker
|
|
from .logger import logger
|
|
|
|
class Hatchet:
|
|
def __init__(self, debug=False):
|
|
# initialize a client
|
|
self.client = new_client()
|
|
|
|
if not debug:
|
|
logger.disable("hatchet_sdk")
|
|
|
|
def workflow(self, name : str='', on_events : list=[], on_crons : list=[]):
|
|
def inner(cls):
|
|
cls.on_events = on_events
|
|
cls.on_crons = on_crons
|
|
cls.name = name or str(cls.__name__)
|
|
cls.client = self.client
|
|
|
|
# Define a new class with the same name and bases as the original, but with WorkflowMeta as its metaclass
|
|
return WorkflowMeta(cls.__name__, cls.__bases__, dict(cls.__dict__))
|
|
|
|
return inner
|
|
|
|
def step(self, name : str='', parents : List[str] = []):
|
|
def inner(func):
|
|
func._step_name = name or func.__name__
|
|
func._step_parents = parents
|
|
|
|
return func
|
|
|
|
return inner
|
|
|
|
def worker(self, name: str, max_threads: int = 200):
|
|
return Worker(name, max_threads)
|