inoreader2readwise/main.py

94 lines
2.9 KiB
Python

import os
import time
import json
import requests
import logging
# Define the path to the persistent data store
DATA_STORE_PATH = "/data/last_update_time.txt"
# Set up logging
logging.basicConfig(level=logging.INFO)
class APIHandler:
def __init__(self, base_url, username, password):
self.base_url = base_url
self.auth = (username, password)
def get(self, endpoint, params=None):
response = requests.get(self.base_url + endpoint, params=params, auth=self.auth)
response.raise_for_status()
return response.json()
def post(self, endpoint, data=None):
response = requests.post(self.base_url + endpoint, data=json.dumps(data), auth=self.auth)
response.raise_for_status()
return response.status_code
def get_last_update_time():
with open(DATA_STORE_PATH, 'r') as file:
return file.read().strip()
def update_last_update_time(new_time):
with open(DATA_STORE_PATH, 'w') as file:
file.write(new_time)
def main():
# Get credentials from environment variables
inoreader = APIHandler(
"https://www.inoreader.com/reader/api/0/stream/contents",
os.getenv("INOREADER_USERNAME"),
os.getenv("INOREADER_PASSWORD")
)
readwise = APIHandler(
"https://readwise.io",
os.getenv("READWISE_USERNAME"),
os.getenv("READWISE_PASSWORD")
)
while True:
try:
last_annotation_time = get_last_update_time()
# Get annotations after the last update time
inoreader_response = inoreader.get(
"/user/-/state/com.google/annotated",
params={
"annotations": 1,
"n": 100,
#"ot": last_annotation_time or None
}
)
data = json.loads(inoreader_response)
all_annotations = []
for item in data["items"]:
annotations = item.get("annotations", [])
all_annotations.extend(annotations)
# Filter annotations
new_annotations = [annotation for annotation in all_annotations if annotation['added_on'] > last_annotation_time]
if new_annotations:
latest_added_on = max(annotation['added_on'] for annotation in new_annotations)
else:
latest_added_on = None
# Push annotations to Readwise
readwise.post("/api/annotations", data=annotations)
# Update the last annotation time
if latest_added_on:
update_last_update_time(latest_added_on)
else:
logging.info("No new annotations found")
# Wait for an hour
time.sleep(3600)
except Exception as e:
logging.error(f"An error occurred: {e}")
time.sleep(60) # Wait a minute before retrying
if __name__ == "__main__":
main()