diff --git a/cps/gdriveutils.py b/cps/gdriveutils.py index 3f29d28a..9ec83059 100644 --- a/cps/gdriveutils.py +++ b/cps/gdriveutils.py @@ -238,6 +238,32 @@ def copyToDrive(drive, uploadFile, createRoot, replaceFiles, driveFile.SetContentFile(os.path.join(prevDir,uploadFile)) driveFile.Upload() +def uploadFileToEbooksFolder(drive, destFile, f): + if not drive: + drive=getDrive() + if drive.auth.access_token_expired: + drive.auth.Refresh() + parent=getEbooksFolder(drive) + splitDir=destFile.split('/') + for i, x in enumerate(splitDir): + if i == len(splitDir)-1: + existingFiles=drive.ListFile({'q' : "title = '%s' and '%s' in parents and trashed = false" % (x, parent['id'])}).GetList() + if len(existingFiles) > 0: + driveFile=existingFiles[0] + else: + driveFile = drive.CreateFile({'title': x, 'parents' : [{"kind": "drive#fileLink", 'id' : parent['id']}], }) + driveFile.SetContentFile(f) + driveFile.Upload() + else: + existingFolder=drive.ListFile({'q' : "title = '%s' and '%s' in parents and trashed = false" % (x, parent['id'])}).GetList() + if len(existingFolder) == 0: + parent = drive.CreateFile({'title': x, 'parents' : [{"kind": "drive#fileLink", 'id' : parent['id']}], + "mimeType": "application/vnd.google-apps.folder" }) + parent.Upload() + else: + parent=existingFolder[0] + + def watchChange(drive, channel_id, channel_type, channel_address, channel_token=None, expiration=None): if not drive: diff --git a/cps/web.py b/cps/web.py index be317349..d6019c18 100755 --- a/cps/web.py +++ b/cps/web.py @@ -2382,9 +2382,17 @@ def edit_book(book_id): if to_save["cover_url"] and os.path.splitext(to_save["cover_url"])[1].lower() == ".jpg": img = requests.get(to_save["cover_url"]) - f = open(os.path.join(config.config_calibre_dir, book.path, "cover.jpg"), "wb") - f.write(img.content) - f.close() + if config.config_use_google_drive: + tmpDir=tempfile.gettempdir() + f = open(os.path.join(tmpDir, "uploaded_cover.jpg"), "wb") + f.write(img.content) + f.close() + gdriveutils.uploadFileToEbooksFolder(Gdrive.Instance().drive, os.path.join(book.path, 'cover.jpg'), os.path.join(tmpDir, f.name)) + else: + f = open(os.path.join(config.config_calibre_dir, book.path, "cover.jpg"), "wb") + f.write(img.content) + f.close() + book.has_cover=1 if book.series_index != to_save["series_index"]: book.series_index = to_save["series_index"]