|
@@ -7,9 +7,9 @@
|
|
|
from utils.params import PARAMS as SYS_ARGS
|
|
|
import requests
|
|
|
import pickle
|
|
|
+import json
|
|
|
from threading import Thread, RLock
|
|
|
ENDPOINT="https://dev.the-phi.com/monitor"
|
|
|
-
|
|
|
class Collector(Thread) :
|
|
|
def __init__(self):
|
|
|
Thread.__init__(self)
|
|
@@ -20,20 +20,44 @@ class Collector(Thread) :
|
|
|
@param id node identifier
|
|
|
|
|
|
"""
|
|
|
+ for id in ['apps','folders']:
|
|
|
+ if id in SYS_ARGS :
|
|
|
+ SYS_ARGS[id] = SYS_ARGS[id].split(',')
|
|
|
scope=list(set(['apps','folders','sandbox'])& set(SYS_ARGS.keys()))
|
|
|
- headers = {"key":SYS_ARGS["key"],"id":SYS_ARGS["id"],"scope":scope}
|
|
|
- r = requests.get("https://the-phi.com/monitor/init",headers=headers)
|
|
|
- self.pool = pickle.loads(r.text)
|
|
|
+ headers = {"key":SYS_ARGS["key"],"id":SYS_ARGS["id"],"scope":json.dumps(scope)}
|
|
|
+ headers['content-type'] = 'application/json'
|
|
|
+ try:
|
|
|
+ r = requests.post("https://the-phi.com/monitor/init/collector",headers=headers,data=json.dumps(SYS_ARGS))
|
|
|
+
|
|
|
+ r = json.loads(r.text)
|
|
|
+
|
|
|
+ self.monitor = pickle.loads(r[0])
|
|
|
+ self.monitor.lock = RLock()
|
|
|
+ #:w
|
|
|
+ #self.monitor.set('lock',RLock())
|
|
|
+ except Exception,e:
|
|
|
+ print e.message
|
|
|
+ self.monitor = None
|
|
|
|
|
|
def run(self):
|
|
|
- for key in self.pool :
|
|
|
- try:
|
|
|
- p = self.pool[key]
|
|
|
- p.init(SYS_ARGS[key])
|
|
|
- p.start()
|
|
|
- except Exception,e:
|
|
|
- print e
|
|
|
+ """
|
|
|
+ This funtion runs the authorized features and
|
|
|
+ """
|
|
|
+ #self.monitor.start()
|
|
|
+ thread = Thread(target=self.monitor.run)
|
|
|
+ thread.start()
|
|
|
+ # print self.monitor.config['store']
|
|
|
+ # for agent in self.pool :
|
|
|
+ # try:
|
|
|
+ # agent = pickle.loads(agent)
|
|
|
+ # agent.start()
|
|
|
+
|
|
|
+ # #p = pickle.loads(self.pool[key])
|
|
|
+ # #p.init(SYS_ARGS[key].split(','))
|
|
|
+ # #p.start()
|
|
|
+ # except Exception,e:
|
|
|
+ # print e
|
|
|
|
|
|
-if name == '__main__' :
|
|
|
+if __name__ == '__main__' :
|
|
|
thread = Collector()
|
|
|
thread.start()
|