main.py

#!/usr/bin/python2
import os
import sys
import signal
import argparse
import logging
import backend.barista.caffe_versions as caffe_versions
from backend.barista.project import dirIsProject
from PyQt5.QtCore import QLocale
from backend.barista.utils.settings import applicationQSetting
from backend.caffe import path_loader
from gui.start_dialog import StartDialog
from PyQt5.QtWidgets import QApplication
from gui.caffepath_dialog import CaffepathDialog
from backend.networking.barista_server import BaristaServer
import threading
import subprocess


if __name__ == "__main__":
    logging.basicConfig(format='[%(asctime)s] %(levelname)s @ %(funcName)s:    %(message)s', datefmt='%d/%m/%Y %H:%M:%S',
                        filename="baristalog.txt", level=logging.INFO)
    QLocale.setDefault(QLocale.c())
    # Parse command line arguments.
    parser = argparse.ArgumentParser()
    parser.add_argument('-s', '--server', help='start a local server and add it to the host manager', default=False, action='store_true')
    parser.add_argument('-d', '--dir', help='local directory to run server sessions in', type=str, default='.')
    parser.add_argument('-o', '--open', help='path to a Barista project to be opened', type=str, default='')
    args = parser.parse_args()
    # Get caffepath out of settings.
    settings = applicationQSetting()
    settings.beginGroup("Path")
    path = settings.value("caffePath")
    app = QApplication(sys.argv)
    signal.signal(signal.SIGINT, signal.SIG_DFL)
    # Start local server on 'localhost:4200' in subprocess.
    if args.server:
        if not os.path.exists(args.dir):
            logging.error("Session folder '%s' does not exist." % (args.dir))
            sys.stderr.write("Session folder '%s' does not exist." % (args.dir))
            exit(2)
        command = './server.py --port {} --dir {}'.format(BaristaServer.DEFAULT_PORT, args.dir)
        pid = subprocess.Popen(command.split()).pid
        # Stop the server when the app is about to quit.
        app.aboutToQuit.connect(lambda: os.kill(pid, signal.SIGTERM))
    # server = BaristaServer(app, ip=None, port=BaristaServer.DEFAULT_PORT, sessionPath='sessions')
    # Set global application stylesheet.
    with open('resources/styles.qss', "r") as stylesFile:
        stylesheet = stylesFile.read()
        app.setStyleSheet(stylesheet)
    # Show caffepath dialog if settings don't contain a valid caffepath.
    caffe_versions.loadVersions()
    if caffe_versions.versionCount() == 0:
        caffedlg = CaffepathDialog("To run Barista you need a valid caffepath.", "Save and start Barista", True)
        caffedlg.show()
        app.exec_()
    # Create start dialog.
    startdlg = StartDialog()
    # If a directory has been passed as an argument, try to open a project from
    # that directory without showing the start dialog.
    if len(args.open) > 0:
        if not dirIsProject(args.open):
            print("The provided path is not a valid Barista project: Please run Barista without flags and create a project first.")
            sys.exit(2)
        args.open = os.path.abspath(args.open)
        #startdlg.setVisible(False)
        startdlg.loadProject(args.open)
    else:
        startdlg.show()
    # Run the application.
    app.exec_()