mirror of
https://github.com/Pardus-LiderAhenk/ahenk
synced 2024-12-23 22:42:17 +03:00
227 lines
9.8 KiB
Python
227 lines
9.8 KiB
Python
#!/usr/bin/python3
|
|
# -*- coding: utf-8 -*-
|
|
# Author: İsmail BAŞARAN <ismail.basaran@tubitak.gov.tr> <basaran.ismaill@gmail.com>
|
|
# Author: Volkan Şahin <volkansah.in> <bm.volkansahin@gmail.com>
|
|
import sqlite3
|
|
import threading
|
|
from base.scope import Scope
|
|
|
|
|
|
class AhenkDbService(object):
|
|
"""
|
|
Sqlite manager for ahenk
|
|
"""
|
|
|
|
def __init__(self):
|
|
scope = Scope.get_instance()
|
|
self.logger = scope.get_logger()
|
|
self.configurationManager = scope.get_configuration_manager()
|
|
self.db_path = self.configurationManager.get('BASE', 'dbPath')
|
|
self.connection = None
|
|
self.cursor = None
|
|
|
|
self.lock = threading.Lock()
|
|
|
|
# TODO get columns anywhere
|
|
# TODO scheduler db init get here
|
|
|
|
def initialize_table(self):
|
|
|
|
self.check_and_create_table('task',
|
|
['id INTEGER', 'create_date TEXT', 'modify_date TEXT', 'command_cls_id TEXT',
|
|
'parameter_map BLOB', 'deleted INTEGER', 'plugin TEXT', 'cron_expr TEXT',
|
|
'file_server TEXT'])
|
|
self.check_and_create_table('policy',
|
|
['id INTEGER PRIMARY KEY AUTOINCREMENT', 'policy_id INTEGER',
|
|
'type TEXT', 'version TEXT', 'name TEXT',
|
|
'execution_id TEXT', 'expiration_date TEXT', 'assign_date TEXT'])
|
|
self.check_and_create_table('profile', ['id INTEGER', 'create_date TEXT', 'label TEXT', 'description TEXT',
|
|
'overridable INTEGER', 'active TEXT', 'deleted TEXT',
|
|
'profile_data TEXT', 'modify_date TEXT', 'plugin TEXT', 'policy_id INTEGER'])
|
|
self.check_and_create_table('plugin',
|
|
['id INTEGER PRIMARY KEY AUTOINCREMENT', 'active TEXT', 'create_date TEXT',
|
|
'deleted TEXT', 'description TEXT', 'machine_oriented TEXT', 'modify_date TEXT',
|
|
'name TEXT', 'policy_plugin TEXT', 'user_oriented TEXT', 'version TEXT',
|
|
'task_plugin TEXT', 'x_based TEXT'])
|
|
self.check_and_create_table('registration',
|
|
['jid TEXT', 'password TEXT', 'registered INTEGER', 'dn TEXT', 'params TEXT',
|
|
'timestamp TEXT'])
|
|
self.check_and_create_table('contract', ['id INTEGER PRIMARY KEY AUTOINCREMENT', 'content BLOB', 'title TEXT',
|
|
'timestamp TEXT'])
|
|
self.check_and_create_table('agreement',
|
|
['id INTEGER PRIMARY KEY AUTOINCREMENT', 'contract_id TEXT', 'username TEXT',
|
|
'timestamp TEXT', 'choice TEXT'])
|
|
self.check_and_create_table('session', ['id INTEGER PRIMARY KEY AUTOINCREMENT', 'username TEXT', 'display TEXT',
|
|
'desktop TEXT', 'timestamp TEXT', 'ip TEXT', 'domain TEXT'])
|
|
|
|
self.check_and_create_table('mail', ['id INTEGER PRIMARY KEY AUTOINCREMENT', 'command TEXT', 'mailstatus INTEGER', 'timestamp TEXT'])
|
|
|
|
self.check_and_create_table('service', ['id INTEGER PRIMARY KEY AUTOINCREMENT', 'serviceName TEXT', 'serviceStatus TEXT','timestamp TEXT','task_id INTEGER'])
|
|
self.check_and_create_table('app_restriction', ['id INTEGER PRIMARY KEY AUTOINCREMENT', 'application_name TEXT', 'username TEXT', 'restriction INTEGER'])
|
|
|
|
|
|
def get_cols(self, table_name):
|
|
if table_name == 'agreement':
|
|
return ['contract_id', 'username', 'timestamp', 'choice']
|
|
elif table_name == 'contract':
|
|
return ['content', 'title', 'timestamp']
|
|
elif table_name == 'session':
|
|
return ['username', 'display', 'desktop', 'timestamp', 'ip', 'domain']
|
|
elif table_name == 'task':
|
|
return ['id', 'create_date', 'modify_date', 'command_cls_id', 'parameter_map', 'deleted', 'plugin',
|
|
'cron_expr', 'file_server']
|
|
elif table_name == 'plugin':
|
|
return ['id', 'active', 'create_date', 'deleted', 'description', 'machine_oriented', 'modify_date', 'name',
|
|
'policy_plugin', 'user_oriented', 'version', 'task_plugin', 'x_based']
|
|
else:
|
|
return None
|
|
|
|
def connect(self):
|
|
try:
|
|
self.connection = sqlite3.connect(self.db_path, check_same_thread=False)
|
|
self.cursor = self.connection.cursor()
|
|
except Exception as e:
|
|
self.logger.error('Database connection error: {0}'.format(str(e)))
|
|
|
|
def check_and_create_table(self, table_name, cols):
|
|
if table_name == 'session':
|
|
if self.get_table_name('session') is not None:
|
|
domain = self.get_column_name('session', 'domain')
|
|
if domain is None:
|
|
self.drop_table('session')
|
|
try:
|
|
self.lock.acquire(True)
|
|
if self.cursor:
|
|
cols = ', '.join([str(x) for x in cols])
|
|
self.cursor.execute('create table if not exists ' + table_name + ' (' + cols + ')')
|
|
else:
|
|
self.logger.warning('Could not create table cursor is None! Table Name : {0}'.format(str(table_name)))
|
|
finally:
|
|
self.lock.release()
|
|
|
|
def drop_table(self, table_name):
|
|
try:
|
|
self.lock.acquire(True)
|
|
sql = 'DROP TABLE ' + table_name
|
|
self.cursor.execute(sql)
|
|
self.connection.commit()
|
|
finally:
|
|
self.lock.release()
|
|
|
|
def update(self, table_name, cols, args, criteria=None):
|
|
try:
|
|
self.lock.acquire(True)
|
|
if self.connection:
|
|
if criteria is None:
|
|
cols = ', '.join([str(x) for x in cols])
|
|
params = ', '.join(['?' for x in args])
|
|
sql = 'INSERT INTO ' + table_name + ' (' + cols + ') VALUES (' + params + ')'
|
|
else:
|
|
update_list = ''
|
|
for index in range(len(cols)):
|
|
update_list = update_list + ' ' + cols[index] + ' = ?,'
|
|
update_list = update_list.strip(',')
|
|
sql = 'UPDATE ' + table_name + ' SET ' + update_list + ' where ' + criteria
|
|
self.cursor.execute(sql, tuple(args))
|
|
self.connection.commit()
|
|
return self.cursor.lastrowid
|
|
else:
|
|
self.logger.warning('Could not update table cursor is None! Table Name : {0}'.format(str(table_name)))
|
|
return None
|
|
except Exception as e:
|
|
self.logger.error(
|
|
'Updating table error ! Table Name : {0} Error Mesage: {1}'.format(str(table_name), str(e)))
|
|
finally:
|
|
self.lock.release()
|
|
|
|
def delete(self, table_name, criteria):
|
|
try:
|
|
self.lock.acquire(True)
|
|
if self.cursor:
|
|
sql = 'DELETE FROM ' + table_name
|
|
if criteria:
|
|
sql += ' where ' + str(criteria)
|
|
self.cursor.execute(sql)
|
|
self.connection.commit()
|
|
finally:
|
|
self.lock.release()
|
|
|
|
def findByProperty(self):
|
|
# Not implemented yet
|
|
pass
|
|
|
|
def select(self, table_name, cols='*', criteria='', orderby=''):
|
|
if self.cursor:
|
|
try:
|
|
self.lock.acquire(True)
|
|
if not cols == '*':
|
|
cols = ', '.join([str(x) for x in cols])
|
|
sql = 'SELECT ' + cols + ' FROM ' + table_name
|
|
if criteria != '':
|
|
sql += ' where '
|
|
sql += criteria
|
|
if orderby != '':
|
|
sql += ' order by '
|
|
sql += orderby
|
|
|
|
self.cursor.execute(sql)
|
|
rows = self.cursor.fetchall()
|
|
return rows
|
|
except:
|
|
raise
|
|
finally:
|
|
self.lock.release()
|
|
else:
|
|
self.logger.warning('Could not select table cursor is None! Table Name : {0}'.format(str(table_name)))
|
|
|
|
def select_one_result(self, table_name, col, criteria=''):
|
|
if self.cursor:
|
|
try:
|
|
self.lock.acquire(True)
|
|
sql = 'SELECT ' + col + ' FROM ' + table_name
|
|
if criteria != '':
|
|
sql += ' where '
|
|
sql += criteria
|
|
self.cursor.execute(sql)
|
|
row = self.cursor.fetchone()
|
|
if row is not None:
|
|
return row[0]
|
|
else:
|
|
return None
|
|
except:
|
|
raise
|
|
finally:
|
|
self.lock.release()
|
|
else:
|
|
self.logger.warning('Could not select table cursor is None! Table Name : {0}'.format(str(table_name)))
|
|
|
|
def close(self):
|
|
try:
|
|
self.cursor.close()
|
|
self.connection.close()
|
|
except Exception as e:
|
|
self.logger.error('Closing database connection error: {0}'.format(str(e)))
|
|
|
|
def get_column_name(self, table_name, column_name):
|
|
try:
|
|
self.lock.acquire(True)
|
|
if self.cursor:
|
|
reader = self.cursor.execute('SELECT * FROM ' + table_name)
|
|
cols = [x[0] for x in reader.description]
|
|
self.connection.commit()
|
|
if column_name in cols:
|
|
return column_name
|
|
else:
|
|
return None
|
|
finally:
|
|
self.lock.release()
|
|
|
|
def get_table_name(self, table_name):
|
|
result = self.cursor.execute('SELECT name FROM sqlite_master WHERE type=\'table\';')
|
|
tables = []
|
|
for name in result:
|
|
tables.append(name[0])
|
|
if table_name in tables:
|
|
return table_name
|
|
else:
|
|
return None
|