def Upload_To_GDrive():
global File_Name
gauth = GoogleAuth()
gauth.LoadCredentialsFile("mycreds.txt")
if gauth.credentials is None:
gauth.GetFlow()
gauth.flow.params.update({'access_type': 'offline'})
gauth.flow.params.update({'approval_prompt': 'force'})
gauth.LocalWebserverAuth()
elif gauth.access_token_expired:
try:
gauth.Refresh()
except Exception as e:
print(e)
else:
gauth.Authorize()
gauth.SaveCredentialsFile("mycreds.txt")
drive = GoogleDrive(gauth)
folder_id = [
'Folder_String_A', 'Folder_String_B', 'Folder_String_C',
'Folder_String_D'
]
for k in range(len(File_Name)):
try:
f = drive.CreateFile({
'title':
f'{File_Name[k]}.csv',
"parents": [{
"kind": "drive#fileLink",
"id": folder_id[k]
}]
})
f.SetContentFile(f'./Data/{File_Name[k]}.csv')
f.Upload()
print("Uploading succeeded!")
except:
print("Uploading failed.")
print(time.ctime())
class GoogledrivebackupPlugin(octoprint.plugin.SettingsPlugin,
octoprint.plugin.AssetPlugin,
octoprint.plugin.TemplatePlugin,
octoprint.plugin.EventHandlerPlugin,
octoprint.plugin.SimpleApiPlugin):
##~~ SettingsPlugin mixin
def __init__(self):
self.gauth = None
def get_settings_defaults(self):
return dict(
cert_saved=False,
cert_authorized=False,
installed_version=self._plugin_version,
)
##~~ SimpleApiPlugin mixin
def get_api_commands(self):
return dict(gen_secret=["json_data"], authorize=["auth_code"])
def on_api_command(self, command, data):
from octoprint.server import user_permission
import flask
if not user_permission.can():
return flask.make_response("Insufficient rights", 403)
from pydrive2.auth import GoogleAuth
config_file = "{}/client_secrets.json".format(
self.get_plugin_data_folder())
credentials_file = "{}/credentials.json".format(
self.get_plugin_data_folder())
if not self.gauth:
self.gauth = GoogleAuth()
if command == "gen_secret":
import json
# write out our client_secrets.json file
with open(config_file, "w") as f:
f.write(json.dumps(data["json_data"]))
self._settings.set(["cert_saved"], True)
self._settings.save()
self.gauth.LoadClientConfigFile(config_file)
self.gauth.GetFlow()
self.gauth.flow.params.update({'access_type': 'offline'})
self.gauth.flow.params.update({'approval_prompt': 'force'})
auth_url = self.gauth.GetAuthUrl()
return flask.jsonify(dict(cert_saved=True, url=auth_url))
if command == "authorize":
self._logger.info("Attempting to authorize Google App")
if not self.gauth:
return flask.jsonify(dict(authorized=False))
# Try to load saved client credentials
self.gauth.Auth(data["auth_code"])
self.gauth.SaveCredentialsFile(credentials_file)
self._settings.set(["cert_authorized"], True)
self._settings.save()
return flask.jsonify(dict(authorized=True))
##~~ AssetPlugin mixin
def get_assets(self):
# Define your plugin's asset files to automatically include in the
# core UI here.
return dict(js=["js/googledrivebackup.js"])
##~~ EventHandlerPlugin mixin
def on_event(self, event, payload):
if event == "plugin_backup_backup_created" and self._settings.get_boolean(
["cert_authorized"]):
self._logger.info(
"{} created, will now attempt to upload to Google Drive".
format(payload["path"]))
from pydrive2.drive import GoogleDrive
from pydrive2.auth import GoogleAuth
credentials_file = "{}/credentials.json".format(
self.get_plugin_data_folder())
gauth = GoogleAuth()
gauth.LoadCredentialsFile(credentials_file)
if gauth.credentials is None:
self._logger.error("not authorized")
self._settings.set(["cert_authorized"], False)
self._settings.save()
return
elif gauth.access_token_expired:
gauth.Refresh()
else:
gauth.Authorize()
gauth.SaveCredentialsFile(credentials_file)
drive = GoogleDrive(gauth)
f = drive.CreateFile({'title': payload["name"]})
f.SetContentFile(payload["path"])
f.Upload()
f = None
##~~ Softwareupdate hook
def get_update_information(self):
return dict(googledrivebackup=dict(
displayName="Google Drive Backup",
displayVersion=self._plugin_version,
# version check: github repository
type="github_release",
user="******",
repo="OctoPrint-GoogleDriveBackup",
current=self._plugin_version,
stable_branch=dict(
name="Stable", branch="master", comittish=["master"]),
prerelease_branches=[
dict(name="Release Candidate",
branch="rc",
comittish=["rc", "master"])
],
# update method: pip
pip=
"https://github.com/jneilliii/OctoPrint-GoogleDriveBackup/archive/{target_version}.zip"
))
