from datetime import datetime
import os
import pandas as pd
import pytz
from waveforms.models import Annotation
from website.settings import base
def update_annotations():
"""
Automatically download the annotations to a CSV file as a cron job.
Parameters
----------
N/A
Returns
-------
N/A
"""
# Update the annotation CSV file each night at midnight.
all_anns = Annotation.objects.all()
csv_df = pd.DataFrame.from_dict({
'username': [a.user.username for a in all_anns],
'project' : [a.project for a in all_anns],
'record': [a.record for a in all_anns],
'event': [a.event for a in all_anns],
'decision': [a.decision for a in all_anns],
'comments': [a.comments for a in all_anns],
'decision_date': [str(a.decision_date) for a in all_anns],
'is_adjudication': [a.is_adjudication for a in all_anns],
})
file_name = datetime.now(pytz.timezone(base.TIME_ZONE)).strftime('all-anns_%H_%M_%d_%m_%Y.csv')
out_file = os.path.join(base.HEAD_DIR, 'backups', file_name)
csv_df.to_csv(out_file, sep=',', index=False)