def __init__(self, dic, vm_name):
dic['results'] = {}
self.job_dic = dic
self.proceed = True
self.job_dic['dest_vm'] = vm_name
self.name = '_'.join( [vm_name, dic.get('sample_name')] )
out = '.'.join([self.job_dic.get('dest_vm') + '-sp3', 'v5-msvc10', self.job_dic.get('sample_name'), 'out'])
err = '.'.join([self.job_dic.get('dest_vm') + '-sp3', 'v5-msvc10', self.job_dic.get('sample_name'), 'err'])
oldmask = os.umask(0007)
log_dir = os.path.join(settings.get('LOG_DIR'), dic.get('job'))
if not os.path.exists(log_dir):
try:
os.makedirs(log_dir)
except os.error:
self.proceed = False
os.umask(oldmask)
self.job_dic['out_path'] = os.path.join(log_dir, out)
self.job_dic['err_path'] = os.path.join(log_dir, err)
self.log_path = os.path.join(log_dir, dic.get('err_path'))
fileFormat = logging.Formatter(fmt='%(asctime)s:%(name)s:%(levelname)s:%(message)s', datefmt='%m/%d/%Y %I:%M:%S %p')
logHandler = logging.FileHandler(self.log_path)
logHandler.setLevel(logging.DEBUG)
logHandler.setFormatter(fileFormat)
self.logger = logging.getLogger(self.name)
self.logger.addHandler(logHandler)
self.log("Task created, init log")
def setParsed(self, tweet_id, val=1):
if SETTINGS.get('debug', False) == True:
return False
try:
query = "UPDATE tracker_tweets SET parsed = %s WHERE tweet_id = %s"
self.queryDB(query, (val, tweet_id))
except Exception, e:
raise Exception("Setting parsed flag for tweet %s failed: %s" % (tweet_id, e))
def setParsed(self, tweet_id, val=1):
if SETTINGS.get('debug', False) == True:
return False
try:
query = "UPDATE tracker_tweets SET parsed = %s WHERE tweet_id = %s"
self.queryDB(query, (val, tweet_id))
except Exception, e:
raise Exception("Setting parsed flag for tweet %s failed: %s" %
(tweet_id, e))
def sendTweet(self, message):
if SETTINGS.get('tweet', False) is True:
try:
self.TW.update_status(message)
except:
self.logError("Tweeting message failed (%s)" % message)
def sendTweet(self, message):
if SETTINGS.get('tweet', False) is True:
try:
self.TW.update_status(message)
except:
self.logError("Tweeting message failed (%s)" % message)
def logDebug(self, message):
if SETTINGS.get('debug', False) is True:
self.log.info('[Debug]: ' + message)
def logDebug(self, message):
if SETTINGS.get('debug', False) is True:
self.log.info('[Debug]: ' + message)
# Checks for valid method in all the imported modules then executes it
def launch_method(self, method, job_dict):
if method in imported_functions:
result = imported_functions[method](job_dict)
else:
print 'Method not found'
result = 'Method not found'
return result
########################################
# Boiler Plate Main
if __name__ == '__main__':
host_addr = SETTINGS.get('HOST_ADDR')
gateway = SETTINGS.get('LAN_GATEWAY')
voodo_port = SETTINGS.get('VOODO_PORT')
# Merge the module dicts together
for module in imported_modules:
imported_functions = dict(imported_functions.items() + module.items())
# Find the host's IP address by creating a test socket, connecting to gateway
test_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
if gateway:
try:
test_socket.connect( (gateway, 65000) )
except socket.error as error:
if error.errno == 9:
print 'Malformed gateway IP address (LAN_GATEWAY value) in config.py:',gateway
print 'Unable to initialize server.'
print error
