mirror of
https://github.com/vale981/jobmanager
synced 2025-03-05 18:01:38 -05:00
95 lines
2.3 KiB
Python
95 lines
2.3 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
from __future__ import division, print_function
|
|
|
|
import os
|
|
import sys
|
|
import time
|
|
|
|
from os.path import abspath, dirname, split
|
|
|
|
# Add parent directory to beginning of path variable
|
|
sys.path = [split(dirname(abspath(__file__)))[0]] + sys.path
|
|
|
|
from jobmanager import decorators, progress
|
|
|
|
@decorators.ProgressBar
|
|
def _my_func_1(arg,
|
|
kwarg = "1",
|
|
count = decorators.progress.UnsignedIntValue(val=0),
|
|
max_count = decorators.progress.UnsignedIntValue(val=1)):
|
|
maxval = 100
|
|
max_count.value = maxval
|
|
|
|
for i in range(maxval):
|
|
count.value += 1
|
|
time.sleep(0.02)
|
|
|
|
return arg+kwarg
|
|
|
|
|
|
def _my_func_2(arg,
|
|
c = decorators.progress.UnsignedIntValue(val=0),
|
|
m = decorators.progress.UnsignedIntValue(val=1),
|
|
kwarg = "2"):
|
|
maxval = 100
|
|
m.value += maxval
|
|
|
|
for i in range(maxval):
|
|
c.value += 1
|
|
time.sleep(0.02)
|
|
|
|
return arg+kwarg
|
|
|
|
|
|
def test_ProgressBar():
|
|
result1 = _my_func_1("decorated function", kwarg=" 1")
|
|
print(result1)
|
|
|
|
wrapper = decorators.ProgressBar(_my_func_2, interval=.1)
|
|
result2 = wrapper("wrapped function", kwarg=" 2")
|
|
print(result2)
|
|
|
|
@decorators.ProgressBar
|
|
def my_func(c, m):
|
|
for i in range(m.value):
|
|
c.value = i
|
|
time.sleep(0.02)
|
|
|
|
def test_decorator():
|
|
c = progress.UnsignedIntValue(val=0)
|
|
m = progress.UnsignedIntValue(val=100)
|
|
my_func(c=c, m=m)
|
|
my_func(c, m)
|
|
|
|
|
|
def my_func_ProgressBarOverrideCount(c = None, m = None):
|
|
maxVal = 100
|
|
if m is not None:
|
|
m.value = maxVal
|
|
|
|
for i in range(maxVal):
|
|
time.sleep(0.03)
|
|
if c is not None:
|
|
c.value = i
|
|
|
|
|
|
def test_ProgressBarOverrideCount():
|
|
print("normal call -> no decoration")
|
|
my_func_ProgressBarOverrideCount()
|
|
print("done!")
|
|
print()
|
|
|
|
my_func_ProgressBarOverrideCount_dec = decorators.ProgressBarOverrideCount(my_func_ProgressBarOverrideCount)
|
|
print("with decorator")
|
|
my_func_ProgressBarOverrideCount_dec()
|
|
print("done!")
|
|
|
|
if __name__ == "__main__":
|
|
test_ProgressBar()
|
|
test_decorator()
|
|
test_ProgressBarOverrideCount()
|
|
|
|
|
|
|
|
|