This commit is contained in:
Swapnil 2024-01-17 17:31:17 +05:30
parent 04dae288ac
commit 0da2a87caf
Signed by: swapnil
GPG Key ID: 58029C48BB100574

88
main.py
View File

@ -4,10 +4,8 @@ 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:
@ -33,14 +31,34 @@ 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
def get_new_annotations(last_annotation_time):
inoreader = APIHandler(
"https://www.inoreader.com/reader/api/0/stream/contents",
headers = {
'Authorization': 'Bearer ' + os.getenv("INOREADER_ACCESS_TOKEN")
})
}
)
inoreader_response = inoreader.get(
"/user/-/state/com.google/annotated",
params={
"annotations": 1,
"n": 100,
}
)
data = json.loads(inoreader_response)
all_annotations = []
for item in data["items"]:
annotations = item.get("annotations", [])
for annotation in annotations:
annotation['title'] = item['title']
annotation['author'] = item['author']
all_annotations.append(annotation)
return [annotation for annotation in all_annotations if annotation['added_on'] > last_annotation_time]
def push_annotations_to_readwise(annotations):
readwise = APIHandler(
"https://readwise.io",
headers = {
@ -49,74 +67,40 @@ def main():
}
)
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
# group annotations by title and author
grouped_annotations = {}
for annotation in new_annotations:
title = annotation['title']
author = annotation['author']
key = (title, author)
if key not in grouped_annotations:
grouped_annotations[key] = []
grouped_annotations[key].append(annotation)
# Push annotations to Readwise
readwise.post(
"/api/v2/highlights/",
# convert the grouped_annotations above for readwise payload
data={
'highlights': [
{
'text': annotation['text'],
'title': key[0],
'author': key[1],
'title': annotation['title'],
'author': annotation['author'],
'note': annotation['note'],
}
for key, annotations in grouped_annotations.items()
for annotation in annotations
]
}
)
# Update the last annotation time
if latest_added_on:
def main():
while True:
try:
last_annotation_time = get_last_update_time()
new_annotations = get_new_annotations(last_annotation_time)
if new_annotations:
latest_added_on = max(annotation['added_on'] for annotation in new_annotations)
push_annotations_to_readwise(new_annotations)
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
time.sleep(60)
if __name__ == "__main__":
main()