cms/scripts/cmsResourceService at future · algorithm-ninja/cms · GitHub

This repository was archived by the owner on Sep 24, 2024. It is now read-only.

History

executable file

84 lines (66 loc) · 2.89 KB

Raw

#!/usr/bin/env python2

# -*- coding: utf-8 -*-

# Contest Management System - http://cms-dev.github.io/

# This program is free software: you can redistribute it and/or modify

# it under the terms of the GNU Affero General Public License as

# published by the Free Software Foundation, either version 3 of the

# License, or (at your option) any later version.

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

# GNU Affero General Public License for more details.

# You should have received a copy of the GNU Affero General Public License

# along with this program. If not, see <http://www.gnu.org/licenses/>.

from __future__ import absolute_import

from __future__ import print_function

from __future__ import unicode_literals

# We enable monkey patching to make many libraries gevent-friendly

# (for instance, urllib3, used by requests)

import gevent.monkey

gevent.monkey.patch_all()

import argparse

import logging

import sys

from cms import ConfigError, contest_id_from_args, get_safe_shard, utf8_decoder

from cms.db import ask_for_contest, test_db_connection

from cms.service.ResourceService import ResourceService

logger = logging.getLogger(__name__)

RESTART_DISABLED = "RESTART_DISABLED"

def main():

"""Parses arguments and launch service.

"""

parser = argparse.ArgumentParser(

description="Resource monitor and service starter for CMS.")

parser.add_argument("-a", "--autorestart", action="store", nargs="?",

type=utf8_decoder, const=None, metavar="CONTEST_ID",

dest="contest_id", default=RESTART_DISABLED,

help="restart automatically services on its machine; "

"a contest id or 'ALL' can be specified")

parser.add_argument("shard", action="store", type=int, nargs="?")

args = parser.parse_args()

try:

args.shard = get_safe_shard("ResourceService", args.shard)

except ValueError:

raise ConfigError("Couldn't autodetect shard number and "

"no shard specified for service %s, "

"quitting." % ("ResourceService", ))

test_db_connection()

autorestart = args.contest_id != RESTART_DISABLED

contest_id = None

if autorestart:

contest_id = contest_id_from_args(args.contest_id, ask_for_contest)

success = ResourceService(args.shard,

contest_id=contest_id,

autorestart=autorestart).run()

return 0 if success is True else 1

if __name__ == "__main__":

try:

sys.exit(main())

except ConfigError as error:

logger.critical(error.message)

sys.exit(1)

cms/scripts/cmsResourceService at future · algorithm-ninja/cms · GitHub

Uh oh!

Expand file tree

Breadcrumbs

cmsResourceService

Latest commit

History

Breadcrumbs

cmsResourceService

File metadata and controls