kalenis_cli.py

import os
import pkg_resources
import subprocess
import appdirs
import click
from pathlib import Path
from shutil import copyfile
from proteus import config, Model

ERROR_COLOR = 'red'
OK_COLOR = 'green'
MODULES = {
    'qc': ['lims_quality_control'],
    'services': ['lims_analysis_sheet'],
    }


def get_config():
    """
    Read configuration file and return its contents
    """
    cfg_dir = appdirs.user_config_dir('kalenis')
    Path(cfg_dir).mkdir(exist_ok=True)
    cfg_file = os.path.join(cfg_dir, 'kalenis.conf')
    if not os.path.isfile(cfg_file):
        source = pkg_resources.resource_filename(
            __name__, '/kalenis_lims/kalenis.conf.dist')
        copyfile(source, cfg_file)
    return cfg_file


@click.group()
@click.version_option()
def cli():
    pass


@cli.command()
@click.option('-d', '--database', default='kalenislims', show_default=True)
@click.option('-l', '--language', required=True,
              type=click.Choice(['en', 'es'], case_sensitive=False))
@click.option('-i', '--industry', required=True,
              type=click.Choice(['qc', 'services'], case_sensitive=False))
def setup(database, language, industry):
    """
    This is the setup command for Kalenis LIMS
    """
    click.echo('Setup Kalenis for %s...' % industry)

    click.echo('Creating user config file...')
    config_file = get_config()

    click.echo('Creating the database...')
    process = subprocess.run(['createdb', database],
        stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    if process.returncode != 0:
        err = process.stdout.decode('utf-8')
        if len(err) == 0:
            err = process.stderr.decode('utf-8')
            click.echo(click.style(err.format(ERROR_COLOR), fg=ERROR_COLOR))
            return process.returncode

    click.echo('Initializing the database...')
    subprocess.run(['trytond-admin', '-d', database, '-c', config_file,
        '--all'])

    click.echo('Installing modules...')
    subprocess.run(['trytond-admin', '-d', database, '-c', config_file, '-u',
        ' '.join(MODULES[industry]), '--activate-dependencies'])
    subprocess.run(['trytond-admin', '-d', database, '-c', config_file, '-u',
        'user_view'])

    if language == 'es':
        click.echo('Loading translations for spanish language...')
        config.set_trytond(database, config_file=config_file)
        User = Model.get('res.user')
        Lang = Model.get('ir.lang')

        lang, = Lang.find([('code', '=', language)])
        lang.translatable = True
        lang.save()

        subprocess.run(['trytond-admin', '-d', database, '-c', config_file,
            '--all'])

        user = User.find()[0]
        user.language = lang
        user.save()

    click.echo('Downloading front-end...')
    front_end_dir = '%s/kalenis_front_end' % os.environ.get('HOME', '')
    front_end_file = 'frontend_dist_6.0.tar.gz'
    Path(front_end_dir).mkdir(exist_ok=True)
    subprocess.run(['wget',
        'https://downloads.kalenislims.com/%s' % front_end_file, '-O',
        '%s/%s' % (front_end_dir, front_end_file)],
        stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    subprocess.run(['tar', 'xzvf', '%s/%s' % (front_end_dir, front_end_file),
        '--directory', front_end_dir],
        stdout=subprocess.PIPE, stderr=subprocess.PIPE)

    click.echo(click.style(
        'Congratulations, the setup process has finished ok. Now you can '
        'execute "kalenis-cli run" to start Kalenis LIMS server'.format(
            OK_COLOR), fg=OK_COLOR))


@cli.command()
@click.option('-d', '--database', default='kalenislims', show_default=True)
def run(database):
    """
    Run Kalenis LIMS service
    """
    click.echo('Starting Kalenis LIMS server')
    click.echo('Kalenis LIMS running, you can go to http://localhost:8000')
    config_file = get_config()
    os.environ['TRYTOND_web__root'] = \
        '%s/kalenis_front_end/frontend_dist_6.0' % os.environ.get('HOME', '')
    subprocess.run(['trytond', '-d', database, '-c', config_file])