diff --git a/requirements.txt b/requirements.txt index df5ae38..53a6226 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1 +1,4 @@ yfinance==0.1.54 + +python-dateutil~=2.8.1 +setuptools~=54.0.0 \ No newline at end of file diff --git a/tests/Json2Yaml.py b/tests/Json2Yaml.py index b13090a..0862022 100644 --- a/tests/Json2Yaml.py +++ b/tests/Json2Yaml.py @@ -1,22 +1,36 @@ # Python script to Read file from JSON and convert to YAML -# Author - HoneyMoose (https://www.ossez.com) +# Author - yucheng.hu@insight.com +import datetime import json import ruamel.yaml as yaml +from dateutil.relativedelta import relativedelta json_filename = 'resources/black_rock_test.json' yaml_filename = 'resources/black_rock_test.yaml' +ELIGIBLE_FOR_RETIREMENT = 'eligible_for_retirement' + + +# Get Difference Years +def get_age(data_input): + date_user = datetime.datetime.strptime(data_input, '%m/%d/%Y') + date_current = datetime.datetime.now() + + time_difference = relativedelta(date_current, date_user) + return time_difference.years + + # Read and process JSON with open(json_filename) as json_file: data = json.load(json_file) - print(type(data)) + data_bod = data['date_of_birth'] - for data_dict in data: - label_id = data_dict['label_id'] - print(str(label_id)) -json_file.close() + if get_age(data_bod) >= 65: + data[ELIGIBLE_FOR_RETIREMENT] = True + else: + data[ELIGIBLE_FOR_RETIREMENT] = False # Write to YAML with open(yaml_filename, 'w') as yaml_file: diff --git a/tests/resources/black_rock_test.json b/tests/resources/black_rock_test.json index b608e34..e26a7fc 100644 --- a/tests/resources/black_rock_test.json +++ b/tests/resources/black_rock_test.json @@ -1,10 +1,5 @@ -[ - { - "label_id": 603, - "user_name": "YUCHENG-L1" - }, - { - "label_id": 604, - "user_name": "YUCHENG-L2" - } -] \ No newline at end of file +{ + "first_name": "John", + "last_name": "Doe", + "date_of_birth": "1/1/2021" +} \ No newline at end of file diff --git a/tests/resources/honeymoose_test.json b/tests/resources/honeymoose_test.json index b608e34..e26a7fc 100644 --- a/tests/resources/honeymoose_test.json +++ b/tests/resources/honeymoose_test.json @@ -1,10 +1,5 @@ -[ - { - "label_id": 603, - "user_name": "YUCHENG-L1" - }, - { - "label_id": 604, - "user_name": "YUCHENG-L2" - } -] \ No newline at end of file +{ + "first_name": "John", + "last_name": "Doe", + "date_of_birth": "1/1/2021" +} \ No newline at end of file