File size: 1,082 Bytes
52bf3e0
 
1c5f2e5
52bf3e0
 
67442ec
 
52bf3e0
1c5f2e5
 
 
 
 
 
52bf3e0
39fe3d1
 
 
 
 
52bf3e0
 
 
 
 
 
 
6907194
88de1f4
 
 
 
 
 
 
 
 
 
1c5f2e5
88de1f4
 
 
1c5f2e5
88de1f4
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
"""Trigger Parquet Snapshot Glue job"""
import os
import logging
import boto3

AWS_ACCESS_KEY_ID = os.environ['AWS_ACCESS_KEY_ID']
AWS_SECRET_ACCESS_KEY = os.environ['AWS_SECRET_ACCESS_KEY']

logging.basicConfig(
    format='%(asctime)s - %(levelname)s - %(funcName)s - %(message)s',
    datefmt="%Y-%m-%d %H:%M:%S",
    level=logging.INFO
)

def get_client_connection():
    """
    Returns a client connection to the AWS Glue service.

    :return: AWS Glue client connection
    """
    return boto3.client(
        service_name='glue',
        region_name='us-east-1',
        aws_access_key_id=AWS_ACCESS_KEY_ID,
        aws_secret_access_key=AWS_SECRET_ACCESS_KEY
    )


def glue_job_run():
    """
    Triggers the Glue job run for the Parquet Snapshot.

    :return: None
    """
    glue = get_client_connection()
    response = glue.start_job_run(
        JobName='Article Snapshot China'
    )
    logging.info(response)
    response = glue.start_job_run(
        JobName='Reference China'
    )
    logging.info(response)

if __name__ == "__main__":
    glue_job_run()