service will run on a separate thread unless explicitly mentioned by the user to block it in main thread
This commit is contained in:
parent
2ab6714040
commit
c0b4aaf089
4 changed files with 55 additions and 30 deletions
23
README.md
23
README.md
|
@ -10,8 +10,7 @@ Inspired by this blog [`here`](https://www.g-loaded.eu/2016/11/24/how-to-termina
|
||||||
pip install timeloop
|
pip install timeloop
|
||||||
```
|
```
|
||||||
|
|
||||||
## writing a job looks like this
|
## Writing jobs
|
||||||
|
|
||||||
```python
|
```python
|
||||||
import time
|
import time
|
||||||
|
|
||||||
|
@ -32,8 +31,28 @@ def sample_job_every_5s():
|
||||||
@tl.job(interval=timedelta(seconds=10))
|
@tl.job(interval=timedelta(seconds=10))
|
||||||
def sample_job_every_10s():
|
def sample_job_every_10s():
|
||||||
print "10s job current time : {}".format(time.ctime())
|
print "10s job current time : {}".format(time.ctime())
|
||||||
|
```
|
||||||
|
|
||||||
|
## Start time loop in separate thread
|
||||||
|
By default timeloop starts in a separate thread.
|
||||||
|
|
||||||
|
Please do not forget to call ```tl.stop``` before exiting the program, Or else the jobs wont shut down gracefully.
|
||||||
|
|
||||||
|
```python
|
||||||
tl.start()
|
tl.start()
|
||||||
|
|
||||||
|
while True:
|
||||||
|
try:
|
||||||
|
time.sleep(1)
|
||||||
|
except KeyboardInterrupt:
|
||||||
|
tl.stop()
|
||||||
|
break
|
||||||
|
```
|
||||||
|
|
||||||
|
## Start time loop in main thread
|
||||||
|
Doing this will automatically shut down the jobs gracefully when the program is killed, so no need to call ```tl.stop```
|
||||||
|
```python
|
||||||
|
tl.start(block=True)
|
||||||
```
|
```
|
||||||
|
|
||||||
## Author
|
## Author
|
||||||
|
|
2
setup.py
2
setup.py
|
@ -6,7 +6,7 @@ with open("README.md", "r") as fh:
|
||||||
|
|
||||||
setup(
|
setup(
|
||||||
name='timeloop',
|
name='timeloop',
|
||||||
version='1.0',
|
version='1.0.1',
|
||||||
packages=['timeloop'],
|
packages=['timeloop'],
|
||||||
license = 'MIT',
|
license = 'MIT',
|
||||||
description = 'An elegant way to run period tasks.',
|
description = 'An elegant way to run period tasks.',
|
||||||
|
|
|
@ -16,33 +16,46 @@ class Timeloop():
|
||||||
logger.setLevel(logging.INFO)
|
logger.setLevel(logging.INFO)
|
||||||
self.logger = logger
|
self.logger = logger
|
||||||
|
|
||||||
def _add_task(self, func, interval, *args, **kwargs):
|
def _add_job(self, func, interval, *args, **kwargs):
|
||||||
j = Job(interval, func, *args, **kwargs)
|
j = Job(interval, func, *args, **kwargs)
|
||||||
self.jobs.append(j)
|
self.jobs.append(j)
|
||||||
|
|
||||||
def job(self, interval):
|
def _block_main_thread(self):
|
||||||
def decorator(f):
|
|
||||||
self._add_task(f, interval)
|
|
||||||
return f
|
|
||||||
return decorator
|
|
||||||
|
|
||||||
def start(self):
|
|
||||||
try:
|
|
||||||
signal.signal(signal.SIGTERM, service_shutdown)
|
signal.signal(signal.SIGTERM, service_shutdown)
|
||||||
signal.signal(signal.SIGINT, service_shutdown)
|
signal.signal(signal.SIGINT, service_shutdown)
|
||||||
|
|
||||||
self.logger.info("Starting Timeloop..")
|
|
||||||
for j in self.jobs:
|
|
||||||
self.logger.info("Registered task {}".format(j.execute))
|
|
||||||
j.start()
|
|
||||||
self.logger.info("Timeloop now started. Tasks will run based on the interval set")
|
|
||||||
|
|
||||||
# block main thead
|
|
||||||
while True:
|
while True:
|
||||||
|
try:
|
||||||
time.sleep(1)
|
time.sleep(1)
|
||||||
|
|
||||||
except ServiceExit:
|
except ServiceExit:
|
||||||
|
self.stop()
|
||||||
|
break
|
||||||
|
|
||||||
|
def _start_jobs(self, block):
|
||||||
for j in self.jobs:
|
for j in self.jobs:
|
||||||
self.logger.info("Stopping task {}".format(j.execute))
|
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()
|
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.")
|
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()
|
||||||
|
|
|
@ -17,10 +17,3 @@ class Job(Thread):
|
||||||
def run(self):
|
def run(self):
|
||||||
while not self.stopped.wait(self.interval.total_seconds()):
|
while not self.stopped.wait(self.interval.total_seconds()):
|
||||||
self.execute(*self.args, **self.kwargs)
|
self.execute(*self.args, **self.kwargs)
|
||||||
|
|
||||||
def sample_func(arg1, arg2, arg3):
|
|
||||||
print "returning {} {} {}".format(arg1, arg2, arg3)
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
|
||||||
j = Job(timedelta(seconds=5), sample_func, **{"arg1": "sankalp", "arg2": "jonna", "arg3": "newsomething"})
|
|
||||||
j.start()
|
|
||||||
|
|
Loading…
Reference in a new issue