35 lines
1.1 KiB
Python
35 lines
1.1 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 'timestamp' in kwargs:
|
|
return True
|
|
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 'timestamp' in kwargs:
|
|
timestamp = kwargs['timestamp']
|
|
if 'timezone' in kwargs:
|
|
timezone = kwargs['timezone']
|
|
else:
|
|
timezone = 'US/Eastern'
|
|
parsed_timestamp = dateparser.parse(timestamp)
|
|
localtime = parsed_timestamp.astimezone(pytz.timezone(timezone))
|
|
return str(localtime)
|
|
|
|
else:
|
|
raise ApiError(code='missing_timestamp',
|
|
message='You must include a timestamp to convert.')
|
|
|