HPI/tweets/__init__.py
2018-10-02 20:25:23 +01:00

59 lines
1.6 KiB
Python

from datetime import date, datetime
from typing import Union, List
KARLICOSS_ID = '119756204'
class Tweet:
def __init__(self, tw):
self.tw = tw
def __getattr__(self, attr):
return getattr(self.tw, attr)
@property
def url(self) -> str:
from twidump.render.tools import make_tweet_permalink
return make_tweet_permalink(self.tw.id_str)
@property
def time(self) -> str:
return self.tw.created_at
@property
def dt(self) -> datetime:
return self.tw.get_utc_datetime()
@property
def text(self) -> str:
return self.tw.text
def __str__(self) -> str:
return str(self.tw)
def __repr__(self) -> str:
return repr(self.tw)
def tweets_all():
import twidump
# add current package to path to discover config?... nah, twidump should be capable of that.
from twidump.data_manipulation.timelines import TimelineLoader
from twidump.component import get_app_injector
tl_loader = get_app_injector().get(TimelineLoader) # type: TimelineLoader
tl = tl_loader.load_timeline(KARLICOSS_ID)
return [Tweet(x) for x in tl]
def predicate(p) -> List[Tweet]:
return [t for t in tweets_all() if p(t)]
def predicate_date(p) -> List[Tweet]:
return predicate(lambda t: p(t.dt.date()))
Datish = Union[date, str]
def tweets_on(*dts: Datish) -> List[Tweet]:
from kython import parse_date_new
# TODO how to make sure we don't miss on 29 feb?
dates = {parse_date_new(d) for d in dts}
return predicate_date(lambda d: d in dates)
on = tweets_on