Add main function to on_modify.py

Make it unit-testable
This commit is contained in:
Thomas Lauf 2023-01-01 22:12:43 +01:00
parent 431db0fb84
commit 056d239f35

View file

@ -42,11 +42,6 @@ try:
except AttributeError: except AttributeError:
input_stream = sys.stdin input_stream = sys.stdin
# Make no changes to the task, simply observe.
old = json.loads(input_stream.readline().decode("utf-8", errors="replace"))
new = json.loads(input_stream.readline().decode("utf-8", errors="replace"))
print(json.dumps(new))
def extract_tags_from(json_obj): def extract_tags_from(json_obj):
# Extract attributes for use as tags. # Extract attributes for use as tags.
@ -69,6 +64,8 @@ def extract_annotation_from(json_obj):
return json_obj['annotations'][0]['description'] return json_obj['annotations'][0]['description']
def main(old, new):
start_or_stop = '' start_or_stop = ''
# Started task. # Started task.
@ -98,3 +95,10 @@ elif 'start' in new and 'start' in old:
if old_annotation != new_annotation: if old_annotation != new_annotation:
subprocess.call(['timew', 'annotate', '@1', new_annotation]) subprocess.call(['timew', 'annotate', '@1', new_annotation])
if __name__ == "__main__":
old = json.loads(input_stream.readline().decode("utf-8", errors="replace"))
new = json.loads(input_stream.readline().decode("utf-8", errors="replace"))
print(json.dumps(new))
main(old, new)