174 lines
4.2 KiB
Python
174 lines
4.2 KiB
Python
from typing import List, Dict, Union, Iterable, Iterator, NamedTuple, Any
|
|
import json
|
|
from collections import OrderedDict
|
|
from pathlib import Path
|
|
import pytz
|
|
import re
|
|
from datetime import datetime
|
|
|
|
from kython import kompress, cproperty
|
|
|
|
|
|
BPATH = Path("/L/backups/reddit")
|
|
|
|
|
|
def _get_backups(all_=True) -> List[Path]:
|
|
bfiles = list(sorted(BPATH.glob('reddit-*.json.xz')))
|
|
if all_:
|
|
return bfiles
|
|
else:
|
|
return bfiles[-1:]
|
|
|
|
|
|
class Save(NamedTuple):
|
|
dt: datetime
|
|
title: str
|
|
url: str
|
|
sid: str
|
|
json: Any = None
|
|
# TODO subreddit-display name?
|
|
|
|
def __hash__(self):
|
|
return hash(self.sid)
|
|
|
|
@cproperty
|
|
def text(self) -> str:
|
|
bb = self.json.get('body', None)
|
|
st = self.json.get('selftext', None)
|
|
if bb is not None and st is not None:
|
|
raise RuntimeError(f'wtf, both body and selftext are not None: {bb}; {st}')
|
|
return bb or st
|
|
|
|
@cproperty
|
|
def subreddit(self) -> str:
|
|
assert self.json is not None
|
|
# pylint: disable=unsubscriptable-object
|
|
return self.json['subreddit']['display_name']
|
|
|
|
|
|
class Misc(NamedTuple):
|
|
pass
|
|
|
|
EventKind = Union[Save, Misc]
|
|
|
|
class Event(NamedTuple):
|
|
dt: datetime
|
|
text: str
|
|
kind: EventKind
|
|
eid: str
|
|
title: str
|
|
url: str
|
|
|
|
|
|
# TODO kython?
|
|
def get_some(d, *keys):
|
|
# TODO only one should be non None??
|
|
for k in keys:
|
|
v = d.get(k, None)
|
|
if v is not None:
|
|
return v
|
|
else:
|
|
return None
|
|
|
|
|
|
Url = str
|
|
|
|
# TODO OrderedDict
|
|
def get_state(bfile: Path) -> Dict[Url, Save]:
|
|
saves: Dict[Url, Save] = {}
|
|
with kompress.open(bfile) as fo:
|
|
jj = json.load(fo)
|
|
|
|
saved = jj['saved']
|
|
for s in saved:
|
|
dt = pytz.utc.localize(datetime.utcfromtimestamp(s['created_utc']))
|
|
url = get_some(s, 'link_permalink', 'url')
|
|
title = get_some(s, 'link_title', 'title')
|
|
save = Save(
|
|
dt=dt,
|
|
title=title,
|
|
url=url,
|
|
sid=s['id'],
|
|
json=s,
|
|
)
|
|
saves[save.url] = save
|
|
|
|
# "created_utc": 1535055017.0,
|
|
# link_title
|
|
# link_text
|
|
return OrderedDict(sorted(saves.items(), key=lambda p: p[1].dt))
|
|
|
|
|
|
def get_events(all_=True):
|
|
backups = _get_backups(all_=all_)
|
|
assert len(backups) > 0
|
|
|
|
events: List[Event] = []
|
|
prev_saves: Dict[str, Save] = {}
|
|
# TODO suppress first batch??
|
|
# TODO for initial batch, treat event time as creation time
|
|
|
|
RE = re.compile(r'reddit-(\d{14})')
|
|
for i, b in enumerate(backups): # TODO when date...
|
|
match = RE.search(b.stem)
|
|
assert match is not None
|
|
btime = pytz.utc.localize(datetime.strptime(match.group(1), "%Y%m%d%H%M%S"))
|
|
|
|
first = i == 0
|
|
saves = get_state(b)
|
|
|
|
def etime(dt: datetime):
|
|
if first:
|
|
return dt
|
|
else:
|
|
return btime
|
|
|
|
for l in set(prev_saves.keys()).symmetric_difference(set(saves.keys())):
|
|
if l in prev_saves:
|
|
s = prev_saves[l]
|
|
# TODO use backup date, that is more precise...
|
|
events.append(Event(
|
|
dt=etime(s.dt),
|
|
text=f"unfavorited",
|
|
kind=s,
|
|
eid=f'unf-{s.sid}',
|
|
url=s.url,
|
|
title=s.title,
|
|
))
|
|
else: # in saves
|
|
s = saves[l]
|
|
events.append(Event(
|
|
dt=etime(s.dt),
|
|
text=f"favorited {' [initial]' if first else ''}",
|
|
kind=s,
|
|
eid=f'fav-{s.sid}',
|
|
url=s.url,
|
|
title=s.title,
|
|
))
|
|
prev_saves = saves
|
|
|
|
return list(sorted(events, key=lambda e: e.dt))
|
|
|
|
def get_saves(all_=True) -> List[Save]:
|
|
# TODO hmm.... do we want ALL reddit saves I ever had?
|
|
# TODO for now even last ones would be ok
|
|
assert all_ is False, 'all saves are not supported yet...'
|
|
backups = _get_backups(all_=all_)
|
|
[backup] = backups
|
|
|
|
saves = get_state(backup)
|
|
return list(saves.values())
|
|
|
|
|
|
def test():
|
|
get_events(all_=False)
|
|
get_saves(all_=False)
|
|
|
|
|
|
def main():
|
|
for e in get_events():
|
|
print(e)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|