Skip to content

Commit 7f04169

Browse files
committed
many variables moved to settings
1 parent 7005e7a commit 7f04169

1 file changed

Lines changed: 14 additions & 9 deletions

File tree

pytrackunit/tucache.py

Lines changed: 14 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -12,19 +12,23 @@ class TuCache:
1212
"""tucache class"""
1313
def __init__(self,**kwargs):
1414
self.cache = WebCache(**kwargs)
15-
self.req_period = 30
16-
self.tdelta_end = kwargs.get("tdelta_end",None)
15+
self.settings = kwargs
16+
self.settings.setdefault('req_period',30)
17+
self.settings.setdefault('tdelta_end',None)
18+
self.settings.setdefault('return_only_cache_files',False)
19+
self.settings.setdefault('dont_return_data',False)
20+
self.settings.setdefault('dont_read_files',False)
1721
def clean(self):
1822
"""deletes all cached data"""
1923
self.cache.clean()
2024
async def get_url(self,url):
2125
"""takes the data from cache if possible. otherwise data is loaded from web"""
2226
data = await self.cache.get(URL_BASE+url)
23-
if self.cache.return_only_cache_files:
27+
if self.settings['return_only_cache_files']:
2428
return [data]
25-
if self.cache.dont_return_data:
29+
if self.settings['dont_return_data']:
2630
return []
27-
if self.cache.dont_read_files and len(data) == 0:
31+
if self.settings['dont_read_files'] and len(data) == 0:
2832
return []
2933
return data.get('list')
3034
async def get_unitlist(self):
@@ -33,7 +37,7 @@ async def get_unitlist(self):
3337
def general_daydiff_get(self,furl,meta,previter=None):
3438
"""returns data for timedependant requests for a given daydelta"""
3539
tdelta = meta["tdelta"]
36-
start,end = start_end_from_tdelta(tdelta,self.tdelta_end)
40+
start,end = start_end_from_tdelta(tdelta,self.settings['tdelta_end'])
3741
meta["start"] = start
3842
meta["end"] = end
3943
return self.general_time_range_get(furl,meta,previter)
@@ -42,10 +46,11 @@ def general_time_range_get(self,furl,meta,previter=None):
4246
start = meta["start"]
4347
end = meta["end"]
4448
days = (end-start).days
49+
req_period = self.settings['req_period']
4550
requests = []
46-
for week in range(ceil(days/self.req_period)):
47-
wstart = start+timedelta(days=week*self.req_period)
48-
wend = wstart+timedelta(days=min(self.req_period,(end-wstart).days))
51+
for week in range(ceil(days/req_period)):
52+
wstart = start+timedelta(days=week*req_period)
53+
wend = wstart+timedelta(days=min(req_period,(end-wstart).days))
4954
requests.append(furl(\
5055
wstart.strftime("%Y-%m-%dT%H:%M:%S"),\
5156
wend.strftime("%Y-%m-%dT%H:%M:%S")))

0 commit comments

Comments
 (0)