89 lines
2.8 KiB
Python
89 lines
2.8 KiB
Python
#
|
|
# Copyright (c) 2016 Wind River Systems, Inc.
|
|
#
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
#
|
|
import os
|
|
import sys
|
|
import argparse
|
|
import sqlite3
|
|
|
|
from sm_api_msg_utils import restart_service as restart_service
|
|
from sm_api_msg_utils import restart_service_safe as restart_service_safe
|
|
from sm_api_msg_utils import database_running_name as database_name
|
|
|
|
|
|
def main():
|
|
filename = os.path.basename(sys.argv[0])
|
|
if "sm-manage" == filename:
|
|
action = "manage"
|
|
elif "sm-unmanage" == filename:
|
|
action = "unmanage"
|
|
elif "sm-restart-safe" == filename:
|
|
action = "restart-safe"
|
|
else:
|
|
action = "restart"
|
|
|
|
try:
|
|
parser = argparse.ArgumentParser(description='SM Action ')
|
|
subparsers = parser.add_subparsers(help='types')
|
|
|
|
# Service
|
|
service_parser = subparsers.add_parser('service', help='service action')
|
|
service_parser.set_defaults(which='service')
|
|
service_parser.add_argument('service', help='service name')
|
|
|
|
args = parser.parse_args()
|
|
|
|
if args.which == 'service':
|
|
database = sqlite3.connect(database_name)
|
|
|
|
cursor = database.cursor()
|
|
|
|
cursor.execute("SELECT * FROM SERVICES WHERE NAME = '%s';"
|
|
% args.service)
|
|
row = cursor.fetchone()
|
|
if row is None:
|
|
print("Given service (%s) does not exist." % args.service)
|
|
sys.exit()
|
|
|
|
database.close()
|
|
|
|
SM_VAR_RUN_SERVICES_DIR = '/var/run/sm/services'
|
|
unmanage_filepath = SM_VAR_RUN_SERVICES_DIR + '/'
|
|
unmanage_filename = "%s.unmanaged" % args.service
|
|
|
|
if 'manage' == action:
|
|
if os.path.exists(SM_VAR_RUN_SERVICES_DIR):
|
|
if os.path.isfile(unmanage_filepath + unmanage_filename):
|
|
os.remove(unmanage_filepath + unmanage_filename)
|
|
|
|
print("Service (%s) is now being managed." % args.service)
|
|
|
|
elif 'unmanage' == action:
|
|
if not os.path.exists(SM_VAR_RUN_SERVICES_DIR):
|
|
os.makedirs(SM_VAR_RUN_SERVICES_DIR)
|
|
|
|
if not os.path.isfile(unmanage_filepath + unmanage_filename):
|
|
open(unmanage_filepath + unmanage_filename, 'w').close()
|
|
|
|
print("Service (%s) is no longer being managed." % args.service)
|
|
|
|
elif 'restart-safe' == action:
|
|
restart_service_safe(args.service)
|
|
print("Service (%s) is restarting." % args.service)
|
|
|
|
else:
|
|
restart_service(args.service)
|
|
|
|
print("Service (%s) is restarting." % args.service)
|
|
|
|
sys.exit(0)
|
|
|
|
except KeyboardInterrupt:
|
|
sys.exit()
|
|
|
|
except Exception as e:
|
|
print(e)
|
|
sys.exit(-1)
|