Python drive.GoogleDrive方法代碼示例

# 需要導入模塊: from pydrive import drive [as 別名]
# 或者: from pydrive.drive import GoogleDrive [as 別名]
def getDrive(drive=None, gauth=None):
    if not drive:
        if not gauth:
            gauth = GoogleAuth(settings_file=SETTINGS_YAML)
        # Try to load saved client credentials
        gauth.LoadCredentialsFile(CREDENTIALS)
        if gauth.access_token_expired:
            # Refresh them if expired
            try:
                gauth.Refresh()
            except RefreshError as e:
                log.error("Google Drive error: %s", e)
            except Exception as e:
                log.exception(e)
        else:
            # Initialize the saved creds
            gauth.Authorize()
        # Save the current credentials to a file
        return GoogleDrive(gauth)
    if drive.auth.access_token_expired:
        try:
            drive.auth.Refresh()
        except RefreshError as e:
            log.error("Google Drive error: %s", e)
    return drive 

# 需要導入模塊: from pydrive import drive [as 別名]
# 或者: from pydrive.drive import GoogleDrive [as 別名]
def __init__(self, *args, **kwargs):
        super(GDriveWriter, self).__init__(*args, **kwargs)
        from pydrive.auth import GoogleAuth
        from pydrive.drive import GoogleDrive
        gauth = GoogleAuth()
        files_tmp_path = tempfile.mkdtemp()
        client_secret_file = os.path.join(files_tmp_path, 'secret.json')
        with open(client_secret_file, 'w') as f:
            f.write(json.dumps(self.read_option('client_secret')))
        gauth.LoadClientConfigFile(client_secret_file)
        credentials_file = os.path.join(files_tmp_path, 'credentials.json')
        with open(credentials_file, 'w') as f:
            f.write(json.dumps(self.read_option('credentials')))
        gauth.LoadCredentialsFile(credentials_file)
        shutil.rmtree(files_tmp_path)
        self.drive = GoogleDrive(gauth)
        self.set_metadata('files_counter', Counter())
        self.set_metadata('files_written', []) 

# 需要導入模塊: from pydrive import drive [as 別名]
# 或者: from pydrive.drive import GoogleDrive [as 別名]
def cleanup(config):
        gauth = GoogleDriveActionBase.GoogleDriveActionBase.authenticate(config)
        drive = GoogleDrive(gauth)
        retain_from_date = datetime.today() - timedelta(days=int(config.config_obj.get('GoogleDriveUploadAction', 'file_retention_days')))
        file_list_len = 1
        logger.debug(drive.GetAbout())
        while file_list_len > 0:
            file_list = drive.ListFile({'q': "properties has { key='source' and value='MotionNotify' and visibility='PRIVATE'} and modifiedDate<'"
                    + retain_from_date.strftime("%Y-%m-%d") + "'"}).GetList()

            file_list_len = file_list.__len__()
            logger.info("GoogleDriveCleanAction - removing " + file_list_len.__str__() + " files")

            print(file_list.__len__())
            for file1 in file_list:
                logger.debug('GoogleDriveUploadAction Removing: title: %s, id: %s, createdDate: %s, parents: %s' % (file1['title'], file1['id'], file1['createdDate'], file1['parents']))
                file1.Delete() 

# 需要導入模塊: from pydrive import drive [as 別名]
# 或者: from pydrive.drive import GoogleDrive [as 別名]
def driveAuthorization():
    """Authorizes access to the Google Drive"""
    
    gauth = GoogleAuth()
    gauth.LocalWebserverAuth() # Creates local webserver and auto handles authentication.
    drive = GoogleDrive(gauth)
    return drive 

# 需要導入模塊: from pydrive import drive [as 別名]
# 或者: from pydrive.drive import GoogleDrive [as 別名]
def get_from_drive(idv):
  gauth = GoogleAuth()
  gauth.credentials = GoogleCredentials.get_application_default()
  drive = GoogleDrive(gauth)
  last_weight_file = drive.CreateFile({'id': idv}) 
  last_weight_file.GetContentFile('DenseNet-40-12-CIFAR10.h5') 

# 需要導入模塊: from pydrive import drive [as 別名]
# 或者: from pydrive.drive import GoogleDrive [as 別名]
def authorize_google_drive():
    global GAUTH
    global DRIVE
    if GAUTH is None or DRIVE is None:
        GAUTH = GoogleAuth()
        # Creates local webserver and auto handles authentication.
        GAUTH.LoadClientConfigFile(client_config_file=GRDIVE_CONFIG_PATH)
        GAUTH.LocalWebserverAuth()
        DRIVE = GoogleDrive(GAUTH)
        return True
    else:
        return True 

# 需要導入模塊: from pydrive import drive [as 別名]
# 或者: from pydrive.drive import GoogleDrive [as 別名]
def setup_drive(config):
        gauth = GoogleDriveActionBase.GoogleDriveActionBase.authenticate(config)
        return GoogleDrive(gauth) 

# 需要導入模塊: from pydrive import drive [as 別名]
# 或者: from pydrive.drive import GoogleDrive [as 別名]
def upload(file_path, description=None):
    try:
        gauth = GoogleAuth()
        # gauth.LocalWebserverAuth()

        # Try to load saved client credentials
        credential_file = os.getenv('GOOGLE_DRIVE_CREDENTIAL_FILE')
        gauth.LoadCredentialsFile(credential_file)
        if gauth.credentials is None:
            # Authenticate if they're not there
            gauth.LocalWebserverAuth()
        elif gauth.access_token_expired:
            # Refresh them if expired
            gauth.Refresh()
        else:
            # Initialize the saved creds
            gauth.Authorize()
        # end if

        # Save the current credentials to a file
        gauth.SaveCredentialsFile(credential_file)

        drive = GoogleDrive(gauth)
        folder_id = os.getenv('GOOGLE_DRIVE_FOLDER_ID')
        filename_w_ext = os.path.basename(file_path)
        filename, file_extension = os.path.splitext(filename_w_ext)

        # Upload file to folder
        f = drive.CreateFile(
            {"parents": [{"kind": "drive#fileLink", "id": folder_id}]})
        f['title'] = filename_w_ext

        # Make sure to add the path to the file to upload below.
        f.SetContentFile(file_path)
        f.Upload()

        logger.info(f['id'])
        return f['id']
    except Exception:
        logger.exception('Failed to upload %s', file_path)
    # end try
    return None
# end def