circleci-failure-tracker/lambda/refresh-grid-view/generate_credentials_module.py at master · kostmo/circleci-failure-tracker · GitHub

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
#!/usr/bin/env python3

import os, sys
import json


THIS_SCRIPT_DIR = os.path.dirname(__file__)
CREDENTIALS_DIR = os.path.join(THIS_SCRIPT_DIR, "../../../circleci-failure-tracker-credentials")


def write_creds_module(source_filename, destination_filename):

    json_source_path = os.path.join(CREDENTIALS_DIR, source_filename)
    github_token_path = os.path.join(CREDENTIALS_DIR, "github-personal-access-token-repo-read-permissions.txt")
    sqs_queue_url_path = os.path.join(CREDENTIALS_DIR, "aws-sqs-queue-url.txt")

    dest_path = os.path.join("dr_ci_view_refresh", destination_filename)

    with open(json_source_path) as json_fh, open(dest_path, "w") as output_module_fh:

        creds_dict = json.load(json_fh)

        output_module_fh.write('# This file is autogenerated!\n')
        output_module_fh.write('db_hostname = "%s"\n' % creds_dict["db-hostname"])
        output_module_fh.write('db_username = "%s"\n' % creds_dict["db-user"])
        output_module_fh.write('db_password = "%s"\n' % creds_dict["db-password"])
        output_module_fh.write('db_name = "%s"\n' % "loganci")

        with open(github_token_path) as token_fh:
            access_token = token_fh.read().strip()
            output_module_fh.write('repo_read_auth_token = "%s"\n' % access_token)


        with open(sqs_queue_url_path) as fh:
            queue_url = fh.read().strip()
            output_module_fh.write('sqs_queue_url = "%s"\n' % queue_url)


if __name__ == "__main__":

    write_creds_module("database-credentials-remote-mview-refresher.json", "db_config.py")
    write_creds_module("database-credentials-remote.json", "logan_db_config.py")