103 lines
3.1 KiB
Python
103 lines
3.1 KiB
Python
#
|
|
# Copyright (c) 2014 Wind River Systems, Inc.
|
|
#
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
#
|
|
import os
|
|
import sys
|
|
import argparse
|
|
import sqlite3
|
|
|
|
database_name = "/var/run/sm/sm.db"
|
|
|
|
|
|
def main():
|
|
|
|
if not os.path.exists(database_name):
|
|
print("%s not available." % database_name)
|
|
sys.exit(0)
|
|
|
|
try:
|
|
parser = argparse.ArgumentParser(description='SM Query')
|
|
subparsers = parser.add_subparsers(help='types')
|
|
|
|
s_parser = subparsers.add_parser('service', help='Query Service')
|
|
s_parser.set_defaults(which='service')
|
|
s_parser.add_argument('service_name', help='service name')
|
|
s_parser = subparsers.add_parser('service-group', help='Query Service Group')
|
|
s_parser.set_defaults(which='service-group')
|
|
s_parser.add_argument('service_group_name', nargs='+', help='service group name')
|
|
s_parser.add_argument("--desired-state", help="display desired state", action="store_true")
|
|
|
|
args = parser.parse_args()
|
|
|
|
if args.which == 'service':
|
|
database = sqlite3.connect(database_name)
|
|
|
|
cursor = database.cursor()
|
|
|
|
cursor.execute("SELECT NAME, DESIRED_STATE, STATE, STATUS FROM "
|
|
"SERVICES WHERE NAME = '%s';"
|
|
% args.service_name)
|
|
|
|
row = cursor.fetchone()
|
|
|
|
if row is None:
|
|
print("%s is disabled." % args.service_name)
|
|
|
|
else:
|
|
service_name = row[0]
|
|
state = row[2]
|
|
status = row[3]
|
|
|
|
if status == 'none':
|
|
print("%s is %s" % (service_name, state))
|
|
else:
|
|
print("%s is %s-%s" % (service_name, state, status))
|
|
|
|
database.close()
|
|
|
|
elif args.which == 'service-group':
|
|
database = sqlite3.connect(database_name)
|
|
|
|
cursor = database.cursor()
|
|
|
|
cursor.execute("SELECT NAME, DESIRED_STATE, STATE FROM "
|
|
"SERVICE_GROUPS WHERE NAME IN (%s) AND PROVISIONED='yes';"
|
|
% ','.join("'%s'" % i for i in args.service_group_name))
|
|
|
|
rows = cursor.fetchall()
|
|
|
|
if args.desired_state:
|
|
fmt = "{0} {2} {1}"
|
|
else:
|
|
fmt = "{0} {1}"
|
|
|
|
found_list = []
|
|
for row in rows:
|
|
service_group_name = row[0]
|
|
found_list.append(service_group_name)
|
|
desired_state = row[1]
|
|
state = row[2]
|
|
|
|
print(fmt.format(service_group_name, state, desired_state))
|
|
|
|
database.close()
|
|
|
|
not_found_list = []
|
|
for g in args.service_group_name:
|
|
if g not in found_list:
|
|
not_found_list.append(g)
|
|
|
|
if len(not_found_list) > 1:
|
|
print("%s are not provisioned" % ','.join((g for g in not_found_list)))
|
|
elif len(not_found_list) == 1:
|
|
print("%s is not provisioned" % ','.join((g for g in not_found_list)))
|
|
|
|
except KeyboardInterrupt:
|
|
sys.exit()
|
|
|
|
except Exception as e:
|
|
print(e)
|
|
sys.exit(-1)
|