94 lines
2.4 KiB
Python
94 lines
2.4 KiB
Python
"""
|
|
Phone calls and SMS messages
|
|
"""
|
|
from datetime import datetime
|
|
from pathlib import Path
|
|
from typing import NamedTuple, Iterator, Set, Tuple
|
|
|
|
import pytz
|
|
from lxml import etree # type: ignore
|
|
|
|
from .core.common import get_files
|
|
|
|
from my.config import smscalls as config
|
|
|
|
|
|
class Call(NamedTuple):
|
|
dt: datetime
|
|
duration_s: int
|
|
who: str
|
|
|
|
@property
|
|
def summary(self) -> str:
|
|
return f"talked with {self.who} for {self.duration_s} secs"
|
|
|
|
|
|
def _extract_calls(path: Path) -> Iterator[Call]:
|
|
tr = etree.parse(str(path))
|
|
for cxml in tr.findall('call'):
|
|
# TODO we've got local tz here, not sure if useful..
|
|
# ok, so readable date is local datetime, changing throughout the backup
|
|
yield Call(
|
|
dt=_parse_dt_ms(cxml.get('date')),
|
|
duration_s=int(cxml.get('duration')),
|
|
who=cxml.get('contact_name') # TODO number if contact is unavail??
|
|
# TODO type? must be missing/outgoing/incoming
|
|
)
|
|
|
|
|
|
def calls() -> Iterator[Call]:
|
|
files = get_files(config.export_path, glob='calls-*.xml')
|
|
|
|
# TODO always replacing with the latter is good, we get better contact names??
|
|
emitted: Set[datetime] = set()
|
|
for p in files:
|
|
for c in _extract_calls(p):
|
|
if c.dt in emitted:
|
|
continue
|
|
emitted.add(c.dt)
|
|
yield c
|
|
|
|
|
|
class Message(NamedTuple):
|
|
dt: datetime
|
|
who: str
|
|
message: str
|
|
phone_number: str
|
|
from_me: bool
|
|
|
|
|
|
def messages() -> Iterator[Message]:
|
|
files = get_files(config.export_path, glob='sms-*.xml')
|
|
|
|
emitted: Set[Tuple[datetime, str, bool]] = set()
|
|
for p in files:
|
|
for c in _extract_messages(p):
|
|
key = (c.dt, c.who, c.from_me)
|
|
if key in emitted:
|
|
continue
|
|
emitted.add(key)
|
|
yield c
|
|
|
|
|
|
def _extract_messages(path: Path) -> Iterator[Message]:
|
|
tr = etree.parse(str(path))
|
|
for mxml in tr.findall('sms'):
|
|
yield Message(
|
|
dt=_parse_dt_ms(mxml.get('date')),
|
|
who=mxml.get('contact_name'),
|
|
message=mxml.get('body'),
|
|
phone_number=mxml.get('address'),
|
|
from_me=mxml.get('type') == '2', # 1 is received message, 2 is sent message
|
|
)
|
|
|
|
def _parse_dt_ms(d: str) -> datetime:
|
|
return pytz.utc.localize(datetime.utcfromtimestamp(int(d) / 1000))
|
|
|
|
|
|
def stats():
|
|
from .core import stat
|
|
|
|
return {
|
|
**stat(calls),
|
|
**stat(messages),
|
|
}
|