simplify provider

This commit is contained in:
Dima Gerasimov 2019-05-08 20:54:37 +01:00
parent 76814f7f06
commit 351528bb64
3 changed files with 30 additions and 18 deletions

23
lastfm/__init__.py Normal file → Executable file
View file

@ -1,12 +1,14 @@
#!/usr/bin/env python3
from functools import lru_cache
from kython import listdir_abs, json_load, JSONType
from typing import Dict, List, NamedTuple
from pytz import UTC
from datetime import datetime
from pathlib import Path
import os
import json
import pytz
_PATH = "/L/backups/lastfm"
_PATH = Path("/L/backups/lastfm")
class Scrobble(NamedTuple):
dt: datetime
@ -17,18 +19,21 @@ class Scrobble(NamedTuple):
# TODO watch out, if we keep the app running it might expire
def _iter_scrobbles():
last = max(listdir_abs(_PATH))
last = max(_PATH.glob('*.json'))
# TODO mm, no timezone? wonder if it's UTC...
j: List[Dict[str, str]]
with open(last, 'r') as fo:
j = json_load(fo)
j = json.loads(last.read_text())
for d in j:
ts = int(d['date'])
dt = datetime.fromtimestamp(ts, tz=UTC)
dt = datetime.fromtimestamp(ts, tz=pytz.utc)
track = f"{d['artist']}{d['name']}"
yield Scrobble(dt, track)
@lru_cache()
@lru_cache(1)
def get_scrobbles():
return list(_iter_scrobbles())
def test():
assert len(get_scrobbles()) > 1000