mirror of
https://github.com/Pardus-LiderAhenk/ahenk
synced 2024-11-23 02:02:20 +03:00
ahenk clean command was added- you can clean ahenk db and conf parameters with this command: sudo python3 ahenkd.py clean
This commit is contained in:
parent
b81d9b079b
commit
1bcf949782
1 changed files with 34 additions and 4 deletions
|
@ -12,6 +12,7 @@ import sys
|
||||||
import threading
|
import threading
|
||||||
import time
|
import time
|
||||||
|
|
||||||
|
from base.system.system import System
|
||||||
from base.Scope import Scope
|
from base.Scope import Scope
|
||||||
from base.config.ConfigManager import ConfigManager
|
from base.config.ConfigManager import ConfigManager
|
||||||
from base.database.AhenkDbService import AhenkDbService
|
from base.database.AhenkDbService import AhenkDbService
|
||||||
|
@ -28,6 +29,7 @@ from base.scheduler.scheduler_factory import SchedulerFactory
|
||||||
from base.task.TaskManager import TaskManager
|
from base.task.TaskManager import TaskManager
|
||||||
|
|
||||||
pidfilePath = '/var/run/ahenk.pid'
|
pidfilePath = '/var/run/ahenk.pid'
|
||||||
|
configFilePath = '/etc/ahenk/ahenk.conf'
|
||||||
|
|
||||||
|
|
||||||
class AhenkDeamon(BaseDaemon):
|
class AhenkDeamon(BaseDaemon):
|
||||||
|
@ -162,7 +164,6 @@ class AhenkDeamon(BaseDaemon):
|
||||||
globalscope = Scope()
|
globalscope = Scope()
|
||||||
globalscope.setInstance(globalscope)
|
globalscope.setInstance(globalscope)
|
||||||
|
|
||||||
configFilePath = '/etc/ahenk/ahenk.conf'
|
|
||||||
configfileFolderPath = '/etc/ahenk/config.d/'
|
configfileFolderPath = '/etc/ahenk/config.d/'
|
||||||
|
|
||||||
# configuration manager must be first load
|
# configuration manager must be first load
|
||||||
|
@ -252,7 +253,7 @@ class AhenkDeamon(BaseDaemon):
|
||||||
if 'login' == str(params[0]):
|
if 'login' == str(params[0]):
|
||||||
logger.debug('[AhenkDeamon] Signal is :{}'.format(str(params[0])))
|
logger.debug('[AhenkDeamon] Signal is :{}'.format(str(params[0])))
|
||||||
login_message = message_manager.login_msg(params[1])
|
login_message = message_manager.login_msg(params[1])
|
||||||
# messenger.send_direct_message(login_message)
|
messenger.send_direct_message(login_message)
|
||||||
get_policy_message = message_manager.policy_request_msg(params[1])
|
get_policy_message = message_manager.policy_request_msg(params[1])
|
||||||
messenger.send_direct_message(get_policy_message)
|
messenger.send_direct_message(get_policy_message)
|
||||||
logger.debug('[AhenkDeamon] login event is handled for user:' + params[1])
|
logger.debug('[AhenkDeamon] login event is handled for user:' + params[1])
|
||||||
|
@ -287,8 +288,29 @@ def set_event(event_param):
|
||||||
config.read(pidfilePath)
|
config.read(pidfilePath)
|
||||||
config.set('PID', 'event', event_param)
|
config.set('PID', 'event', event_param)
|
||||||
|
|
||||||
with open(pidfilePath, 'w') as config_file:
|
with open(pidfilePath, 'w') as conf_file:
|
||||||
config.write(config_file)
|
config.write(conf_file)
|
||||||
|
|
||||||
|
|
||||||
|
def clean():
|
||||||
|
try:
|
||||||
|
config = configparser.ConfigParser()
|
||||||
|
config._interpolation = configparser.ExtendedInterpolation()
|
||||||
|
config.read(configFilePath)
|
||||||
|
db_path = config.get('BASE', 'dbPath')
|
||||||
|
|
||||||
|
os.remove(db_path)
|
||||||
|
|
||||||
|
config.set('BASE', 'uid', '')
|
||||||
|
config.set('BASE', 'password', '')
|
||||||
|
|
||||||
|
with open(configFilePath, 'w') as file:
|
||||||
|
config.write(file)
|
||||||
|
|
||||||
|
file.close()
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
print('Error while running clean command. Error Message {}'.format(str(e)))
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
|
@ -300,11 +322,19 @@ if __name__ == '__main__':
|
||||||
print('starting')
|
print('starting')
|
||||||
ahenkdaemon.run()
|
ahenkdaemon.run()
|
||||||
elif sys.argv[1] == 'stop':
|
elif sys.argv[1] == 'stop':
|
||||||
|
print('stopping')
|
||||||
|
# TODO
|
||||||
ahenkdaemon.stop()
|
ahenkdaemon.stop()
|
||||||
elif sys.argv[1] == 'restart':
|
elif sys.argv[1] == 'restart':
|
||||||
|
# TODO
|
||||||
|
print('restarting')
|
||||||
ahenkdaemon.restart()
|
ahenkdaemon.restart()
|
||||||
elif sys.argv[1] == 'status':
|
elif sys.argv[1] == 'status':
|
||||||
|
# TODO
|
||||||
print('status')
|
print('status')
|
||||||
|
elif sys.argv[1] == 'clean':
|
||||||
|
print('cleaning')
|
||||||
|
clean()
|
||||||
else:
|
else:
|
||||||
print('Unknown command. Usage : %s start|stop|restart|status' % sys.argv[0])
|
print('Unknown command. Usage : %s start|stop|restart|status' % sys.argv[0])
|
||||||
sys.exit(2)
|
sys.exit(2)
|
||||||
|
|
Loading…
Reference in a new issue