generate_analog_map.py

#!/usr/bin/env python
'''Generate map.txt for analog stats parsing'''

import os
from urllib.parse import urlparse
import logging
import pwd
# pylint: disable=import-error
import psycopg2
from funcy import compose
from funcy import partial

LOGGER = logging.getLogger(__name__)
LOGGER.setLevel(logging.DEBUG)
LOGGER.addHandler(logging.StreamHandler())

DATABASE_HOST = os.environ['DATABASE_HOST']
DATABASE_PORT = os.environ['DATABASE_PORT']
DATABASE_NAME = os.environ['DATABASE_NAME']
DATABASE_USER = os.environ['DATABASE_USER']
DATABASE_PASSWORD = os.environ['DATABASE_PASSWORD']

DB_URL = 'postgresql://%s:%s@%s:%s/%s' % (DATABASE_USER,
                                          DATABASE_PASSWORD,
                                          DATABASE_HOST,
                                          DATABASE_PORT,
                                          DATABASE_NAME)

MAP_DIR = os.environ.get('ANALOG_MAP_DIR', '/home/httpd/homecu/stats')


def create_analog_map():
    '''Query DB and create analog map.txt'''
    if not os.path.exists(MAP_DIR):
        os.makedirs(MAP_DIR)
    map_filename = os.path.join(MAP_DIR, 'map.txt')
    with __create_conn__() as conn:
        compose(partial(__write_analog_map_rows__, map_filename),
                __query_cu_info__)(conn)


def __create_conn__():
    '''Create Database connection'''
    return psycopg2.connect(DB_URL)


def __query_cu_info__(conn):
    '''Query cuinfo for username and urls'''
    assert conn
    with conn.cursor() as cur:
        cur.execute('SELECT user_name, web_host FROM cuinfo;')
        return cur.fetchall()


def __write_analog_map_rows__(filename, rows):
    '''Write map.txt with provided rows'''
    with open(filename, 'w') as map_file:
        for (user_name, web_host) in rows:
            map_file.write('%s:%s %s' % (user_name.upper(),
                                         user_name.lower(),
                                         urlparse(web_host).hostname))
        # write EOF \n
        map_file.write(os.linesep)
    return filename

if __name__ == '__main__':
    UID = pwd.getpwnam('www-data').pw_uid
    os.seteuid(UID)
    LOGGER.info("Starting analog map.txt generation ...")
    create_analog_map()
    LOGGER.info("Done.")