129 lines
3.7 KiB
Python
129 lines
3.7 KiB
Python
"""
|
|
Phone calls and SMS messages
|
|
Exported using https://play.google.com/store/apps/details?id=com.riteshsahu.SMSBackupRestore&hl=en_US
|
|
"""
|
|
|
|
REQUIRES = ['lxml']
|
|
|
|
from .core import Paths, dataclass
|
|
from my.config import smscalls as user_config
|
|
|
|
@dataclass
|
|
class smscalls(user_config):
|
|
# path[s] that SMSBackupRestore syncs XML files to
|
|
export_path: Paths
|
|
|
|
from .core.cfg import make_config
|
|
config = make_config(smscalls)
|
|
|
|
from datetime import datetime, timezone
|
|
from pathlib import Path
|
|
from typing import NamedTuple, Iterator, Set, Tuple, Optional
|
|
|
|
from lxml import etree # type: ignore
|
|
|
|
from my.core.common import get_files, Stats
|
|
from my.core.error import Res
|
|
|
|
|
|
class Call(NamedTuple):
|
|
dt: datetime
|
|
dt_readable: Optional[str]
|
|
duration_s: Optional[int]
|
|
who: Optional[str]
|
|
|
|
@property
|
|
def summary(self) -> str:
|
|
return f"talked with {self.who} for {self.duration_s} secs"
|
|
|
|
|
|
def _extract_calls(path: Path) -> Iterator[Res[Call]]:
|
|
tr = etree.parse(str(path))
|
|
for cxml in tr.findall('call'):
|
|
date_str = cxml.get('date')
|
|
if date_str is None:
|
|
yield RuntimeError(f"no date in {etree.tostring(cxml).decode('utf-8')}")
|
|
continue
|
|
duration = cxml.get('duration')
|
|
# 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(date_str),
|
|
dt_readable=cxml.get('readable_date'),
|
|
duration_s=int(duration) if duration is not None else None,
|
|
who=cxml.get('contact_name') # TODO number if contact is unavail??
|
|
# TODO type? must be missing/outgoing/incoming
|
|
)
|
|
|
|
|
|
def calls() -> Iterator[Res[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 isinstance(c, Exception):
|
|
yield c
|
|
continue
|
|
if c.dt in emitted:
|
|
continue
|
|
emitted.add(c.dt)
|
|
yield c
|
|
|
|
|
|
class Message(NamedTuple):
|
|
dt: datetime
|
|
dt_readable: Optional[str]
|
|
who: Optional[str]
|
|
message: Optional[str]
|
|
phone_number: Optional[str]
|
|
from_me: bool
|
|
|
|
|
|
def messages() -> Iterator[Res[Message]]:
|
|
files = get_files(config.export_path, glob='sms-*.xml')
|
|
|
|
emitted: Set[Tuple[datetime, Optional[str], Optional[bool]]] = set()
|
|
for p in files:
|
|
for c in _extract_messages(p):
|
|
if isinstance(c, Exception):
|
|
yield c
|
|
continue
|
|
key = (c.dt, c.who, c.from_me)
|
|
if key in emitted:
|
|
continue
|
|
emitted.add(key)
|
|
yield c
|
|
|
|
|
|
def _extract_messages(path: Path) -> Iterator[Res[Message]]:
|
|
tr = etree.parse(str(path))
|
|
for mxml in tr.findall('sms'):
|
|
date_str = mxml.get('date')
|
|
if date_str is None:
|
|
yield RuntimeError(f"no date in {etree.tostring(mxml).decode('utf-8')}")
|
|
continue
|
|
yield Message(
|
|
dt=_parse_dt_ms(date_str),
|
|
dt_readable=mxml.get('readable_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
|
|
)
|
|
|
|
|
|
# See https://github.com/karlicoss/HPI/pull/90#issuecomment-702422351
|
|
# for potentially parsing timezone from the readable_date
|
|
def _parse_dt_ms(d: str) -> datetime:
|
|
return datetime.fromtimestamp(int(d) / 1000, tz=timezone.utc)
|
|
|
|
|
|
def stats() -> Stats:
|
|
from .core import stat
|
|
|
|
return {
|
|
**stat(calls),
|
|
**stat(messages),
|
|
}
|