mirror-github-bspeice/metrik
1
0
Fork 0
mirror of https://github.com/bspeice/metrik synced 2025-07-03 06:45:07 -04:00
Code Issues Releases Wiki Activity

Switch live to a BoolParameter

Browse Source 
Instead of constructor argument
This commit is contained in:
Bradlee Speice
2016-08-23 19:01:09 -04:00
parent 495a491096
commit 49953e9da4
2 changed files with 17 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
metrik/tasks/base.py
View File

@ -3,7 +3,7 @@ from __future__ import print_function
import logging
from luigi import Task
from luigi.parameter import DateMinuteParameter
from luigi.parameter import DateMinuteParameter, BoolParameter
from metrik.targets.mongo import MongoTarget
from metrik.targets.noop import NoOpTarget
@ -38,19 +38,20 @@ class MongoNoBackCreateTask(MongoCreateTask):
# Have one parameter to make sure that the MongoTarget created by `super`
# doesn't blow up.
current_datetime = DateMinuteParameter()
live = BoolParameter()
def __init__(self, live=False, *args, **kwargs):
def __init__(self, *args, **kwargs):
super(MongoNoBackCreateTask, self).__init__(*args, **kwargs)
self.live = live
child_name = type(self).__name__
if not live:
logging.warning('Trying to create {child_name} without running'
if not self.live:
logging.warning('Trying to create {} without running'
' live, errors potentially to ensue.'.format(child_name))
def output(self):
if self.live:
return super(MongoNoBackCreateTask, self).output()
else:
# TODO: return target closest to self.current_datetime
return NoOpTarget()
def run(self):