diff --git a/main.py b/main.py index 29b450e..d473e32 100644 --- a/main.py +++ b/main.py @@ -45,6 +45,12 @@ def _get_content(obj): return obj.get("content") if isinstance(obj, dict) else getattr(obj, "content", "") +def _children(obj): + if isinstance(obj, dict): + return obj.get("children") or [] + return getattr(obj, "children", []) or [] + + def _translate_month_en_ru(name: str) -> str: mapping = { "January": "Январь", @@ -101,11 +107,10 @@ def save_daily(text: str) -> str: # Ищем/создаём месяц month_folder = None - if hasattr(daily_root, "children"): - for child in daily_root.children: - if _get_title(child) == month_name: - month_folder = child - break + for child in _children(daily_root): + if _get_title(child) == month_name: + month_folder = child + break if not month_folder: month_folder = ea.create_note( @@ -117,11 +122,10 @@ def save_daily(text: str) -> str: # Ищем/создаём день day_folder = None - if hasattr(month_folder, "children"): - for child in month_folder.children: - if _get_title(child) == day_name: - day_folder = child - break + for child in _children(month_folder): + if _get_title(child) == day_name: + day_folder = child + break if not day_folder: day_folder = ea.create_note(parentNoteId=month_id, title=day_name, type="text") @@ -131,11 +135,10 @@ def save_daily(text: str) -> str: # Ищем заметку по дате existing_note = None - if hasattr(day_folder, "children"): - for child in day_folder.children: - if _get_title(child) == today_date: - existing_note = child - break + for child in _children(day_folder): + if _get_title(child) == today_date: + existing_note = child + break if existing_note: existing_content = _get_content(existing_note)