45 lines
1.9 KiB
Python
45 lines
1.9 KiB
Python
#!/usr/bin/env python3
|
|
"""Export stable desktop session/exercise equipment associations."""
|
|
import argparse
|
|
import json
|
|
import sqlite3
|
|
import os
|
|
from datetime import datetime
|
|
from pathlib import Path
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("output", type=Path)
|
|
parser.add_argument("--database", type=Path,
|
|
default=Path(os.environ.get("XDG_DATA_HOME", str(Path.home() / ".local/share"))) / "trainlog" / "trainlog.db")
|
|
args = parser.parse_args()
|
|
connection = sqlite3.connect(args.database)
|
|
try:
|
|
if connection.execute("PRAGMA user_version;").fetchone()[0] != 7:
|
|
raise ValueError("schema desktop v7 requis")
|
|
rows = connection.execute(
|
|
"SELECT s.session_id,se.entry_id,e.exercise_id,se.equipment_id FROM session_exercises se "
|
|
"JOIN sessions s ON s.id=se.session_row_id JOIN exercises e ON e.id=se.exercise_row_id "
|
|
"ORDER BY s.started_at,s.id,se.position").fetchall()
|
|
payload = {"format": "trainlog-equipment-associations", "version": 2,
|
|
"generated_at": datetime.now().astimezone().isoformat(), "associations": []}
|
|
for session_id, entry_id, exercise_id, equipment_id in rows:
|
|
item = {"session_id": session_id, "entry_id": entry_id, "exercise_id": exercise_id,
|
|
"state": "set" if equipment_id is not None else "cleared"}
|
|
if equipment_id is not None:
|
|
item["equipment_id"] = equipment_id
|
|
payload["associations"].append(item)
|
|
args.output.write_text(json.dumps(payload, ensure_ascii=False, separators=(",", ":")), encoding="utf-8")
|
|
print("EQUIPMENT_ASSOCIATIONS_EXPORT=PASS")
|
|
print(f"associations={len(rows)}")
|
|
finally:
|
|
connection.close()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
try:
|
|
main()
|
|
except Exception as error:
|
|
print(f"EQUIPMENT_ASSOCIATIONS_EXPORT=FAIL {error}")
|
|
raise SystemExit(1)
|