671 lines
27 KiB
Python
671 lines
27 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
"""
|
|
IRC Bot Plugin for Note Taking using TinyDB.
|
|
|
|
This plugin provides a single !note command with subcommands to manage both
|
|
daily and general notes. It supports the following subcommands:
|
|
|
|
today [--add|--replace] [<content>...]
|
|
View or update today's daily note. If content is provided and a note for
|
|
today already exists, a warning is shown unless you specify --add (to append)
|
|
or --replace (to replace).
|
|
|
|
daily [--date <date>] [--add|--replace] [<content>...]
|
|
View or update a daily note for the specified date (default is today). If
|
|
content is provided and a note already exists, a warning is shown unless you
|
|
specify --add (to append) or --replace (to replace).
|
|
|
|
general <title> [--tags <tags>] [<content>...]
|
|
View or update a general note with the given title.
|
|
Optionally specify tags.
|
|
|
|
list <type>
|
|
List all notes of the specified type (either 'daily' or 'general'). The
|
|
content of each note is displayed truncated.
|
|
|
|
search <type> <keyword>
|
|
Search for a keyword in notes of the specified type.
|
|
|
|
summary
|
|
Display a summary of note counts.
|
|
|
|
Notes are stored persistently in the TinyDB database file "notes.json". Daily
|
|
notes are keyed by date (YYYY-MM-DD) and general notes by a slugified title.
|
|
|
|
Dependencies:
|
|
- irc3
|
|
- ircstyle
|
|
- tinydb
|
|
|
|
Author: Your Name
|
|
Version: 1.3
|
|
Date: 2025-02-18
|
|
"""
|
|
|
|
import re
|
|
from datetime import datetime, date
|
|
from tinydb import TinyDB, Query
|
|
import irc3
|
|
from irc3.plugins.command import command
|
|
import ircstyle
|
|
import logging
|
|
|
|
# -----------------------------------------------------------------------------
|
|
# Helper Functions
|
|
# -----------------------------------------------------------------------------
|
|
|
|
def slugify(text):
|
|
"""
|
|
Convert text to a slug (lowercase, underscores, alphanumerics only).
|
|
|
|
Args:
|
|
text (str): The text to slugify.
|
|
|
|
Returns:
|
|
str: A slugified version of the text.
|
|
"""
|
|
text = text.lower()
|
|
text = re.sub(r'\s+', '_', text)
|
|
text = re.sub(r'[^a-z0-9_-]', '', text)
|
|
return text
|
|
|
|
|
|
def now_str():
|
|
"""
|
|
Return the current datetime as an ISO formatted string.
|
|
|
|
Returns:
|
|
str: The current datetime.
|
|
"""
|
|
return datetime.now().isoformat()
|
|
|
|
|
|
def today_str():
|
|
"""
|
|
Return today's date in ISO format (YYYY-MM-DD).
|
|
|
|
Returns:
|
|
str: Today's date.
|
|
"""
|
|
return date.today().isoformat()
|
|
|
|
|
|
def default_daily_template(day):
|
|
"""
|
|
Generate a default daily note template.
|
|
|
|
Args:
|
|
day (str): The date string (YYYY-MM-DD).
|
|
|
|
Returns:
|
|
str: Default content for a daily note.
|
|
"""
|
|
return f"""# Daily Note for {day}
|
|
|
|
## To do:
|
|
|
|
## Tomorrow:
|
|
|
|
## Reminder:
|
|
|
|
## Journal:
|
|
"""
|
|
|
|
|
|
def default_general_template(title, tags):
|
|
"""
|
|
Generate a default general note template.
|
|
|
|
Args:
|
|
title (str): The note title.
|
|
tags (str): Tags for the note.
|
|
|
|
Returns:
|
|
str: Default content for a general note.
|
|
"""
|
|
tag_line = f"Tags: {tags}" if tags else ""
|
|
return f"""# General Note: {title}
|
|
{tag_line} Created on: {now_str()}
|
|
|
|
"""
|
|
|
|
|
|
def truncate(text, length=50):
|
|
"""
|
|
Truncate the text to a specified length, appending "..." if truncated.
|
|
|
|
Args:
|
|
text (str): The text to truncate.
|
|
length (int): Maximum length of the returned text.
|
|
|
|
Returns:
|
|
str: The truncated text.
|
|
"""
|
|
if not isinstance(text, str):
|
|
text = str(text)
|
|
return text if len(text) <= length else text[:length] + "..."
|
|
|
|
|
|
# -----------------------------------------------------------------------------
|
|
# NoteTaking Plugin
|
|
# -----------------------------------------------------------------------------
|
|
|
|
@irc3.plugin
|
|
class NoteTaking:
|
|
"""
|
|
IRC3 plugin for unified note taking.
|
|
|
|
Provides a single !note command with subcommands to create, update, view,
|
|
list, and search both daily and general notes stored in TinyDB.
|
|
"""
|
|
|
|
requires = [
|
|
'plugins.users', # Ensure userlist plugin is loaded, if using user-specific notes
|
|
]
|
|
|
|
def __init__(self, bot):
|
|
"""
|
|
Initialize the NoteTaking plugin.
|
|
|
|
Args:
|
|
bot (irc3.IrcBot): The IRC bot instance.
|
|
"""
|
|
self.bot = bot
|
|
# Configuration options with defaults
|
|
self.db_filename = self.bot.config.get('note_db_filename', 'notes.json')
|
|
self.daily_template = self.bot.config.get('daily_template', default_daily_template)
|
|
self.general_template = self.bot.config.get('general_template', default_general_template)
|
|
|
|
try:
|
|
self.db = TinyDB(self.db_filename)
|
|
self.daily_table = self.db.table('daily')
|
|
self.general_table = self.db.table('general')
|
|
except Exception as e:
|
|
self.bot.log.error(f"Error initializing TinyDB: {e}")
|
|
raise # Re-raise to prevent the plugin from loading
|
|
|
|
self.User = Query()
|
|
self.log = logging.getLogger(__name__)
|
|
|
|
@command(aliases=['notes'], public=True, options_first=False, use_shlex=True)
|
|
async def note(self, mask, target, args):
|
|
"""
|
|
%%note today [--add|--replace] [<content>...]
|
|
%%note daily [--date=<date>] [--add|--replace] [<content>...]
|
|
%%note general <title> [--tags=<tags>] [<content>...]
|
|
%%note list [<type>]
|
|
%%note search <type> <keyword>
|
|
%%note summary
|
|
%%note
|
|
"""
|
|
# If no subcommand is provided, show the help message.
|
|
if not any(args.get(key) for key in ('today', 'daily', 'general', 'list', 'search', 'summary')):
|
|
# If there's content provided without a subcommand, treat it as a daily note update.
|
|
if args.get('<content>'):
|
|
await self._note_today(mask, target, args)
|
|
else:
|
|
self._show_help(target)
|
|
return
|
|
|
|
try:
|
|
if args.get('today'):
|
|
await self._note_today(mask, target, args)
|
|
elif args.get('daily'):
|
|
await self._note_daily(mask, target, args)
|
|
elif args.get('general'):
|
|
await self._note_general(mask, target, args)
|
|
elif args.get('list'):
|
|
await self._note_list(target, args)
|
|
elif args.get('search'):
|
|
await self._note_search(target, args)
|
|
elif args.get('summary'):
|
|
await self._note_summary(target)
|
|
except Exception as e:
|
|
self.bot.log.error(f"Error processing note command: {e}", exc_info=True)
|
|
self.bot.privmsg(
|
|
target,
|
|
ircstyle.style(f"❌ Error processing command: {e}", fg="red", bold=True)
|
|
)
|
|
|
|
def _show_help(self, target):
|
|
help_lines = [
|
|
ircstyle.style("Usage: !note <subcommand> [options]", fg="red", bold=True),
|
|
"Subcommands:",
|
|
f"{ircstyle.style(' today [--add|--replace] [<content>...]', fg='yellow')} - View or update today's daily note.",
|
|
f"{ircstyle.style(' daily [--date <date>] [--add|--replace] [<content>...]', fg='yellow')} - View or update a daily note for a specific date (default is today).",
|
|
f"{ircstyle.style(' general <title> [--tags <tags>] [<content>...]', fg='yellow')} - View or update a general note with the given title. Optionally, specify tags.",
|
|
f"{ircstyle.style(' list [daily|general]', fg='yellow')} - List all notes of a specific type with truncated content.",
|
|
f"{ircstyle.style(' search <type> <keyword>', fg='yellow')} - Search for a keyword in notes of a specific type.",
|
|
f"{ircstyle.style(' summary', fg='yellow')} - Display a summary of note counts.",
|
|
]
|
|
|
|
for line in help_lines:
|
|
self.bot.privmsg(target, line)
|
|
|
|
async def _note_today(self, mask, target, args):
|
|
"""
|
|
Handle the 'today' subcommand for daily notes.
|
|
|
|
If content is provided, update today's note.
|
|
If a note already exists and content is provided, warn the user unless
|
|
--add or --replace is specified.
|
|
"""
|
|
day = today_str()
|
|
content_list = args.get('<content>')
|
|
try:
|
|
note = self._get_daily_note(day)
|
|
if content_list:
|
|
new_content = " ".join(content_list)
|
|
# Check if both flags are provided
|
|
if args.get('--add') and args.get('--replace'):
|
|
self.bot.privmsg(
|
|
target,
|
|
ircstyle.style("❌ Error: Cannot use both --add and --replace simultaneously. Please choose one.", fg="red", bold=True)
|
|
)
|
|
return
|
|
|
|
if note:
|
|
# If note exists but neither flag is provided, warn the user.
|
|
if not (args.get('--add') or args.get('--replace')):
|
|
warning = ircstyle.style(
|
|
f"⚠️ Warning: Daily note for {day} already exists. Use --add to append to it or --replace to overwrite it.", fg="red", bold=True
|
|
)
|
|
self.bot.privmsg(target, warning)
|
|
return
|
|
if args.get('--add'):
|
|
appended_content = note.get('content') + "\n" + new_content
|
|
self.daily_table.update(
|
|
{'content': appended_content, 'updated': now_str()},
|
|
self.User.date == day
|
|
)
|
|
msg = ircstyle.style("✅ ", fg="green") + ircstyle.style(
|
|
f"Daily note for {day} updated (content added).", fg="blue", bold=True
|
|
)
|
|
elif args.get('--replace'):
|
|
self.daily_table.update(
|
|
{'content': new_content, 'updated': now_str()},
|
|
self.User.date == day
|
|
)
|
|
msg = ircstyle.style("✅ ", fg="green") + ircstyle.style(
|
|
f"Daily note for {day} replaced.", fg="blue", bold=True
|
|
)
|
|
else:
|
|
# No existing note; create a new one.
|
|
self.daily_table.insert({
|
|
'date': day,
|
|
'content': new_content,
|
|
'created': now_str(),
|
|
'updated': now_str()
|
|
})
|
|
msg = ircstyle.style("✅ ", fg="green") + ircstyle.style(
|
|
f"Daily note for {day} created.", fg="blue", bold=True
|
|
)
|
|
else:
|
|
if not note:
|
|
default_content = self.daily_template(day)
|
|
self.daily_table.insert({
|
|
'date': day,
|
|
'content': default_content,
|
|
'created': now_str(),
|
|
'updated': now_str()
|
|
})
|
|
note = self._get_daily_note(day)
|
|
msg = ircstyle.style(
|
|
f"📝 Daily Note for {day}:\n", fg="yellow", bold=True
|
|
) + note.get('content')
|
|
self.bot.privmsg(target, msg)
|
|
|
|
except Exception as e:
|
|
self.bot.log.error(f"Error in _note_today: {e}", exc_info=True)
|
|
self.bot.privmsg(
|
|
target,
|
|
ircstyle.style(f"❌ Error: {e}", fg="red", bold=True)
|
|
)
|
|
|
|
async def _note_daily(self, mask, target, args):
|
|
"""
|
|
Handle the 'daily' subcommand for daily notes with an optional date.
|
|
|
|
If content is provided, update the note.
|
|
If a note already exists and content is provided, warn the user unless
|
|
--add or --replace is specified.
|
|
"""
|
|
day = args.get('--date') or today_str()
|
|
try:
|
|
datetime.strptime(day, '%Y-%m-%d') # Validate date format
|
|
except ValueError:
|
|
self.bot.privmsg(
|
|
target,
|
|
ircstyle.style("❌ Invalid date format. Use YYYY-MM-DD.", fg="red", bold=True)
|
|
)
|
|
return
|
|
|
|
content_list = args.get('<content>')
|
|
|
|
try:
|
|
note = self._get_daily_note(day)
|
|
if content_list:
|
|
new_content = " ".join(content_list)
|
|
# Check if both flags are provided
|
|
if args.get('--add') and args.get('--replace'):
|
|
self.bot.privmsg(
|
|
target,
|
|
ircstyle.style("❌ Error: Cannot use both --add and --replace simultaneously. Please choose one.", fg="red", bold=True)
|
|
)
|
|
return
|
|
|
|
if note:
|
|
if not (args.get('--add') or args.get('--replace')):
|
|
self.bot.privmsg(
|
|
target,
|
|
ircstyle.style(f"⚠️ Warning: Daily note for {day} already exists. Use --add to append or --replace to overwrite.", fg="red", bold=True)
|
|
)
|
|
return
|
|
if args.get('--add'):
|
|
appended_content = note.get('content') + "\n" + new_content
|
|
self.daily_table.update(
|
|
{'content': appended_content, 'updated': now_str()},
|
|
self.User.date == day
|
|
)
|
|
msg = ircstyle.style("✅ ", fg="green") + ircstyle.style(
|
|
f"Daily note for {day} updated (content added).", fg="blue", bold=True
|
|
)
|
|
elif args.get('--replace'):
|
|
self.daily_table.update(
|
|
{'content': new_content, 'updated': now_str()},
|
|
self.User.date == day
|
|
)
|
|
msg = ircstyle.style("✅ ", fg="green") + ircstyle.style(
|
|
f"Daily note for {day} replaced.", fg="blue", bold=True
|
|
)
|
|
else:
|
|
# Create new note if none exists.
|
|
self.daily_table.insert({
|
|
'date': day,
|
|
'content': new_content,
|
|
'created': now_str(),
|
|
'updated': now_str()
|
|
})
|
|
msg = ircstyle.style("✅ ", fg="green") + ircstyle.style(
|
|
f"Daily note for {day} created.", fg="blue", bold=True
|
|
)
|
|
else:
|
|
if not note:
|
|
default_content = self.daily_template(day)
|
|
self.daily_table.insert({
|
|
'date': day,
|
|
'content': default_content,
|
|
'created': now_str(),
|
|
'updated': now_str()
|
|
})
|
|
note = self._get_daily_note(day)
|
|
msg = ircstyle.style(
|
|
f"📝 Daily Note for {day}:\n", fg="yellow", bold=True
|
|
) + note.get('content')
|
|
self.bot.privmsg(target, msg)
|
|
|
|
except Exception as e:
|
|
self.bot.log.error(f"Error in _note_daily: {e}", exc_info=True)
|
|
self.bot.privmsg(
|
|
target,
|
|
ircstyle.style(f"❌ Error: {e}", fg="red", bold=True)
|
|
)
|
|
|
|
async def _note_general(self, mask, target, args):
|
|
"""
|
|
Handle the 'general' subcommand for general notes.
|
|
|
|
If content is provided, update the note; otherwise, display it.
|
|
"""
|
|
nick = mask.nick # Get the user's nickname
|
|
|
|
# Retrieve <title> as a string (not a list)
|
|
title = args.get('<title>')
|
|
if not title:
|
|
self.bot.privmsg(
|
|
target,
|
|
ircstyle.style("❌ ", fg="red")
|
|
+ ircstyle.style("Title is required.", fg="red", bold=True)
|
|
)
|
|
return
|
|
|
|
slug = slugify(title)
|
|
tags = args.get('--tags') or ""
|
|
content_list = args.get('<content>')
|
|
|
|
try:
|
|
# Now filter by owner (nick) so users only access their own general notes
|
|
note = self._get_general_note(slug, nick)
|
|
if content_list:
|
|
new_content = " ".join(content_list)
|
|
update_fields = {'content': new_content, 'updated': now_str()}
|
|
if tags:
|
|
update_fields['tags'] = tags
|
|
if note:
|
|
self.general_table.update(
|
|
update_fields,
|
|
(self.User.slug == slug) & (self.User.owner == nick)
|
|
)
|
|
msg = ircstyle.style("✅ ", fg="green") + ircstyle.style(
|
|
f"General note '{title}' updated.", fg="blue", bold=True
|
|
)
|
|
else:
|
|
self.general_table.insert({
|
|
'title': title,
|
|
'slug': slug,
|
|
'tags': tags,
|
|
'content': new_content,
|
|
'created': now_str(),
|
|
'updated': now_str(),
|
|
'owner': nick # Store the owner of the note
|
|
})
|
|
msg = ircstyle.style("✅ ", fg="green") + ircstyle.style(
|
|
f"General note '{title}' created.", fg="blue", bold=True
|
|
)
|
|
|
|
else:
|
|
if not note:
|
|
default_content = self.general_template(title, tags)
|
|
self.general_table.insert({
|
|
'title': title,
|
|
'slug': slug,
|
|
'tags': tags,
|
|
'content': default_content,
|
|
'created': now_str(),
|
|
'updated': now_str(),
|
|
'owner': nick # Store the owner of the note
|
|
})
|
|
note = self._get_general_note(slug, nick)
|
|
header = ircstyle.style(
|
|
f"📝 General Note: {note.get('title')}\n", fg="yellow", bold=True
|
|
)
|
|
tag_line = ircstyle.style(
|
|
f"Tags: {note.get('tags')}\n", fg="cyan"
|
|
)
|
|
msg = header + tag_line + note.get('content')
|
|
self.bot.privmsg(target, msg)
|
|
|
|
except Exception as e:
|
|
self.bot.log.error(f"Error in _note_general: {e}", exc_info=True)
|
|
self.bot.privmsg(
|
|
target,
|
|
ircstyle.style(f"❌ Error: {e}", fg="red", bold=True)
|
|
)
|
|
|
|
async def _note_list(self, target, args):
|
|
"""
|
|
Handle the 'list' subcommand to list notes with truncated content.
|
|
|
|
Usage:
|
|
%%note list [<type>]
|
|
where <type> is optional and can be 'daily' or 'general'.
|
|
If <type> is not provided, both daily and general notes are listed.
|
|
"""
|
|
note_type = args.get('<type>')
|
|
try:
|
|
if not note_type:
|
|
# List both daily and general notes with content truncated.
|
|
daily_notes = self.daily_table.all()
|
|
general_notes = self.general_table.all()
|
|
|
|
if not daily_notes and not general_notes:
|
|
msg = ircstyle.style("No notes found.", fg="red", bold=True)
|
|
else:
|
|
msg_lines = [ircstyle.style("📝 All Notes:", fg="blue", bold=True)]
|
|
if daily_notes:
|
|
msg_lines.append(ircstyle.style("📅 Daily Notes:", fg="yellow", bold=True))
|
|
# Sort daily notes by date.
|
|
daily_sorted = sorted(
|
|
[n for n in daily_notes if isinstance(n.get('date'), str)],
|
|
key=lambda x: x['date']
|
|
)
|
|
for note in daily_sorted:
|
|
date_str = note.get('date')
|
|
snippet = truncate(note.get('content', ''))
|
|
msg_lines.append(f" • {ircstyle.style(date_str, fg='green')}: {ircstyle.style(snippet, fg='white')}")
|
|
if general_notes:
|
|
msg_lines.append(ircstyle.style("📋 General Notes:", fg="cyan", bold=True))
|
|
# Sort general notes by title.
|
|
general_sorted = sorted(
|
|
[n for n in general_notes if isinstance(n.get('title'), str)],
|
|
key=lambda x: x['title']
|
|
)
|
|
for note in general_sorted:
|
|
title = note.get('title')
|
|
snippet = truncate(note.get('content', ''))
|
|
msg_lines.append(f" • {ircstyle.style(title, fg='green')}: {ircstyle.style(snippet, fg='white')}")
|
|
msg = "\n".join(msg_lines)
|
|
elif note_type == 'daily':
|
|
notes = self.daily_table.all()
|
|
valid_notes = [n for n in notes if isinstance(n.get('date'), str)]
|
|
if not valid_notes:
|
|
msg = ircstyle.style("No daily notes found.", fg="red", bold=True)
|
|
else:
|
|
msg_lines = [ircstyle.style("📅 Daily Notes:", fg="blue", bold=True)]
|
|
daily_sorted = sorted(valid_notes, key=lambda x: x['date'])
|
|
for note in daily_sorted:
|
|
date_str = note.get('date')
|
|
snippet = truncate(note.get('content', ''))
|
|
msg_lines.append(f" • {ircstyle.style(date_str, fg='green')}: {ircstyle.style(snippet, fg='white')}")
|
|
msg = "\n".join(msg_lines)
|
|
elif note_type == 'general':
|
|
notes = self.general_table.all()
|
|
valid_notes = [n for n in notes if isinstance(n.get('title'), str)]
|
|
if not valid_notes:
|
|
msg = ircstyle.style("No general notes found.", fg="red", bold=True)
|
|
else:
|
|
msg_lines = [ircstyle.style("📋 General Notes:", fg="blue", bold=True)]
|
|
general_sorted = sorted(valid_notes, key=lambda x: x['title'])
|
|
for note in general_sorted:
|
|
title = note.get('title')
|
|
snippet = truncate(note.get('content', ''))
|
|
msg_lines.append(f" • {ircstyle.style(title, fg='green')}: {ircstyle.style(snippet, fg='white')}")
|
|
msg = "\n".join(msg_lines)
|
|
else:
|
|
msg = ircstyle.style("Usage: !note list [daily|general]", fg="red", bold=True)
|
|
for line in msg.split('\n'):
|
|
self.bot.privmsg(target, line)
|
|
except Exception as e:
|
|
self.bot.log.error(f"Error in _note_list: {e}", exc_info=True)
|
|
self.bot.privmsg(target, ircstyle.style(f"❌ Error: {e}", fg="red", bold=True))
|
|
|
|
async def _note_search(self, target, args):
|
|
"""
|
|
Handle the 'search' subcommand to search notes by keyword.
|
|
Usage:
|
|
%%note search <type> <keyword>
|
|
"""
|
|
note_type = args.get('<type>')
|
|
keyword = args.get('<keyword>')
|
|
|
|
if note_type not in ('daily', 'general') or not keyword:
|
|
msg = ircstyle.style("Usage: !note search <daily|general> <keyword>", fg="red", bold=True)
|
|
self.bot.privmsg(target, msg)
|
|
return
|
|
try:
|
|
if note_type == 'daily':
|
|
notes = self.daily_table.search(self.User.content.search(keyword, flags=re.IGNORECASE))
|
|
header = "Daily"
|
|
else:
|
|
notes = self.general_table.search(self.User.content.search(keyword, flags=re.IGNORECASE))
|
|
header = "General"
|
|
if not notes:
|
|
msg = ircstyle.style(f"No {note_type} notes found with '{keyword}'.", fg="red", bold=True)
|
|
else:
|
|
results = []
|
|
for n in notes:
|
|
identifier = n.get('date') if note_type == 'daily' else n.get('title')
|
|
content_snippet = n.get('content')[:50] + "..." # Show a snippet of the content
|
|
results.append(f"{identifier} ({content_snippet})")
|
|
msg = ircstyle.style(f"{header} Notes matching '{keyword}':\n", fg="blue", bold=True) + "\n".join(results)
|
|
for line in msg.split('\n'):
|
|
self.bot.privmsg(target, line)
|
|
|
|
except Exception as e:
|
|
self.bot.log.error(f"Error in _note_search: {e}", exc_info=True)
|
|
self.bot.privmsg(
|
|
target,
|
|
ircstyle.style(f"❌ Error: {e}", fg="red", bold=True)
|
|
)
|
|
|
|
async def _note_summary(self, target):
|
|
"""
|
|
Handle the 'summary' subcommand to display note counts.
|
|
"""
|
|
try:
|
|
daily_count = len(self.daily_table.all())
|
|
general_count = len(self.general_table.all())
|
|
lines = [
|
|
ircstyle.style("===== NOTE SUMMARY =====", fg="teal", bold=True),
|
|
ircstyle.style(f"Daily notes: {daily_count}", fg="yellow"),
|
|
ircstyle.style(f"General notes: {general_count}", fg="cyan"),
|
|
ircstyle.style("===== END SUMMARY =====", fg="teal", bold=True)
|
|
]
|
|
for line in lines:
|
|
self.bot.privmsg(target, line)
|
|
except Exception as e:
|
|
self.bot.log.error(f"Error in _note_summary: {e}", exc_info=True)
|
|
self.bot.privmsg(
|
|
target,
|
|
ircstyle.style(f"❌ Error: {e}", fg="red", bold=True)
|
|
)
|
|
|
|
def _get_daily_note(self, day):
|
|
"""
|
|
Retrieve a daily note for the given day.
|
|
|
|
Args:
|
|
day (str): The date (YYYY-MM-DD).
|
|
|
|
Returns:
|
|
dict or None: The note if found, else None.
|
|
"""
|
|
try:
|
|
results = self.daily_table.search(self.User.date == day)
|
|
return results[0] if results else None
|
|
except Exception as e:
|
|
self.bot.log.error(f"Error in _get_daily_note: {e}", exc_info=True)
|
|
return None
|
|
|
|
def _get_general_note(self, slug, owner=None):
|
|
"""
|
|
Retrieve a general note by its slug and (optionally) owner.
|
|
|
|
Args:
|
|
slug (str): The slugified title.
|
|
owner (str, optional): The owner of the note.
|
|
|
|
Returns:
|
|
dict or None: The note if found, else None.
|
|
"""
|
|
try:
|
|
if owner:
|
|
results = self.general_table.search((self.User.slug == slug) & (self.User.owner == owner))
|
|
else:
|
|
results = self.general_table.search(self.User.slug == slug)
|
|
return results[0] if results else None
|
|
except Exception as e:
|
|
self.bot.log.error(f"Error in _get_general_note: {e}", exc_info=True)
|
|
return None
|