HPI/my/pocket.py

66 lines
1.4 KiB
Python

"""
[[https://getpocket.com][Pocket]] bookmarks and highlights
"""
REQUIRES = [
'git+https://github.com/karlicoss/pockexport',
]
from dataclasses import dataclass
from typing import Optional
from .core import Paths
from my.config import pocket as user_config
@dataclass
class pocket(user_config):
'''
Uses [[https://github.com/karlicoss/pockexport][pockexport]] outputs
'''
# paths[s]/glob to the exported JSON data
export_path: Paths
from .core.cfg import make_config
config = make_config(pocket)
try:
from pockexport import dal
except ModuleNotFoundError as e:
from .core.compat import pre_pip_dal_handler
dal = pre_pip_dal_handler('pockexport', e, config, requires=REQUIRES)
############################
Article = dal.Article
from pathlib import Path
from typing import Sequence, Iterable
# todo not sure if should be defensive against empty?
def _dal() -> dal.DAL:
from .core import get_files
inputs = get_files(config.export_path)
return dal.DAL(inputs)
def articles() -> Iterable[Article]:
yield from _dal().articles()
def stats():
from itertools import chain
from more_itertools import ilen
# todo make stats more defensive?
return {
'articles' : ilen(articles()),
'highlights': ilen(chain.from_iterable(a.highlights for a in articles())),
}
# todo deprecate?
def get_articles() -> Sequence[Article]:
return list(articles())