Example #1
0
        db.session.commit()
        time.sleep(settings.interval)


if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument("instance",
                        type=str,
                        help="Name of this node instance.")
    parser.add_argument("-v",
                        "--verbose",
                        action="store_true",
                        help="Enable verbosity for debug information.")
    args = parser.parse_args()

    app = create_app()

    if args.verbose:
        logging.basicConfig(level=logging.DEBUG)
    else:
        logging.basicConfig(level=logging.INFO)

    logging.getLogger("requests").setLevel(logging.WARNING)
    logging.getLogger("urllib3").setLevel(logging.WARNING)
    log = logging.getLogger("dist-%s" % args.instance)

    with app.app_context():
        if args.instance == "dist.scheduler":
            scheduler()
        elif args.instance == "dist.status":
            status_caching()
Example #2
0
File: app.py Project: CIRCL/cuckoo
import logging
import os.path
import sys

from distributed.app import create_app

sys.path.insert(0, os.path.join(os.path.abspath(os.path.dirname(__file__)), ".."))

from lib.cuckoo.core.startup import drop_privileges

logging.basicConfig(level=logging.INFO)
logging.getLogger("requests").setLevel(logging.WARNING)
logging.getLogger("urllib3").setLevel(logging.WARNING)
log = logging.getLogger("cuckoo.distributed")

application = create_app()

if __name__ == "__main__":
    p = argparse.ArgumentParser()
    p.add_argument("host", nargs="?", default="127.0.0.1", help="Host to listen on.")
    p.add_argument("port", nargs="?", type=int, default=9003, help="Port to listen on.")
    p.add_argument("-u", "--user", type=str, help="Drop user privileges to this user.")
    p.add_argument("-v", "--verbose", action="store_true", help="Enable verbose logging.")
    args = p.parse_args()

    if args.user:
        drop_privileges(args.user)

    log.setLevel(logging.DEBUG)
    application.run(host=args.host, port=args.port, debug=True)
Example #3
0
            t.started = datetime.datetime.strptime(task["started_on"],
                                                   "%Y-%m-%d %H:%M:%S")
            t.completed = datetime.datetime.now()

        log.debug("Fetched %d reports from %s", len(tasks), node.name)

        db.session.commit()
        time.sleep(settings.interval)

if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument("instance", type=str, help="Name of this node instance.")
    parser.add_argument("-v", "--verbose", action="store_true", help="Enable verbosity for debug information.")
    args = parser.parse_args()

    app = create_app()

    if args.verbose:
        logging.basicConfig(level=logging.DEBUG)
    else:
        logging.basicConfig(level=logging.INFO)

    logging.getLogger("requests").setLevel(logging.WARNING)
    logging.getLogger("urllib3").setLevel(logging.WARNING)
    log = logging.getLogger("dist-%s" % args.instance)

    with app.app_context():
        if args.instance == "dist.scheduler":
            scheduler()
        elif args.instance == "dist.status":
            status_caching()
Example #4
0
import logging
import os.path
import sys

from distributed.app import create_app

sys.path.append(os.path.join(os.path.abspath(os.path.dirname(__file__)), ".."))

from lib.cuckoo.core.startup import drop_privileges

logging.basicConfig(level=logging.INFO)
logging.getLogger("requests").setLevel(logging.WARNING)
logging.getLogger("urllib3").setLevel(logging.WARNING)
log = logging.getLogger("cuckoo.distributed")

application = create_app()

if __name__ == "__main__":
    p = argparse.ArgumentParser()
    p.add_argument("host",
                   nargs="?",
                   default="127.0.0.1",
                   help="Host to listen on.")
    p.add_argument("port",
                   nargs="?",
                   type=int,
                   default=9003,
                   help="Port to listen on.")
    p.add_argument("-u",
                   "--user",
                   type=str,