添加日期时间的代码

tests/DateTime.py

@@ -0,0 +1,39 @@
+# -*- coding: utf-8 -*-
+
+# Python script to Process DateTime
+# Author - yucheng.hu@ingith.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)
+    data_bod = data['date_of_birth']
+
+    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:
+    yaml.dump(data, yaml_file, allow_unicode=True)