refactor: move Anki API interactions to separate anki.py module

This commit is contained in:
Yandrik (aider) 2025-03-20 11:28:59 +01:00
parent 35df3eaab8
commit 9a29db84ef
2 changed files with 62 additions and 43 deletions

56
anki.py Normal file
View File

@ -0,0 +1,56 @@
import requests
import logging
from typing import List, Dict
logger = logging.getLogger("flashcard-creator")
class AnkiConnect:
def __init__(self, url="http://127.0.0.1:8765"):
self.url = url
def get_decks(self) -> List[str]:
"""Get the list of decks from Anki using AnkiConnect."""
try:
response = requests.post(
self.url, json={"action": "deckNames", "version": 6}
)
data = response.json()
return data.get("result", [])
except Exception as e:
logger.error(f"Error getting Anki decks: {e}")
return []
def create_cards(self, deck: str, cards: List[Dict[str, str]]) -> bool:
"""Create flashcards in Anki using AnkiConnect."""
try:
# Prepare notes for addition
notes = []
logger.info(f"Preparing to create {len(cards)} Anki cards in deck: {deck}")
for card in cards:
notes.append(
{
"deckName": deck,
"modelName": "Basic", # Using the basic model
"fields": {"Front": card["front"], "Back": card["back"]},
"options": {"allowDuplicate": False},
"tags": [f"{self.source_language}-{self.target_language}"],
}
)
# Add notes to Anki
logger.info("Sending request to AnkiConnect to add notes")
response = requests.post(
self.url,
json={"action": "addNotes", "version": 6, "params": {"notes": notes}},
)
data = response.json()
success = all(id is not None for id in data.get("result", []))
if success:
logger.info("Successfully added all cards to Anki")
else:
logger.warning("Some cards failed to be added to Anki")
return success
except Exception as e:
logger.error(f"Error creating Anki cards: {e}")
return False

49
main.py
View File

@ -16,8 +16,9 @@ logger = logging.getLogger("flashcard-creator")
class FlashcardCreator:
def __init__(self):
# AnkiConnect configuration
self.anki_connect_url = "http://127.0.0.1:8765"
# Anki configuration
from anki import AnkiConnect
self.anki = AnkiConnect()
# File picker
self.file_picker = ft.FilePicker()
@ -640,49 +641,11 @@ class FlashcardCreator:
def get_anki_decks(self) -> List[str]:
"""Get the list of decks from Anki using AnkiConnect."""
try:
response = requests.post(
self.anki_connect_url, json={"action": "deckNames", "version": 6}
)
data = response.json()
return data.get("result", [])
except Exception as e:
print(f"Error getting Anki decks: {e}")
return []
return self.anki.get_decks()
def create_anki_cards(self, deck: str, cards: List[Dict[str, str]]) -> bool:
"""Create flashcards in Anki using AnkiConnect."""
try:
# Prepare notes for addition
notes = []
logging.info(f"Preparing to create {len(cards)} Anki cards in deck: {deck}")
for card in cards:
notes.append(
{
"deckName": deck,
"modelName": "Basic", # Using the basic model
"fields": {"Front": card["front"], "Back": card["back"]},
"options": {"allowDuplicate": False},
"tags": [f"{self.source_language}-{self.target_language}"],
}
)
# Add notes to Anki
logging.info("Sending request to AnkiConnect to add notes")
response = requests.post(
self.anki_connect_url,
json={"action": "addNotes", "version": 6, "params": {"notes": notes}},
)
data = response.json()
success = all(id is not None for id in data.get("result", []))
if success:
logging.info("Successfully added all cards to Anki")
else:
logging.warning("Some cards failed to be added to Anki")
return success
except Exception as e:
logging.error(f"Error creating Anki cards: {e}")
return False
"""Create flashcards in Anki using Anki-connect."""
return self.anki.create_cards(deck, cards)
def export_as_yaml(self, e):
"""Export the flashcards as a YAML file."""