HPI/my/hypothesis.py
2019-11-13 00:03:19 +00:00

71 lines
1.5 KiB
Python

from pathlib import Path
from my_configuration import paths
import my_configuration.repos.hypexport.model as hypexport
def get_model() -> hypexport.Model:
export_dir = Path(paths.hypexport.export_dir)
sources = list(sorted(export_dir.glob('*.json')))
model = hypexport.Model(sources)
return model
Highlight = hypexport.Highlight
from typing import Dict, List, NamedTuple, Optional, Sequence
from pathlib import Path
from datetime import datetime
from .common import group_by_key, the, cproperty
class Page(NamedTuple):
"""
Represents annotated page along with the highlights
"""
highlights: Sequence[Highlight]
@cproperty
def link(self):
return the(h.page_link for h in self.highlights)
@cproperty
def title(self):
return the(h.page_title for h in self.highlights)
@cproperty
def dt(self) -> datetime:
return min(h.dt for h in self.highlights)
def _iter():
yield from get_model().iter_highlights()
def get_pages() -> List[Page]:
grouped = group_by_key(_iter(), key=lambda e: e.page_link)
pages = []
for link, group in grouped.items():
sgroup = tuple(sorted(group, key=lambda e: e.dt))
pages.append(Page(highlights=sgroup))
pages = list(sorted(pages, key=lambda p: p.dt))
# TODO fixme page tag??
return pages
def get_highlights():
return list(sorted(_iter(), key=lambda h: h.dt))
def test():
get_pages()
get_highlights()
def _main():
for page in get_pages():
print(page)
if __name__ == '__main__':
_main()