misp-dashboard/updates.py

import redis
import os
import configparser
import logging

DATABASE_VERSION = [
    1
]


configfile = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'config/config.cfg')
cfg = configparser.ConfigParser()
cfg.read(configfile)
serv_log = redis.StrictRedis(
        host=cfg.get('RedisGlobal', 'host'),
        port=cfg.getint('RedisGlobal', 'port'),
        db=cfg.getint('RedisLog', 'db'))
serv_redis_db = redis.StrictRedis(
        host=cfg.get('RedisGlobal', 'host'),
        port=cfg.getint('RedisGlobal', 'port'),
        db=cfg.getint('RedisDB', 'db'))
serv_list = redis.StrictRedis(
        host=cfg.get('RedisGlobal', 'host'),
        port=cfg.getint('RedisGlobal', 'port'),
        db=cfg.getint('RedisLIST', 'db'))

# logger
logDir = cfg.get('Log', 'directory')
logfilename = cfg.get('Log', 'update_filename')
logPath = os.path.join(logDir, logfilename)
if not os.path.exists(logDir):
    os.makedirs(logDir)
handler = logging.FileHandler(logPath)
formatter = logging.Formatter('%(asctime)s:%(levelname)s:%(name)s:%(message)s')
handler.setFormatter(formatter)
update_logger = logging.getLogger(__name__)
update_logger.setLevel(logging.INFO)
update_logger.addHandler(handler)


def check_for_updates():
    db_version = serv_redis_db.get(cfg.get('RedisDB', 'dbVersion'))
    db_version = int(db_version) if db_version is not None else 0
    updates_to_be_done = find_updates(db_version)
    if len(updates_to_be_done) == 0:
        update_logger.info('database up-to-date')
    else:
        for i in updates_to_be_done:
            exec_updates(i)


def find_updates(db_version):
    updates_to_be_done = []
    for i in DATABASE_VERSION:
        if db_version < i:
            updates_to_be_done.append(i)
    return updates_to_be_done


def exec_updates(db_version):
    result = False

    if db_version == 1:
        result = apply_update_1()

    if result:
        serv_redis_db.set(cfg.get('RedisDB', 'dbVersion'), db_version)
        update_logger.warning('dbVersion sets to {}'.format(db_version))
    else:
        update_logger.error('Something went wrong. {}'.format(result))


# Data format changed. Wipe the key.
def apply_update_1():
    serv_redis_db.delete('TEMP_CACHE_LIVE:Attribute')
    log_text = 'Executed update 1. Deleted Redis key `TEMP_CACHE_LIVE:Attribute`'
    print(log_text)
    update_logger.info(log_text)
    return True