Function share_task

coreplugins/lightning/api.py:93–275

(task_name, project_name, project, cloud_token, cloud_url, resources, resources_base_path, progress_callback, should_cancel)

        return Response({'size': get_size_bytes(resources)})

def share_task(task_name, project_name, project, cloud_token, cloud_url, resources, resources_base_path, progress_callback, should_cancel):
    import uuid
    import requests
    import os
    import time
    from zipstream.ng import ZipStream
    import logging
    import jwt
    logger = logging.getLogger('app.logger')

    CHUNK_SIZE = 8 * 1024 * 1024  # 8MB
    cloud_url = cloud_url.rstrip('/')
    session = requests.Session()
    session.headers.update({
        'Authorization': 'JWT {}'.format(cloud_token)
    })

    def check_refresh_token():
        nonlocal cloud_token

        try:
            meta = jwt.decode(cloud_token, None, False)
            exp = meta.get('exp', time.time())

            # Refresh token if less than 1 hour remaining to expiry
            if exp - time.time() < 60 * 60:
                res = session.post(cloud_url + '/api/token-auth/refresh/', json={'token': cloud_token})
                if res.status_code == 200:
                    j = res.json()
                    if 'token' in j:
                        cloud_token = j['token']
                        session.headers.update({
                            'Authorization': 'JWT {}'.format(cloud_token)
                        })
        except Exception as e:
            logger.warning(f"Cannot check refresh token: {str(e)}")

    check_refresh_token()
    cleanup = lambda: None

    # If project is None, create a new project on the remote
    if project is None:
        res = session.post(cloud_url + '/api/projects/', json={'name': project_name})
        if res.status_code != 201:
            logger.info(res.content.decode("utf-8"))
            return {'error': 'Failed to create Lightning project. Try again in a bit.'}
        
        project = res.json().get('id')

        def cleanup():
            try:
                session.delete(cloud_url + f'/api/projects/{project}')
            except Exception as e:
                logger.warning(f"Cannot cleanup project: {str(e)}")

    dzuuid = str(uuid.uuid4())
    zs = ZipStream(sized=True)
    zs.comment = "Generated by WebODM Lightning Plugin"