some code...
This commit is contained in:
parent
f9c99d4070
commit
cac342582b
19 changed files with 788 additions and 0 deletions
0
lib/core/__init__.py
Normal file
0
lib/core/__init__.py
Normal file
28
lib/core/command.py
Normal file
28
lib/core/command.py
Normal file
|
|
@ -0,0 +1,28 @@
|
|||
import subprocess
|
||||
|
||||
|
||||
class Command:
|
||||
def __init__(self, command):
|
||||
self.command = command
|
||||
self.output = ""
|
||||
self.error = ""
|
||||
|
||||
def __str__(self):
|
||||
return command
|
||||
|
||||
def hasProcessed(self):
|
||||
return self.output != "" and self.error != ""
|
||||
|
||||
def doProcess(self):
|
||||
process = subprocess.Popen([self.command], stdout=subprocess.PIPE)
|
||||
self.output, self.error = process.communicate()
|
||||
|
||||
def getOutput(self):
|
||||
if not self.hasProcessed():
|
||||
self.doProcess()
|
||||
return self.output
|
||||
|
||||
def getError(self):
|
||||
if not self.hasProcessed():
|
||||
self.doProcess()
|
||||
return self.error
|
||||
132
lib/core/daemon.py
Normal file
132
lib/core/daemon.py
Normal file
|
|
@ -0,0 +1,132 @@
|
|||
#!/usr/bin/env python
|
||||
|
||||
import sys, os, time, atexit
|
||||
from signal import SIGTERM
|
||||
|
||||
|
||||
class Daemon:
|
||||
"""
|
||||
A generic daemon class.
|
||||
|
||||
Usage: subclass the Daemon class and override the run() method
|
||||
"""
|
||||
|
||||
def __init__(self, pidfile, stdin='/dev/null', stdout='/dev/null', stderr='/dev/null'):
|
||||
self.stdin = stdin
|
||||
self.stdout = stdout
|
||||
self.stderr = stderr
|
||||
self.pidfile = pidfile
|
||||
|
||||
def daemonize(self):
|
||||
"""
|
||||
do the UNIX double-fork magic, see Stevens' "Advanced
|
||||
Programming in the UNIX Environment" for details (ISBN 0201563177)
|
||||
http://www.erlenstar.demon.co.uk/unix/faq_2.html#SEC16
|
||||
"""
|
||||
try:
|
||||
pid = os.fork()
|
||||
if pid > 0:
|
||||
# exit first parent
|
||||
sys.exit(0)
|
||||
except OSError, e:
|
||||
sys.stderr.write("fork #1 failed: %d (%s)\n" % (e.errno, e.strerror))
|
||||
sys.exit(1)
|
||||
|
||||
# decouple from parent environment
|
||||
os.chdir("/")
|
||||
os.setsid()
|
||||
os.umask(0)
|
||||
|
||||
# do second fork
|
||||
try:
|
||||
pid = os.fork()
|
||||
if pid > 0:
|
||||
# exit from second parent
|
||||
sys.exit(0)
|
||||
except OSError, e:
|
||||
sys.stderr.write("fork #2 failed: %d (%s)\n" % (e.errno, e.strerror))
|
||||
sys.exit(1)
|
||||
|
||||
# redirect standard file descriptors
|
||||
sys.stdout.flush()
|
||||
sys.stderr.flush()
|
||||
si = file(self.stdin, 'r')
|
||||
so = file(self.stdout, 'a+')
|
||||
se = file(self.stderr, 'a+', 0)
|
||||
os.dup2(si.fileno(), sys.stdin.fileno())
|
||||
os.dup2(so.fileno(), sys.stdout.fileno())
|
||||
os.dup2(se.fileno(), sys.stderr.fileno())
|
||||
|
||||
# write pidfile
|
||||
atexit.register(self.delpid)
|
||||
pid = str(os.getpid())
|
||||
file(self.pidfile, 'w+').write("%s\n" % pid)
|
||||
|
||||
def delpid(self):
|
||||
os.remove(self.pidfile)
|
||||
|
||||
def start(self):
|
||||
"""
|
||||
Start the daemon
|
||||
"""
|
||||
# Check for a pidfile to see if the daemon already runs
|
||||
try:
|
||||
pf = file(self.pidfile, 'r')
|
||||
pid = int(pf.read().strip())
|
||||
pf.close()
|
||||
except IOError:
|
||||
pid = None
|
||||
|
||||
if pid:
|
||||
message = "pidfile %s already exist. daemon already running?\n"
|
||||
sys.stderr.write(message % self.pidfile)
|
||||
sys.exit(1)
|
||||
|
||||
# Start the daemon
|
||||
self.daemonize()
|
||||
self.run()
|
||||
|
||||
def stop(self):
|
||||
"""
|
||||
Stop the daemon
|
||||
"""
|
||||
# Get the pid from the pidfile
|
||||
try:
|
||||
pf = file(self.pidfile, 'r')
|
||||
pid = int(pf.read().strip())
|
||||
pf.close()
|
||||
except IOError:
|
||||
pid = None
|
||||
|
||||
if not pid:
|
||||
message = "pidfile %s does not exist. daemon not running?\n"
|
||||
sys.stderr.write(message % self.pidfile)
|
||||
return # not an error in a restart
|
||||
|
||||
# Try killing the daemon process
|
||||
try:
|
||||
while 1:
|
||||
os.kill(pid, SIGTERM)
|
||||
time.sleep(0.1)
|
||||
except OSError, err:
|
||||
err = str(err)
|
||||
if err.find("No such process") > 0:
|
||||
if os.path.exists(self.pidfile):
|
||||
os.remove(self.pidfile)
|
||||
else:
|
||||
print str(err)
|
||||
sys.exit(1)
|
||||
|
||||
def restart(self):
|
||||
"""
|
||||
Restart the daemon
|
||||
"""
|
||||
self.stop()
|
||||
self.start()
|
||||
|
||||
def run(self):
|
||||
"""
|
||||
You should override this method when you subclass Daemon.
|
||||
It will be called after the process has been
|
||||
daemonized by start() or restart().
|
||||
"""
|
||||
12
lib/core/job.py
Normal file
12
lib/core/job.py
Normal file
|
|
@ -0,0 +1,12 @@
|
|||
"""
|
||||
This is what job_function needs as parameter for each job to succesfully
|
||||
execute.
|
||||
"""
|
||||
|
||||
|
||||
class Job:
|
||||
def __init__(self, command=None, members=None, host=None, service=None):
|
||||
self.command = command
|
||||
self.members = members
|
||||
self.host = host
|
||||
self.service = service
|
||||
16
lib/core/linspector_daemon.py
Normal file
16
lib/core/linspector_daemon.py
Normal file
|
|
@ -0,0 +1,16 @@
|
|||
from ..core import logger
|
||||
from ..core.daemon import Daemon
|
||||
|
||||
|
||||
class LinspectorDaemon(Daemon):
|
||||
def run(self):
|
||||
while True:
|
||||
try:
|
||||
a = 2
|
||||
logger.writeLogToFile(_logfile, "Running!")
|
||||
print "running!"
|
||||
except Exception as err:
|
||||
#logger.writeLogToFile(_logfile, str(err))
|
||||
print "failed"
|
||||
sys.exit(1)
|
||||
time.sleep(1)
|
||||
40
lib/core/logger.py
Normal file
40
lib/core/logger.py
Normal file
|
|
@ -0,0 +1,40 @@
|
|||
from datetime import datetime
|
||||
|
||||
DEBUG = "[debug]"
|
||||
NOTICE = "[notice]"
|
||||
WARNING = "[warning]"
|
||||
|
||||
|
||||
def logVerbose(message, verbose=True):
|
||||
if verbose:
|
||||
print DEBUG + " " + str(message)
|
||||
|
||||
|
||||
def logNotice(message, verbose=True):
|
||||
if verbose:
|
||||
print NOTICE + " " + str(message)
|
||||
|
||||
|
||||
def logWarning(message):
|
||||
print WARNING + " " + message
|
||||
|
||||
|
||||
def logWarningConfig(file="file", missing="missing"):
|
||||
logWarning("in " + file + ": The " + missing + " is not defined")
|
||||
|
||||
|
||||
def writeLogToFile(logfile, message):
|
||||
f = open(logfile, 'a')
|
||||
f.write("[" + str(datetime.now()) + "] " + message + '\n')
|
||||
f.close()
|
||||
|
||||
|
||||
class Logger:
|
||||
def __init__(self, logfile="/dev/null"):
|
||||
self.logfile = logfile
|
||||
|
||||
def logSomething(message, verbose=False):
|
||||
f = open(self.logfile, 'a')
|
||||
f.write("[" + str(datetime.now()) + "] " + message + '\n')
|
||||
f.close()
|
||||
|
||||
25
lib/core/logger.py.orig
Normal file
25
lib/core/logger.py.orig
Normal file
|
|
@ -0,0 +1,25 @@
|
|||
from datetime import datetime
|
||||
|
||||
DEBUG = "[debug]"
|
||||
NOTICE = "[notice]"
|
||||
WARNING = "[warning]"
|
||||
|
||||
def logVerbose(message, verbose = False):
|
||||
if verbose:
|
||||
print DEBUG + " " + str(message)
|
||||
|
||||
def logNotice(message, verbose = False):
|
||||
if verbose:
|
||||
print NOTICE + " " + str(message)
|
||||
|
||||
def logWarning(message):
|
||||
print WARNING + " " + message
|
||||
|
||||
def logWarningConfig(file = "file", missing = "missing"):
|
||||
logWarning("in " + file + ": The " + missing + " is not defined")
|
||||
|
||||
def writeLogToFile(logfile, message):
|
||||
f = open(logfile, 'a')
|
||||
f.write("[" + str(datetime.now()) + "] " + message + '\n')
|
||||
f.close()
|
||||
|
||||
Loading…
Add table
Add a link
Reference in a new issue