fetch_from_messages.py

#!/usr/bin/env python3

"""Fetch all famly.co pictures of your kid

Auth has two versions:

 - "non-v2" has a `?accessToken=XXX` as a GET-parameter
 - v2-urls demands a `x-famly-accesstoken: XXX` header

"""

import argparse
import json
import shutil
import urllib.request


class FamlyClient:
    _access_token = None

    def __init__(self, email, password):
        self._base = "https://app.famly.co"

        login_data = self.login(email, password)
        self._access_token = login_data["accessToken"]

    def _request_json(self, method, url, body=None):
        """Get some JSON and hope we get some back"""
        b = None
        if body:
            b = json.dumps(body).encode("utf-8")

        headers = {"Content-Type": "application/json"}
        if self._access_token:
            headers["x-famly-accesstoken"] = self._access_token

        req = urllib.request.Request(url=url, headers=headers, method=method, data=b)

        with urllib.request.urlopen(req) as f:
            body = f.read().decode("utf-8")
            if f.status != 200:
                raise "B0rked! %" % body

            try:
                return json.loads(body)
            except Exception as e:
                return body

    def _auth_request(self, method, path, body=None, request_params=None):
        # This should perhaps be split in two functions, one for "classic"
        # requests and auth, and then one for the "v2"-stuff...
        if not request_params:
            request_params = {}

        if "v2" not in path:
            request_params["accessToken"] = self._access_token

        # Serialize request-params
        key_values = ["%s=%s" % (k, v) for k, v in request_params.items()]

        return self._request_json(
            method, self._base + path + "?" + ("&".join(key_values)), body
        )

    def login(self, email, password):
        """Get an access token"""
        return self._request_json(
            "POST",
            self._base + "/api/login/login/authenticate",
            body={"email": email, "password": password},
        )

    def me_me_me(self):
        """Know about thy self..."""
        return self._auth_request("GET", "/api/me/me/me")


if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="Fetch kids' images from famly.co")
    parser.add_argument("email", help="Auth email")
    parser.add_argument("password", help="Auth password")
    args = parser.parse_args()
    client = FamlyClient(args.email, args.password)

    my_info = client.me_me_me()

    conversationsIds = client._auth_request("GET", "/api/v2/conversations")

    img_number = 1

    for conv_id in conversationsIds:
        conversation = client._auth_request(
            "GET", "/api/v2/conversations/%s" % (conv_id["conversationId"])
        )

        for msg in conversation["messages"]:
            for img in msg["images"]:
                url = "%s/%sx%s/%s" % (
                    img["prefix"],
                    img["height"],
                    img["width"],
                    img["key"],
                )
                print("%s" % (url))
                req = urllib.request.Request(url=url)

                with urllib.request.urlopen(req) as r, open(
                    "%i.jpg" % (img_number), "wb"
                ) as f:
                    if r.status != 200:
                        raise "B0rked! %s" % body
                    shutil.copyfileobj(r, f)
                img_number += 1