mirror of https://github.com/MISP/misp-dashboard
80 lines
2.3 KiB
Python
80 lines
2.3 KiB
Python
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
|