Compare commits
6 commits
Author | SHA1 | Date | |
---|---|---|---|
5a7d4e04fb | |||
07aeb8d9bf | |||
2af9c8dd87 | |||
e29bf2baf6 | |||
|
d3e58dbe3b | ||
|
100a3c7798 |
8 changed files with 27 additions and 109 deletions
16
README.md
16
README.md
|
@ -1,16 +1,15 @@
|
|||
# Timeloop
|
||||
Timeloop is a service that can be used to run periodic tasks after a certain interval.
|
||||
|
||||

|
||||
|
||||
Each job runs on a separate thread and when the service is shut down, it waits till all tasks currently being executed are completed.
|
||||
|
||||
Inspired by this blog [`here`](https://www.g-loaded.eu/2016/11/24/how-to-terminate-running-python-threads-using-signals/)
|
||||
|
||||
## Fork
|
||||
This fork aims to provide some improvements to the original library. Mainly to be able to set a start time or start offset for tasks.
|
||||
|
||||
## Installation
|
||||
```sh
|
||||
pip install timeloop
|
||||
```
|
||||
Has to be installed manually atm. Since it is just a fork of the original and I still have to set that up and want to respect the work of Sankalp Jonna. Feel free to help tho!
|
||||
|
||||
## Writing jobs
|
||||
```python
|
||||
|
@ -30,9 +29,10 @@ def sample_job_every_5s():
|
|||
print "5s job current time : {}".format(time.ctime())
|
||||
|
||||
|
||||
@tl.job(interval=timedelta(seconds=10))
|
||||
def sample_job_every_10s():
|
||||
print "10s job current time : {}".format(time.ctime())
|
||||
# Added support for initial offset!
|
||||
@tl.job(interval=timedelta(seconds=10), offset=timedelta(hours=1))
|
||||
def sample_job_after_an_hour_every_10s():
|
||||
print "after an hour 10s job current time : {}".format(time.ctime())
|
||||
```
|
||||
|
||||
## Start time loop in separate thread
|
||||
|
|
|
@ -1 +0,0 @@
|
|||
from timeloop.app import Timeloop
|
|
@ -1,65 +0,0 @@
|
|||
import logging
|
||||
import sys
|
||||
import signal
|
||||
import time
|
||||
|
||||
from timeloop.exceptions import ServiceExit
|
||||
from timeloop.job import Job
|
||||
from timeloop.helpers import service_shutdown
|
||||
|
||||
|
||||
class Timeloop():
|
||||
def __init__(self):
|
||||
self.jobs = []
|
||||
logger = logging.getLogger('timeloop')
|
||||
ch = logging.StreamHandler(sys.stdout)
|
||||
ch.setLevel(logging.INFO)
|
||||
formatter = logging.Formatter('[%(asctime)s] [%(name)s] [%(levelname)s] %(message)s')
|
||||
ch.setFormatter(formatter)
|
||||
logger.addHandler(ch)
|
||||
logger.setLevel(logging.INFO)
|
||||
self.logger = logger
|
||||
|
||||
def _add_job(self, func, interval, *args, **kwargs):
|
||||
j = Job(interval, func, *args, **kwargs)
|
||||
self.jobs.append(j)
|
||||
|
||||
def _block_main_thread(self):
|
||||
signal.signal(signal.SIGTERM, service_shutdown)
|
||||
signal.signal(signal.SIGINT, service_shutdown)
|
||||
|
||||
while True:
|
||||
try:
|
||||
time.sleep(1)
|
||||
except ServiceExit:
|
||||
self.stop()
|
||||
break
|
||||
|
||||
def _start_jobs(self, block):
|
||||
for j in self.jobs:
|
||||
j.daemon = not block
|
||||
j.start()
|
||||
self.logger.info("Registered job {}".format(j.execute))
|
||||
|
||||
def _stop_jobs(self):
|
||||
for j in self.jobs:
|
||||
self.logger.info("Stopping job {}".format(j.execute))
|
||||
j.stop()
|
||||
|
||||
def job(self, interval):
|
||||
def decorator(f):
|
||||
self._add_job(f, interval)
|
||||
return f
|
||||
return decorator
|
||||
|
||||
def stop(self):
|
||||
self._stop_jobs()
|
||||
self.logger.info("Timeloop exited.")
|
||||
|
||||
def start(self, block=False):
|
||||
self.logger.info("Starting Timeloop..")
|
||||
self._start_jobs(block=block)
|
||||
|
||||
self.logger.info("Timeloop now started. Jobs will run based on the interval set")
|
||||
if block:
|
||||
self._block_main_thread()
|
|
@ -1,6 +0,0 @@
|
|||
class ServiceExit(Exception):
|
||||
"""
|
||||
Custom exception which is used to trigger the clean exit
|
||||
of all running threads and the main program.
|
||||
"""
|
||||
pass
|
|
@ -1,4 +0,0 @@
|
|||
from timeloop.exceptions import ServiceExit
|
||||
|
||||
def service_shutdown(signum, frame):
|
||||
raise ServiceExit
|
|
@ -1,19 +0,0 @@
|
|||
from threading import Thread, Event
|
||||
from datetime import timedelta
|
||||
|
||||
class Job(Thread):
|
||||
def __init__(self, interval, execute, *args, **kwargs):
|
||||
Thread.__init__(self)
|
||||
self.stopped = Event()
|
||||
self.interval = interval
|
||||
self.execute = execute
|
||||
self.args = args
|
||||
self.kwargs = kwargs
|
||||
|
||||
def stop(self):
|
||||
self.stopped.set()
|
||||
self.join()
|
||||
|
||||
def run(self):
|
||||
while not self.stopped.wait(self.interval.total_seconds()):
|
||||
self.execute(*self.args, **self.kwargs)
|
|
@ -1,3 +1,4 @@
|
|||
from datetime import datetime, timedelta
|
||||
import logging
|
||||
import sys
|
||||
import signal
|
||||
|
@ -20,8 +21,8 @@ class Timeloop():
|
|||
logger.setLevel(logging.INFO)
|
||||
self.logger = logger
|
||||
|
||||
def _add_job(self, func, interval, *args, **kwargs):
|
||||
j = Job(interval, func, *args, **kwargs)
|
||||
def _add_job(self, func, interval: timedelta, offset: timedelta=None, *args, **kwargs):
|
||||
j = Job(interval, func, offset=offset, *args, **kwargs)
|
||||
self.jobs.append(j)
|
||||
|
||||
def _block_main_thread(self):
|
||||
|
@ -46,9 +47,15 @@ class Timeloop():
|
|||
self.logger.info("Stopping job {}".format(j.execute))
|
||||
j.stop()
|
||||
|
||||
def job(self, interval):
|
||||
def job(self, interval: timedelta, offset: timedelta=None):
|
||||
"""Decorator to define a timeloop for the decorated function.
|
||||
|
||||
Args:
|
||||
interval (timedelta): How long to wait after every execution until the next one.
|
||||
offset (timedelta, optional): Positive offset until the first execution of the function. If None, will wait with first execution until the first interval passed. If timedelta with length 0 (or smaller) will execute immediately. Defaults to None.
|
||||
"""
|
||||
def decorator(f):
|
||||
self._add_job(f, interval)
|
||||
self._add_job(f, interval, offset=offset)
|
||||
return f
|
||||
return decorator
|
||||
|
||||
|
|
|
@ -1,12 +1,14 @@
|
|||
from threading import Thread, Event
|
||||
from datetime import timedelta
|
||||
from time import sleep
|
||||
|
||||
class Job(Thread):
|
||||
def __init__(self, interval, execute, *args, **kwargs):
|
||||
def __init__(self, interval: timedelta, execute, offset: timedelta=None, *args, **kwargs):
|
||||
Thread.__init__(self)
|
||||
self.stopped = Event()
|
||||
self.interval = interval
|
||||
self.interval: timedelta = interval
|
||||
self.execute = execute
|
||||
self.offset: timedelta = offset
|
||||
self.args = args
|
||||
self.kwargs = kwargs
|
||||
|
||||
|
@ -15,5 +17,9 @@ class Job(Thread):
|
|||
self.join()
|
||||
|
||||
def run(self):
|
||||
if self.offset:
|
||||
sleep(self.offset.total_seconds())
|
||||
self.execute(*self.args, **self.kwargs)
|
||||
|
||||
while not self.stopped.wait(self.interval.total_seconds()):
|
||||
self.execute(*self.args, **self.kwargs)
|
||||
|
|
Loading…
Reference in a new issue