mirror of
https://github.com/sartography/cr-connect-workflow.git
synced 2025-02-23 13:18:35 +00:00
1f9c80d70d
Modified the workflow so the first task is a form with booleans that allows me to test all the permutations. Modified tests to cover all the permutations for calling the script
40 lines
1.4 KiB
Python
40 lines
1.4 KiB
Python
from crc.api.common import ApiError
|
|
from crc.scripts.script import Script
|
|
|
|
import dateparser
|
|
import pytz
|
|
|
|
|
|
class GetLocaltime(Script):
|
|
|
|
def get_description(self):
|
|
return """Converts a UTC Datetime object into a Datetime object with a different timezone.
|
|
Defaults to US/Eastern"""
|
|
|
|
def do_task_validate_only(self, task, study_id, workflow_id, *args, **kwargs):
|
|
if len(args) > 0 or 'timestamp' in kwargs:
|
|
return self.do_task(task, study_id, workflow_id, *args, **kwargs)
|
|
raise ApiError(code='missing_timestamp',
|
|
message='You must include a timestamp to convert.')
|
|
|
|
def do_task(self, task, study_id, workflow_id, *args, **kwargs):
|
|
if len(args) > 0 or 'timestamp' in kwargs:
|
|
if 'timestamp' in kwargs:
|
|
timestamp = kwargs['timestamp']
|
|
else:
|
|
timestamp = args[0]
|
|
if 'timezone' in kwargs:
|
|
timezone = kwargs['timezone']
|
|
elif len(args) > 1:
|
|
timezone = args[1]
|
|
else:
|
|
timezone = 'US/Eastern'
|
|
parsed_timestamp = dateparser.parse(timestamp)
|
|
localtime = parsed_timestamp.astimezone(pytz.timezone(timezone))
|
|
return localtime
|
|
|
|
else:
|
|
raise ApiError(code='missing_timestamp',
|
|
message='You must include a timestamp to convert.')
|
|
|