graphql.py

import json
import sys

import requests

def exit_plugin(msg=None, err=None):
    if msg is None and err is None:
        msg = "plugin ended"
    output_json = {"output": msg, "error": err}
    print(json.dumps(output_json))
    sys.exit()

def doRequest(query, variables=None, port=9999, session=None, scheme="http", raise_exception=True):
    # Session cookie for authentication
    graphql_port = port
    graphql_scheme = scheme
    graphql_cookies = {
        'session': session
    }

    graphql_headers = {
        "Accept-Encoding": "gzip, deflate",
        "Content-Type": "application/json",
        "Accept": "application/json",
        "Connection": "keep-alive",
        "DNT": "1"
    }
    graphql_domain = 'localhost'
    # Stash GraphQL endpoint
    graphql_url = graphql_scheme + "://" + graphql_domain + ":" + str(graphql_port) + "/graphql"

    json = {'query': query}
    if variables is not None:
        json['variables'] = variables
    try:
        response = requests.post(graphql_url, json=json,headers=graphql_headers, cookies=graphql_cookies, timeout=20)
    except Exception as e:
         exit_plugin(err=f"[FATAL] Exception with GraphQL request. {e}")
    if response.status_code == 200:
        result = response.json()
        if result.get("error"):
            for error in result["error"]["errors"]:
                if raise_exception:
                    raise Exception(f"GraphQL error: {error}")
                else:
                    log.LogError(f"GraphQL error: {error}")
            return None
        if result.get("data"):
            return result.get("data")
    elif response.status_code == 401:
        exit_plugin(err="HTTP Error 401, Unauthorised.")
    else:
        raise ConnectionError(f"GraphQL query failed: {response.status_code} - {response.content}")

def update_scene_title(scene_id, scene_title, port, session, scheme):
    query = """
    mutation UpdateSceneTitle($id: ID!, $title: String) {
            sceneUpdate(
                input: {id: $id, title: $title}
                ) {
                    title
                  }
    }
    """
    variables = {
        "id": scene_id,
        "title": scene_title
    }
    result = doRequest(query=query, variables=variables, port=port, session=session, scheme=scheme)
    return result.get('sceneUpdate')

def get_scene_base(scene_id, port, session, scheme):
    query = """
    query FindScene($id: ID!, $checksum: String) {
        findScene(id: $id, checksum: $checksum) {
            files {
                basename
            }
        }
    }
    """
    variables = {
        "id": scene_id
    }
    result = doRequest(query=query, variables=variables, port=port, session=session, scheme=scheme)
    return result.get('findScene')

def get_api_version(port, session, scheme):
    query = """
    query SystemStatus {
        systemStatus {
            databaseSchema
            appSchema
        }
    }
    """
    result = doRequest(query=query, port=port, session=session, scheme=scheme)
    return result.get('systemStatus')