def reset(uid, sid, e_msg):
"""
Checks in all licences on error.
:param uid: User ID
:param sid: Server ID
:param e_msg: Error message
"""
Product.reset(sid)
History.reset(sid)
Updates.end(uid, 'ERROR', e_msg)
def test_end(self):
# test start update
start_id = Updates.start(server_id=1)
self.assertEqual(start_id, 1)
# test end update
Updates.end(start_id, 'UP', 'test-message')
query = Updates.query.order_by(Updates.id.desc()).limit(1).first()
self.assertEqual(query.id, 1)
self.assertEqual(query.status, 'UP')
self.assertEqual(query.info, 'test-message')
def update():
form = AddUpdateForm()
if form.validate_on_submit():
update = Updates(update=form.update.data, user_id=current_user.id, update_title=form.update_title.data)
db.session.add(update)
db.session.commit()
flash('Your update is now live!')
return redirect(url_for('index'))
return render_template("update.html", title='Post Update', form=form)
def run():
"""
Entry point for updating licenses. This script can optionally be run stand alone (cron job or windows task scheduler).
Relies on the lmutil.exe to collect server information.
"""
checked_out_history_ids = []
error = False
try:
for server in license_servers:
print('Reading from {} on port {}...'.format(
server['name'], server['port']))
s = server['name']
port = server['port']
info = None
status = None
server_id = Server.upsert(server)
update_id = Updates.start(server_id)
check_year(server_id)
try:
# Server info
lm_util_lines = os.popen(lm_util + " lmstat -c " + port + "@" +
s).readlines()
if len(lm_util_lines) == 0:
m = 'Unable to read license data. Check path to lmutil.exe and license server parameters are correct.'
raise Exception(m)
for idx, line in enumerate(lm_util_lines):
l = line.split(':')
if 'error' in line.lower():
raise Exception(line)
elif status is None:
if l[0] == s:
info = l[1].strip()
if 'license server up' in info.lower():
status = 'OK'
info = None
else:
raise Exception(
'Unable to parse license data. Check license server is functioning properly.'
)
elif (idx + 1) == len(lm_util_lines) and status is None:
raise Exception('No license data from lmutil.exe')
except Exception as e:
reset(update_id, server_id, str(e))
error = True
continue
try:
# Product info
for idx, val in enumerate(products):
lm_util_lines = os.popen(lm_util + " lmstat -f " +
val['internal-name'] + " -c " +
port + "@" + s).readlines()
for idx, line in enumerate(lm_util_lines):
if 'Users of ' + val['internal-name'] in line:
lic = line.lower().split('total of ')
lic_total = int(lic[1][:2])
lic_out = int(lic[2][:2])
Product.upsert(server_id=server_id,
internal_name=val['internal-name'],
common_name=val['common-name'],
category=val['category'],
type=val['type'],
license_out=lic_out,
license_total=lic_total)
# User info
if ") (" + s + "/" + port in line:
cols = line.split()
if 'activated' in line.lower():
user = '******'
computer = cols[2]
else:
user = cols[0]
computer = cols[1]
# Format date for sql
month = cols[-2].split("/")[0]
day = cols[-2].split("/")[1]
if len(day) == 1:
day = '0' + day
if len(month) == 1:
month = '0' + month
t = cols[-1]
hour = t.split(":")[0]
min = t.split(":")[1]
date_4_db = datetime(datetime.now().year,
int(month), int(day),
int(hour), int(min))
workstation_id = Workstation.add(
workstation=computer)
user_id = User.add(username=user)
product_id = Product.query(val['internal-name'],
server_id)
history_id = History.add(user_id, workstation_id,
server_id, product_id,
update_id, date_4_db)
checked_out_history_ids.append(history_id)
dt = datetime.now().replace(second=0, microsecond=0)
checked_out = History.time_in_none(server_id)
for c in checked_out:
if c.id not in checked_out_history_ids:
History.update(c.id, dt, server_id)
Updates.end(update_id, status, info)
except Exception as e:
reset(update_id, server_id, str(e))
print(e)
error = True
continue
except Exception as e:
error = True
print(e)
finally:
print("Read process finished.")
return error
def populate():
server_1 = get_server_id('gis-license-1')
server_2 = get_server_id('gis-license-2')
update_1 = Updates.start(server_1)
update_2 = Updates.start(server_2)
dummy_data = [
# server 1
(update_1, server_1, {
'user_id': get_user_id('Gus'),
'workstation_id': get_workstation_id('Gus-PC'),
'product_id': get_product_id(server_1, 'ARC/INFO', 0, 3),
'time_out': random_date()
}, False),
(update_1, server_1, {
'user_id': get_user_id('Ted'),
'workstation_id': get_workstation_id('Ted-PC'),
'product_id': get_product_id(server_1, 'ARC/INFO', 0, 3),
'time_out': random_date()
}, False),
(update_1, server_1, {
'user_id': get_user_id('Lin'),
'workstation_id': get_workstation_id('Lin-PC'),
'product_id': get_product_id(server_1, 'VIEWER', 0, 5),
'time_out': random_date()
}, True),
(update_1, server_1, {
'user_id': get_user_id('Sue'),
'workstation_id': get_workstation_id('Ted-PC'),
'product_id': get_product_id(server_1, 'EDITOR', 5, 10),
'time_out': random_date()
}, False),
(update_1, server_1, {
'user_id': get_user_id('Ike'),
'workstation_id': get_workstation_id('Ike-PC'),
'product_id': get_product_id(server_1, 'EDITOR', 5, 10),
'time_out': random_date()
}, False),
(update_1, server_1, {
'user_id': get_user_id('Mary'),
'workstation_id': get_workstation_id('Mary-PC'),
'product_id': get_product_id(server_1, 'EDITOR', 5, 10),
'time_out': random_date()
}, False),
(update_1, server_1, {
'user_id': get_user_id('Sally'),
'workstation_id': get_workstation_id('Sally-PC'),
'product_id': get_product_id(server_1, 'EDITOR', 5, 10),
'time_out': random_date()
}, False),
(update_1, server_1, {
'user_id': get_user_id('Ron'),
'workstation_id': get_workstation_id('Ron-PC'),
'product_id': get_product_id(server_1, 'EDITOR', 5, 10),
'time_out': random_date()
}, False),
(update_1, server_1, {
'user_id': get_user_id('Bill'),
'workstation_id': get_workstation_id('Bill-PC'),
'product_id': get_product_id(server_1, 'EDITOR', 5, 10),
'time_out': random_date()
}, True),
(update_1, server_1, {
'user_id': get_user_id('Nancy'),
'workstation_id': get_workstation_id('Nancy-PC'),
'product_id': get_product_id(server_1, 'EDITOR', 5, 10),
'time_out': random_date()
}, True),
(update_1, server_1, {
'user_id': get_user_id('Will'),
'workstation_id': get_workstation_id('Will-PC'),
'product_id': get_product_id(server_1, 'EDITOR', 5, 10),
'time_out': random_date()
}, True),
(update_1, server_1, {
'user_id': get_user_id('Sam'),
'workstation_id': get_workstation_id('Sam-PC'),
'product_id': get_product_id(server_1, 'DESKTOPBASICP', 2, 2),
'time_out': random_date()
}, False),
(update_1, server_1, {
'user_id': get_user_id('Pat'),
'workstation_id': get_workstation_id('Pat-PC'),
'product_id': get_product_id(server_1, 'DESKTOPBASICP', 2, 2),
'time_out': random_date()
}, False),
(update_1, server_1, {
'user_id': get_user_id('Alf'),
'workstation_id': get_workstation_id('Alf-PC'),
'product_id': get_product_id(server_1, 'DESKTOPADVP', 1, 4),
'time_out': random_date()
}, False),
(update_1, server_1, {
'user_id': get_user_id('Sam'),
'workstation_id': get_workstation_id('Sam-PC'),
'product_id': get_product_id(server_1, '3DANALYSTP', 2, 4),
'time_out': random_date()
}, False),
(update_1, server_1, {
'user_id': get_user_id('Pat'),
'workstation_id': get_workstation_id('Pat-PC'),
'product_id': get_product_id(server_1, '3DANALYSTP', 2, 4),
'time_out': random_date()
}, False),
# server 2
(update_2, server_2, {
'user_id': get_user_id('Bob'),
'workstation_id': get_workstation_id('Pat-PC'),
'product_id': get_product_id(server_2, 'ARC/INFO', 2, 3),
'time_out': random_date()
}, False),
(update_2, server_2, {
'user_id': get_user_id('Kim'),
'workstation_id': get_workstation_id('Alf-PC'),
'product_id': get_product_id(server_2, 'ARC/INFO', 2, 3),
'time_out': random_date()
}, False),
(update_2, server_2, {
'user_id': get_user_id('Joe'),
'workstation_id': get_workstation_id('Joe-PC'),
'product_id': get_product_id(server_2, 'VIEWER', 0, 10),
'time_out': random_date()
}, True),
(update_2, server_2, {
'user_id': get_user_id('Liz'),
'workstation_id': get_workstation_id('Joe-PC'),
'product_id': get_product_id(server_2, 'VIEWER', 0, 10),
'time_out': random_date()
}, True),
(update_2, server_2, {
'user_id': get_user_id('Pam'),
'workstation_id': get_workstation_id('Pam-PC'),
'product_id': get_product_id(server_2, 'EDITOR', 1, 10),
'time_out': random_date()
}, False),
(update_2, server_2, {
'user_id': get_user_id('Ann'),
'workstation_id': get_workstation_id('Ann-PC'),
'product_id': get_product_id(server_2, 'DESKTOPBASICP', 0, 2),
'time_out': random_date()
}, True),
(update_2, server_2, {
'user_id': get_user_id('Bob'),
'workstation_id': get_workstation_id('Bob-PC'),
'product_id': get_product_id(server_2, 'NETWORK', 2, 2),
'time_out': random_date()
}, False),
(update_2, server_2, {
'user_id': get_user_id('Kim'),
'workstation_id': get_workstation_id('Kim-PC'),
'product_id': get_product_id(server_2, 'NETWORK', 2, 2),
'time_out': random_date()
}, False),
]
for d in dummy_data:
id = History.add(d[0], d[1], **d[2])
if d[3]:
check_in(server_id=d[1], h=id, data=d[2])
Updates.end(update_1, 'UP', '')
Updates.end(update_2, 'UP', '')
def clear_log():
try:
Updates.clear()
return make_response('ok', 200)
except Exception as e:
return make_response(str(e), 500)
def read(license_file=None):
"""
entry point for reading license data from a FlexLM license server.
:param license_file: manually pass in a license file in the same format the lmutil.exe displays data.
:return:
"""
for s in license_servers:
try:
info = ''
server_id = Server.upsert(s['hostname'], s['port'])
update_id = Updates.start(server_id)
check_year(server_id)
checked_out_history_ids = []
if license_file:
with open(license_file) as process:
lines = process.read()
else:
process = subprocess.Popen([
lm_util, "lmstat", "-f", "-c", "{}@{}".format(
s['port'], s['hostname'])
],
stdout=subprocess.PIPE,
stderr=subprocess.PIPE,
bufsize=1,
universal_newlines=True)
lines = process.stdout.read()
has_error = parse_error_info(lines)
if has_error:
reset(update_id, server_id,
'{}@{}: {}'.format(s['port'], s['hostname'], has_error))
raise Exception(has_error)
license_data = split_license_data(lines)
updates = {'update_id': update_id, 'status': None}
server_information = parse_server_info(lines)
if server_information:
updates['status'] = server_information[2]
else:
updates['status'] = "DOWN"
reset(update_id, server_id,
'{}@{} is DOWN'.format(s['port'], s['hostname']))
raise Exception(has_error)
for lic in license_data:
split_line = lic.split('FLOATING LICENSE')
product_id = add_product(split_line[0], server_id=server_id)
users_and_workstations = add_users_and_workstations(
split_line[-1])
if product_id and len(users_and_workstations):
data = map_product_id(product_id, users_and_workstations)
for kwargs in data:
history_id = History.add(update_id=update_id,
server_id=server_id,
**kwargs)
checked_out_history_ids.append(history_id)
dt = datetime.now().replace(second=0, microsecond=0)
checked_out = History.time_in_none(server_id)
for c in checked_out:
if c.id not in checked_out_history_ids:
History.update(c.id, dt, server_id)
except Exception as e:
info = "{} error: {}".format(s['hostname'], str(e))
logger.error(str(e))
pass
finally:
logger.info('Finished reading data from \'{}\'. '
'Update details | id:{} | status:{} | info:{}.'.format(
s['hostname'], update_id, updates['status'], info))
Updates.end(update_id, updates['status'], info)