#!/usr/bin/python

import csv
from datetime import datetime as dt
import os

DATA_DIR=os.path.expanduser('~/Journal/2026/Data')

data_files = {}
for root, _, files in os.walk(DATA_DIR):
    for file in files:
        name, ext = os.path.splitext(file)
        if ext == '.csv':
            data_files[name] = os.path.join(root, file)

print(data_files.keys())
choice = input()

file = data_files[choice]

print(file)

with open(file, 'r') as f:
    reader = csv.reader(f, delimiter=',', quotechar='"')
    fields = next(reader)  # require header row
    first_example = next(reader)
    for line in reader:
        last_example = next(reader)

print(first_example)
print("...")
print(last_example)
    

DATE_FORMAT = '%Y-%m-%d'

def today():
    return dt.now().strftime(DATE_FORMAT)

values = {}
for key in fields:
    default = ''
    if key == 'date':
        default = today()
    values[key] = input(f"{key} [{default}]\n").strip() or default

print("Break with Ctrl+C if incorrect:")
print(values)
input()  # Assume break with Ctrl+C if incorrect

with open(file, 'a') as f:
    writer = csv.DictWriter(f, fields)
    writer.writerow(values)
    print(f"Updated {file}")