hiro/kindle_fetch
1
0
Fork 0
mirror of https://github.com/vale981/kindle_fetch synced 2025-03-04 16:41:38 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

add logging

Browse source
This commit is contained in:
Valentin Boettcher 2024-06-08 13:32:57 -04:00
parent 62408b2126
commit c76e4137f8
No known key found for this signature in database
GPG key ID: E034E12B7AF56ACE
1 changed files with 18 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
kindle_fetch/kindle_fetch.py
View file

@ -5,6 +5,7 @@ import shutil
import urllib.request import urllib.request
import asyncio import asyncio
import argparse import argparse
import logging
import subprocess import subprocess
from aioimaplib import aioimaplib from aioimaplib import aioimaplib
from collections import namedtuple from collections import namedtuple
@ -14,6 +15,8 @@ from collections import namedtuple
from email.parser import BytesHeaderParser, BytesParser from email.parser import BytesHeaderParser, BytesParser
from dataclasses import dataclass from dataclasses import dataclass
logger = logging.getLogger(__name__)
@dataclass @dataclass
class Options: class Options:
@ -112,7 +115,7 @@ async def fetch_messages_headers(imap_client: aioimaplib.IMAP4_SSL, max_uid: int
message_headers = BytesHeaderParser().parsebytes(response.lines[i + 1]) message_headers = BytesHeaderParser().parsebytes(response.lines[i + 1])
new_max_uid = uid new_max_uid = uid
else: else:
print("error %s" % response) logger.error("error %s" % response)
return new_max_uid, message_headers return new_max_uid, message_headers
@ -140,7 +143,7 @@ async def wait_for_new_message(imap_client, options: Options):
) )
while True: while True:
print("Waiting for new message") logger.debug("waiting for new message")
idle_task = await imap_client.idle_start(timeout=60) idle_task = await imap_client.idle_start(timeout=60)
msg = await imap_client.wait_server_push() msg = await imap_client.wait_server_push()
@ -176,7 +179,7 @@ async def wait_for_new_message(imap_client, options: Options):
filename += ".pdf" filename += ".pdf"
outpath = options.kindle_dir / filename outpath = options.kindle_dir / filename
print(f"Downloading '{doc_title}' -> '{outpath}'") logger.info(f"downloading '{doc_title}' -> '{outpath}'")
urllib.request.urlretrieve(link, outpath) urllib.request.urlretrieve(link, outpath)
shutil.copy(outpath, options.latest_path) shutil.copy(outpath, options.latest_path)
@ -202,7 +205,7 @@ async def make_client(host, user, password, folder):
return imap_client return imap_client
def parse_args(): def parse_args_and_configure_logging():
parser = argparse.ArgumentParser( parser = argparse.ArgumentParser(
prog="kindle_fetch", prog="kindle_fetch",
description="Monitors you Email and automatically downloads the notes sent to it.", description="Monitors you Email and automatically downloads the notes sent to it.",
@ -233,6 +236,11 @@ def parse_args():
help="the folder to monitor for new messages", help="the folder to monitor for new messages",
default="INBOX", default="INBOX",
) )
parser.add_argument(
"--loglevel",
default="info",
help="logging level; example --loglevel debug, default=warning",
)
args = parser.parse_args() args = parser.parse_args()
@ -241,6 +249,8 @@ def parse_args():
kindle_dir.mkdir(exist_ok=True, parents=True) kindle_dir.mkdir(exist_ok=True, parents=True)
latest_path = Path(kindle_dir / args.current_file).with_suffix(".pdf") latest_path = Path(kindle_dir / args.current_file).with_suffix(".pdf")
logging.basicConfig(level=args.loglevel.upper())
return Options( return Options(
server=args.server, server=args.server,
user=args.user, user=args.user,
@ -252,16 +262,18 @@ def parse_args():
def main(): def main():
options = parse_args() options = parse_args_and_configure_logging()
loop = asyncio.get_event_loop() loop = asyncio.get_event_loop()
logger.info("logging in")
try: try:
client = loop.run_until_complete( client = loop.run_until_complete(
make_client(options.server, options.user, options.password, options.mailbox) make_client(options.server, options.user, options.password, options.mailbox)
) )
except Exception as e: except Exception as e:
print(f"Failed to connect to the server: {e}") logger.error(f"Failed to connect to the server: {e}")
exit(1) exit(1)
logger.info("starting monitor")
loop.run_until_complete(wait_for_new_message(client, options)) loop.run_until_complete(wait_for_new_message(client, options))
loop.run_until_complete(client.logout()) loop.run_until_complete(client.logout())