AirFrame: Hacking together a WiFi photo frame with a Toshiba FlashAir SD card and Flickr

class Flickr(object):

    def __init__(self):
        self.set_keys(*self.read_keys())
        self.get_auth2()

    def read_keys(self):
        """
            Read the flickr API key and secret from a local file
        """
        with open("flickr_api.yaml") as f:
            api = yaml.load(f)
        return (api["key"], api["secret"])

    def set_keys(self, key, secret):
        self.api_key = key
        self.api_secret = secret

    def get_auth2(self):
        print("Authenticating to Flickr")
        self.flickr = flickrapi.FlickrAPI(self.api_key, self.api_secret)
        (token,frob) = self.flickr.get_token_part_one(perms='read')
        if not token: raw_input("Press ENTER after you authorized this program")
        self.flickr.get_token_part_two((token,frob))
        print("Authentication succeeded")

def get_tagged(self, tags, count, download_dir="photos"):
    """ Get photos with the given list of tags
    """
    print ("connecting to flickr, and getting %d photos with tags %s" % (count, tags))
    x = self.flickr.photos_search(api_key = self.api_key, user_id="me", tags=','.join(tags), per_page=count)
    photos = self._extract_photos_from_xml(x)
    photo_filenames = self._sync_photos(photos, download_dir)
    print("Found %d photos" % len(photos))
    return photo_filenames

def _extract_photos_from_xml(self, xml):
    photos = []
    for i in xml.iter():
        if i.tag == 'rsp':
            # the response header.  stat member should be 'ok'
            if i.get('stat') == 'ok':
                continue
            else:
                # error, so just break
                break
        if i.tag == 'photo':
            photos.append(Photo(i))
    return photos

<photos page="2" pages="89" perpage="10" total="881">
    <photo id="2636" owner="47058503995@N01"
        secret="a123456" server="2" title="test_04"
        ispublic="1" isfriend="0" isfamily="0" />
    <photo id="2635" owner="47058503995@N01"
        secret="b123456" server="2" title="test_03"
        ispublic="0" isfriend="1" isfamily="1" />
    <photo id="2633" owner="47058503995@N01"
        secret="c123456" server="2" title="test_01"
        ispublic="1" isfriend="0" isfamily="0" />
    <photo id="2610" owner="12037949754@N01"
        secret="d123456" server="2" title="00_tall"
        ispublic="1" isfriend="0" isfamily="0" />
</photos>

class Photo(object):
    def __init__(self, photo_element):
        """Construct a photo object out of the XML response from Flickr"""
        attrs = { 'farm': 'farmid', 'server':'serverid','id':'photoid','secret':'secret'}
        for flickr_attr, py_attr in attrs.items():
            setattr(self, py_attr, photo_element.get(flickr_attr))

    def _construct_flickr_url(self):
        url = "http://farm%s.staticflickr.com/%s/%s_%s_b.jpg" % (self.farmid,self.serverid, self.photoid, self.secret)
        return url

    def download_photo(self, dirname, cache=False, tgt_filename=None):
        if not os.path.exists(dirname):
            os.makedirs(dirname)
        tgt = os.path.join(dirname, "%s.jpg" % self.photoid)
        if cache:
            if os.path.isfile(tgt):
                return tgt
        urllib.urlretrieve(self._construct_flickr_url(), tgt)
        return tgt

def _set_write_protect(self):
    # Change the upload directory on the card
    url = "http://%s/upload.cgi" % self.hostname

    payload = {'WRITEPROTECT': "ON"}
    r = requests.get(url, params=payload)
    r.raise_for_status()
    if not "SUCCESS" in r.content:
        print("Could not put card into host-write-protect mode")

def get_file_list(self):
     payload = {"op":100, "DIR":self.card_path}
     r = requests.get("http://%s/command.cgi" % self.hostname, params=payload)
     r.raise_for_status()                                                           print r.text
     # Divide the returned text by newline, and ignore the first line "WLANSD_FILELIST"
     lines = r.text.split('\n')
     assert lines[0].strip()=='WLANSD_FILELIST'

     filenames = []
     for line in lines[1:]:
         # Split each line by ',' and take the second column for the filename
         values = line.split(',')
         if len(values) > 2:
             filename = values[1].strip()
             filenames.append(filename)

     print filenames
     return filenames

def _get_renamed_filename(self, filename):
        # First, separate out the path from the filename
        path = os.path.dirname(filename)
        fn = os.path.basename(filename)
        h = hashlib.sha1()
        h.update(fn)
        hash_filename = "%s.JPG" % (h.hexdigest()[:8].upper())

        hash_full_filename = os.path.join(path, hash_filename)
        logging.debug("Hashed file: %s" % (hash_full_filename))
        return hash_full_filename

def set_timestamp(self, t):
        """
            :param t: The time stamp
            :type t: time.struct_time
        """
        fat32_time = ((t.tm_year-1980)<<25) | (t.tm_mon << 21) | (t.tm_mday << 16) | (t.tm_hour << 11) | (t.tm_min << 5) | (t.tm_sec >>1)

        url = "http://%s/upload.cgi" % self.hostname
        payload = {'FTIME': "0x%0.8X" % fat32_time}
        r = requests.get(url, params=payload)
        r.raise_for_status()

def upload_file(self, filename):
     # First, set the time of the photo to right now
     self.set_timestamp(time.localtime())
     # Change the upload directory on the card
     url = "http://%s/upload.cgi" % self.hostname

     payload = {'UPDIR':self.card_path}
     r = requests.get(url, params=payload)
     r.raise_for_status()

     # Now, upload the file
     hash_full_filename = self._get_renamed_filename(filename)
     hash_filename = os.path.basename(hash_full_filename)

     files = {'file':(hash_filename, open(filename,'rb'))}
     r = requests.post(url, files=files)
     r.raise_for_status()