2
2
from dataclasses import dataclass
3
3
from dataclasses_json import dataclass_json , LetterCase , Undefined
4
4
from datetime import datetime , timedelta
5
+ from enum import Enum , IntEnum
5
6
6
7
T = TypeVar ('T' )
7
8
Table = TypeVar ('Table' )
@@ -469,7 +470,7 @@ class WorkerStats:
469
470
retries : int = 0
470
471
failed : int = 0
471
472
running_job : Optional [int ] = None
472
- running_time : Optional [datetime . timedelta ] = None
473
+ running_time : Optional [timedelta ] = None
473
474
474
475
@dataclass_json (letter_case = LetterCase .CAMEL , undefined = Undefined .EXCLUDE )
475
476
@dataclass
@@ -482,8 +483,8 @@ class BackgroundJobBase:
482
483
batch_id : Optional [str ] = None
483
484
callback : Optional [str ] = None
484
485
depends_on : Optional [int ] = None
485
- run_after : Optional [datetime . datetime ] = None
486
- created_date : datetime . datetime = datetime . datetime (1 , 1 , 1 )
486
+ run_after : Optional [datetime ] = None
487
+ created_date : datetime = datetime (1 , 1 , 1 )
487
488
created_by : Optional [str ] = None
488
489
request_id : Optional [str ] = None
489
490
request_type : Optional [str ] = None
@@ -494,17 +495,17 @@ class BackgroundJobBase:
494
495
response : Optional [str ] = None
495
496
response_body : Optional [str ] = None
496
497
state : Optional [BackgroundJobState ] = None
497
- started_date : Optional [datetime . datetime ] = None
498
- completed_date : Optional [datetime . datetime ] = None
499
- notified_date : Optional [datetime . datetime ] = None
498
+ started_date : Optional [datetime ] = None
499
+ completed_date : Optional [datetime ] = None
500
+ notified_date : Optional [datetime ] = None
500
501
retry_limit : Optional [int ] = None
501
502
attempts : int = 0
502
503
duration_ms : int = 0
503
504
timeout_secs : Optional [int ] = None
504
505
progress : Optional [float ] = None
505
506
status : Optional [str ] = None
506
507
logs : Optional [str ] = None
507
- last_activity_date : Optional [datetime . datetime ] = None
508
+ last_activity_date : Optional [datetime ] = None
508
509
reply_to : Optional [str ] = None
509
510
error_code : Optional [str ] = None
510
511
error : Optional [ResponseStatus ] = None
@@ -525,16 +526,16 @@ class JobSummary:
525
526
worker : Optional [str ] = None
526
527
tag : Optional [str ] = None
527
528
batch_id : Optional [str ] = None
528
- created_date : datetime . datetime = datetime . datetime (1 , 1 , 1 )
529
+ created_date : datetime = datetime (1 , 1 , 1 )
529
530
created_by : Optional [str ] = None
530
531
request_type : Optional [str ] = None
531
532
command : Optional [str ] = None
532
533
request : Optional [str ] = None
533
534
response : Optional [str ] = None
534
535
user_id : Optional [str ] = None
535
536
callback : Optional [str ] = None
536
- started_date : Optional [datetime . datetime ] = None
537
- completed_date : Optional [datetime . datetime ] = None
537
+ started_date : Optional [datetime ] = None
538
+ completed_date : Optional [datetime ] = None
538
539
state : Optional [BackgroundJobState ] = None
539
540
duration_ms : int = 0
540
541
attempts : int = 0
@@ -547,7 +548,7 @@ class BackgroundJobOptions:
547
548
ref_id : Optional [str ] = None
548
549
parent_id : Optional [int ] = None
549
550
worker : Optional [str ] = None
550
- run_after : Optional [datetime . datetime ] = None
551
+ run_after : Optional [datetime ] = None
551
552
callback : Optional [str ] = None
552
553
depends_on : Optional [int ] = None
553
554
user_id : Optional [str ] = None
@@ -557,7 +558,7 @@ class BackgroundJobOptions:
557
558
batch_id : Optional [str ] = None
558
559
created_by : Optional [str ] = None
559
560
timeout_secs : Optional [int ] = None
560
- timeout : Optional [datetime . timedelta ] = None
561
+ timeout : Optional [timedelta ] = None
561
562
args : Optional [Dict [str , str ]] = None
562
563
run_command : Optional [bool ] = None
563
564
@@ -566,14 +567,14 @@ class BackgroundJobOptions:
566
567
class ScheduledTask :
567
568
id : int = 0
568
569
name : Optional [str ] = None
569
- interval : Optional [datetime . timedelta ] = None
570
+ interval : Optional [timedelta ] = None
570
571
cron_expression : Optional [str ] = None
571
572
request_type : Optional [str ] = None
572
573
command : Optional [str ] = None
573
574
request : Optional [str ] = None
574
575
request_body : Optional [str ] = None
575
576
options : Optional [BackgroundJobOptions ] = None
576
- last_run : Optional [datetime . datetime ] = None
577
+ last_run : Optional [datetime ] = None
577
578
last_job_id : Optional [int ] = None
578
579
579
580
@dataclass_json (letter_case = LetterCase .CAMEL , undefined = Undefined .EXCLUDE )
0 commit comments