score_repo.py

#!/usr/bin/env python3

import argparse
from lib.core import establish_database_connection
from lib.core import init_attribute_plugins
from lib.core import load_attribute_plugins
from lib.core import process_configuration
from lib.core import process_repository
from lib.core import save_result
from lib.core import get_run_id
import os
import random
import sys
import threading
import time
from lib.utilities import is_dir


def process_arguments():
    """
    Uses the argparse module to parse commandline arguments.

    Returns:
        Dictionary of parsed commandline arguments.
    """
    parser = argparse.ArgumentParser(
        description='Calculate the score of a repository.'
    )
    parser.add_argument(
        '-c',
        '--config',
        type=argparse.FileType('r'),
        default='config.json',
        dest='config_file',
        help='Path to the configuration file.'
    )
    parser.add_argument(
        '-p',
        '--plugins',
        type=is_dir,
        default='attributes',
        dest='plugins_dir',
        help='Path to the folder containing your set of attribute plugins.'
    )
    parser.add_argument(
        'repository_id',
        type=int,
        nargs=1,
        help='Identifier for a project as it appears in the \
              GHTorrent database.'
    )
    parser.add_argument(
        'repository_path',
        type=is_dir,
        nargs=1,
        help='Path to the repository source code.'
    )

    if len(sys.argv) < 2:
        parser.print_help()
        sys.exit(1)

    return parser.parse_args()


def main():
    """
    Main execution flow.
    """
    args = process_arguments()
    config = process_configuration(args.config_file)
    connection = establish_database_connection(config['options']['datasource'])
    attributes = config['attributes']
    load_attribute_plugins(args.plugins_dir, attributes)
    init_attribute_plugins(attributes, connection)

    score, results = process_repository(
        args.repository_id[0],
        args.repository_path[0],
        attributes,
        connection
    )

    if config['options'].get('persistResult', False):
        cursor = connection.cursor()

        run_id = get_run_id()
        save_result(run_id, args.repository_id, results, cursor)

        cursor.close()

        print('\rResult saved to datasource.')
    else:
        print('\rRaw score: {0}'.format(score))

        if 'DEBUG' in os.environ:
            print(results)

    connection.close()


def spin(stop_condition):
    while not stop_condition:
        print(chr(random.randint(int('2800', 16), int('2880', 16))), end='')
        sys.stdout.flush()
        print('\b', end='')
        time.sleep(0.1)

if __name__ == '__main__':
    stop_condition = False
    spinner_thread = threading.Thread(target=spin, args=([stop_condition]))
    spinner_thread.daemon = True
    spinner_thread.start()
    try:
        main()
    except KeyboardInterrupt:
        print('\rCaught interrupt, exiting.')
    finally:
        stop_condition = True