80 lines
2.3 KiB
Python
80 lines
2.3 KiB
Python
"""Google Reader/FreshRSS API code for Youtube2FreshRSS."""
|
|
|
|
import requests
|
|
|
|
|
|
class FreshRss:
|
|
"""FreshRSS greader API helper class."""
|
|
base_path = "/api/greader.php"
|
|
base_api_path = f"{base_path}/reader/api/0/"
|
|
|
|
def __init__(self, baseurl, username, password):
|
|
self.base_url = baseurl
|
|
self.username = username
|
|
self.password = password
|
|
self.auth_header = ""
|
|
|
|
def get_auth_token(self):
|
|
"Retrieve greader auth token"
|
|
res = requests.post(
|
|
f"{self.base_url}{self.base_path}/accounts/ClientLogin",
|
|
params={'Email': self.username, 'Passwd': self.password},
|
|
timeout=15
|
|
)
|
|
auth_token = res.text.split("=")[-1].strip()
|
|
self.auth_header = f"GoogleLogin auth={auth_token}"
|
|
|
|
def post(self, path, params=None, data=None):
|
|
"Make a POST request to a greader API."
|
|
res = requests.post(
|
|
f"{self.base_url}{self.base_api_path}{path}",
|
|
data=data,
|
|
params=params,
|
|
headers={"Authorization": self.auth_header},
|
|
timeout=15
|
|
)
|
|
return res.text
|
|
|
|
def get(self, path, params=None):
|
|
"""Make a GET request to a greader API."""
|
|
res = requests.get(
|
|
f"{self.base_url}{self.base_api_path}{path}?output=json",
|
|
params=params,
|
|
headers={"Authorization": self.auth_header},
|
|
timeout=15
|
|
)
|
|
return res.json()
|
|
|
|
def get_starred(self):
|
|
"""Retrieve starred items."""
|
|
return self.get("/stream/contents/user/-/state/com.google/starred")
|
|
|
|
def parse_items(self, items):
|
|
"""Parse URLs from a list of items."""
|
|
parsed = []
|
|
|
|
for item in items:
|
|
cat = ["rss"]
|
|
for category in item['categories']:
|
|
if category.startswith("user/-/label/"):
|
|
cat.append(category[13:])
|
|
|
|
parsed.append({
|
|
"id": item['id'],
|
|
"title": item['title'],
|
|
"url": item['canonical'][0]['href'],
|
|
"categories": ",".join(cat)
|
|
})
|
|
|
|
return parsed
|
|
|
|
def unstarr(self, item_id):
|
|
"""Unstar an item given its ID."""
|
|
self.post(
|
|
"/edit-tag",
|
|
data={
|
|
"i": item_id,
|
|
"r": "user/-/state/com.google/starred"
|
|
}
|
|
)
|