contentloader

#!/usr/bin/env python3

# Auto-install and activate a virtualenv if possible
import autovenv
autovenv.run()

import contextlib
import fcntl
import inspect
import os
import subprocess

from utils import script_directory

from config import (
    DB_NAME,
    JOBS_PRE,
    JOBS_POST,
)


@contextlib.contextmanager
def process_lock(name=None):
    '''
    Use a file lock to ensure only one process runs at a time
    '''
    caller = inspect.getouterframes(inspect.currentframe())[2]
    caller_dir = os.path.dirname(os.path.realpath(caller.filename))
    if not name:
        name = os.path.splitext(os.path.basename(caller.filename))
    with open(os.path.join(caller_dir, f'{name}.lock'), 'w') as f:
        try:
            fcntl.lockf(f, fcntl.LOCK_EX)
            yield
        finally:
            fcntl.lockf(f, fcntl.LOCK_UN)


with script_directory():
    for shell_command in JOBS_PRE:
        subprocess.run(shell_command, shell=True)

    with process_lock(DB_NAME):
        import contentloader
        contentloader.ContentLoader()

    for shell_command in JOBS_POST:
        subprocess.run(shell_command, shell=True)