graphql_service.py

import os
import requests
import exceptions

def mutate(mutation, variables):
  url=os.getenv('HYGRAPH_URL')
  token = os.getenv("HYGRAPH_TOKEN")
  headers = {"Authorization": f"Bearer {token}"}

  try:
    r = requests.post(url, json={"query": mutation, "variables": variables}, headers=headers)
    json_data = r.json()
    print("MUTATioN", mutation)
    print("vas: ---------", variables)
    print(json_data)
    r.raise_for_status()
    return json_data
  
  except requests.exceptions.HTTPError as error: 
    raise exceptions.CustomException("HTTPError error " + str(error))
  
  except requests.exceptions.ConnectionError as error:
    raise exceptions.CustomException("ConnectionError" + str(error))
  
  except requests.exceptions.Timeout as error:
    raise exceptions.CustomException("Timeout error" + str(error))
  
  except requests.exceptions.TooManyRedirects as error:
    raise exceptions.CustomException("TooManyRedirects error" + str(error))
  
  except requests.exceptions.RequestException as error:
    raise exceptions.CustomException("RequestException(catastrophic) error" + str(error))


def create_episode(audio_id, program_slug, release_date, audio_url, transcript = ""):
  mutation = """
    mutation upsertEpisode(
      $transcript: String,
      $programSlug: String,
      $releaseDate: Date,
      $audioUrl: String,
      $audioId: String
    ) {
      upsertEpisode(
        where:{ audioId: $audioId },
        upsert: {
          create: {
            transcript: $transcript
            program: {
              connect: {
                slug: $programSlug
              }
            }
            releaseDate: $releaseDate,
            audioUrl: $audioUrl,
            audioId: $audioId
          },
          update: {
            transcript: $transcript
            program: {
              connect: {
                slug: $programSlug
              }
            }
            releaseDate: $releaseDate,
            audioUrl: $audioUrl,
            audioId: $audioId
          }
      }) {
        id
        program {
          id
        }
      }
    }
  """

  variables = {
    "transcript": transcript,
    "programSlug": program_slug,
    "releaseDate": release_date,
    "audioUrl": audio_url,
    "audioId": audio_id
  }
  return mutate(mutation, variables)

def publish_episode(audio_id):
  mutation = """
    mutation publishEpisode($audioId: String) {
      publishEpisode(where: { audioId: $audioId }, to: PUBLISHED) {
        audioId
      }
    }
  """
  variables = {
    "audioId": audio_id
  }
  return mutate(mutation, variables)


def update_transcript(audio_id, transcript):
  mutation = """
    mutation UpdateEpisodeTranscript($audioId: String, $transcript: String){
      updateEpisode(
        where: { audioId: $audioId },
        data: {
          transcript: $transcript
        }) {
        id
      }
    }
  """
  variables = {
    "audioId": audio_id,
    "transcript": transcript
  }
  return mutate(mutation, variables)

def gql_create_episode_and_publish(audio_id, program_slug, release_date, audio_url, transcript = ""):
  try:
    create_episode(audio_id, program_slug, release_date, audio_url, transcript)
    publish_episode(audio_id)
  except Exception as error:
    raise exceptions.CustomException(str(error))

def gql_update_episode_transcript_and_publish(audio_id, transcript):
  try:
    update_transcript(audio_id, transcript)
    publish_episode(audio_id)
    print('gql_update_episode_transcript_and_publish')
  except Exception as error:
    print(error)
    raise exceptions.CustomException(str(error))