Fix to pickle error and Nonetype comparison

This commit is contained in:
bobloy 2021-01-21 10:44:25 -05:00
parent 837cff7a26
commit 3f997fa804
2 changed files with 14 additions and 1 deletions

View File

@ -1,5 +1,15 @@
import sys
from .fifo import FIFO from .fifo import FIFO
# Applying fix from: https://github.com/Azure/azure-functions-python-worker/issues/640
# [Fix] Create a wrapper for importing imgres
from .date_trigger import *
from . import CustomDateTrigger
# [Fix] Register imgres into system modules
sys.modules["CustomDateTrigger"] = CustomDateTrigger
async def setup(bot): async def setup(bot):
cog = FIFO(bot) cog = FIFO(bot)

View File

@ -4,4 +4,7 @@ from apscheduler.triggers.date import DateTrigger
class CustomDateTrigger(DateTrigger): class CustomDateTrigger(DateTrigger):
def get_next_fire_time(self, previous_fire_time, now): def get_next_fire_time(self, previous_fire_time, now):
next_run = super().get_next_fire_time(previous_fire_time, now) next_run = super().get_next_fire_time(previous_fire_time, now)
return next_run if next_run >= now else None return next_run if next_run is not None and next_run >= now else None
def __getstate__(self):
return {"version": 1, "run_date": self.run_date}