Beispiel #1
0
class GoogleDocs:
    def __init__(self, client, cache_dir):
        self.client = client
        self.config = UserConfig(dir_name=cache_dir, config_file="cache", def_config_callback=lambda x: {})
        self.cache_dir = cache_dir
        if not self.config.read_config():
            self.config.initialize_dir(None)

    def getFolders(self):
        folders = self.get_list("/feeds/default/private/full/-/folder")
        if not folders.entry:
            return None
        return [Folder(self, x) for x in folders.entry]

    def getFolder(self, descriptor, etag=None):
        return Folder(self, self.client.GetResourceById(descriptor[id], etag=etag))

    def __cached_entry(self, id):
        return os.path.join(self.cache_dir, id)

    def get_list(self, url):
        feed = self.client.GetResources(uri=url)
        if not feed.entry:
            return None
        if feed.GetNextLink():
            feed.entry += self.get_list(feed.GetNextLink().href).entry
        return feed

    def get_cache_descriptor(self, id):
        if self.config.has_key(id):
            self.config[id]
        return None

    def open_cached_file(self, id, **kwargs):
        return open(self.__cached_entry(id), **kwargs)

    def download(self, id, extra_params=None):
        item_etag = None
        if self.config.has_key(id):
            item_etag = self.config[id][etag]
        entry = self.client.GetResourceById(id, etag=item_etag)
        self.client.DownloadResource(entry, self.__cached_entry(id), extra_params=extra_params)
        self.config[id] = create_descriptor(entry)
        self.config.write_config()

    def create(self, title, folder_entry, mime_type="text/plain"):
        ms = gdata.data.MediaSource(file_handle=StringIO(" "), content_type=mime_type, content_length=1)
        entry = gdata.docs.data.Resource(type="file", title=title)
        return self.client.CreateResource(entry, media=ms, collection=folder_entry)

    def write(self, entry, stream, length, mime_type="text/plain"):
        ms = gdata.data.MediaSource(file_handle=stream, content_type=mime_type, content_length=length)
        self.client.UpdateResource(entry, media=ms)
Beispiel #2
0
 def __init__(self, client, cache_dir):
     self.client = client
     self.config = UserConfig(dir_name=cache_dir, config_file="cache", def_config_callback=lambda x: {})
     self.cache_dir = cache_dir
     if not self.config.read_config():
         self.config.initialize_dir(None)