This commit is contained in:
parent
6b606475df
commit
c74920e478
|
@ -6,6 +6,7 @@ Sub-module for importing time-based data into Obsidian.
|
||||||
import datetime
|
import datetime
|
||||||
from logging import getLogger
|
from logging import getLogger
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
|
from typing import Any
|
||||||
|
|
||||||
from personal_data.util import load_csv_file
|
from personal_data.util import load_csv_file
|
||||||
|
|
||||||
|
@ -13,13 +14,7 @@ from .obsidian import ObsidianVault
|
||||||
|
|
||||||
logger = getLogger(__name__)
|
logger = getLogger(__name__)
|
||||||
|
|
||||||
|
def import_workout_csv(vault: ObsidianVault, rows: list[dict[str,Any]]) -> int:
|
||||||
def import_data(obsidian_path: Path, dry_run=True):
|
|
||||||
vault = ObsidianVault(obsidian_path, read_only=dry_run and 'silent' or None)
|
|
||||||
|
|
||||||
data_path = Path('/home/jmaa/Notes/workout.csv')
|
|
||||||
rows = load_csv_file(data_path)
|
|
||||||
logger.info('Loaded CSV with %d lines', len(rows))
|
|
||||||
num_updated = 0
|
num_updated = 0
|
||||||
for row in rows:
|
for row in rows:
|
||||||
date = row['Date']
|
date = row['Date']
|
||||||
|
@ -43,5 +38,41 @@ def import_data(obsidian_path: Path, dry_run=True):
|
||||||
if was_updated:
|
if was_updated:
|
||||||
num_updated += 1
|
num_updated += 1
|
||||||
del row, date
|
del row, date
|
||||||
|
return num_updated
|
||||||
|
|
||||||
|
def import_step_counts_csv(vault: ObsidianVault, rows: list[dict[str,Any]]) -> int:
|
||||||
|
MINIMUM = 300
|
||||||
|
|
||||||
|
num_updated = 0
|
||||||
|
|
||||||
|
rows_per_day = {}
|
||||||
|
for row in rows:
|
||||||
|
date = row['Start'].date()
|
||||||
|
rows_per_day.setdefault(date, [])
|
||||||
|
rows_per_day[date].append(row)
|
||||||
|
del date, row
|
||||||
|
|
||||||
|
|
||||||
|
steps_per_day = { date: sum(row['Steps'] for row in rows) for date, rows in rows_per_day.items()}
|
||||||
|
|
||||||
|
for date, steps in steps_per_day.items():
|
||||||
|
if steps < MINIMUM:
|
||||||
|
continue
|
||||||
|
was_updated = vault.add_statistic(date, 'Steps', steps)
|
||||||
|
if was_updated:
|
||||||
|
num_updated += 1
|
||||||
|
del date, steps, was_updated
|
||||||
|
|
||||||
|
return num_updated
|
||||||
|
|
||||||
|
def import_data(obsidian_path: Path, dry_run=True):
|
||||||
|
vault = ObsidianVault(obsidian_path, read_only=dry_run and 'silent' or None)
|
||||||
|
|
||||||
|
#data_path = Path('/home/jmaa/Notes/workout.csv')
|
||||||
|
data_path = Path('/home/jmaa/personal-archive/misc-data/step_counts_2023-07-26_to_2024-09-21.csv')
|
||||||
|
rows = load_csv_file(data_path)
|
||||||
|
logger.info('Loaded CSV with %d lines', len(rows))
|
||||||
|
#num_updated = import_workout_csv(vault, rows)
|
||||||
|
num_updated = import_step_counts_csv(vault, rows)
|
||||||
|
|
||||||
logger.info('Updated %d files', num_updated)
|
logger.info('Updated %d files', num_updated)
|
||||||
|
|
|
@ -41,9 +41,10 @@ class ObsidianVault:
|
||||||
) -> bool:
|
) -> bool:
|
||||||
if self.read_only == 'silent':
|
if self.read_only == 'silent':
|
||||||
logger.info(
|
logger.info(
|
||||||
'Real only ObsidianVault ignoring add_statistic(%s, "%s", ?)',
|
'Real only ObsidianVault ignoring add_statistic(%s, "%s", %s)',
|
||||||
date,
|
date,
|
||||||
statistic_key,
|
statistic_key,
|
||||||
|
amount,
|
||||||
)
|
)
|
||||||
return False
|
return False
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue
Block a user