from config.settings import *
from config.amqp import *
from modules.logger import Log
from modules.amqp import AMQPWorker
from modules.worker import WorkerThread, WorkerTasks
from netbox import Netbox
import argparse, sys, json, time, signal
import ssdp, requests, re, copy
from sets import Set
from urlparse import urlparse
log = Log(__name__)
worker_info = {}
amqp_info = {}
def shutdown(signum, stack):
for key, val in amqp_info.iteritems():
log.info('stopping ' + key)
val['amqp'].stop()
for key, val in worker_info.iteritems():
val['worker'].stop()
sys.exit(0)
def add_amqp_listener(amqp, name):
amqp_info[name] = {'amqp': amqp}
def add_worker(worker, task, name):
worker_info[name] = {'worker': worker, 'task': task}
from config.api2_0_config import *
from on_http_api2_0 import ApiApi as WorkflowApi
from on_http_api2_0 import rest
from modules.logger import Log
from datetime import datetime
from proboscis.asserts import *
from proboscis import SkipTest
from proboscis import test
from json import dumps, loads
import json
LOG = Log(__name__)
@test(groups=['workflowTasks_api2.tests'])
class WorkflowTasksTests(object):
def __init__(self):
self.__client = config.api_client
self.__workflows = None
self.workflowTaskDict = {
"friendlyName": "fn_1",
"injectableName": "in_1",
"implementsTask": "im_1",
"options": {},
"properties": {}
}
@test(groups=['workflowTasks_api2.tests', 'api2_workflowTasks_get'])
def test_workflowTasks__get(self):
""" Testing GET:/worflows/tasks"""
WorkflowApi().workflows_get_all_tasks()
from config.netbox_config import *
from modules.logger import Log
from modules.worker import WorkerThread, WorkerTasks
from netbox_rest import ApidcimApi as Dcim
from netbox_rest import ApicircuitsApi as Circuit
from netbox_rest import ApiipamApi as Ipam
from netbox_rest import ApisecretsApi as Secret
from netbox_rest import ApitenancyApi as Tenancy
import sys, json, time
import requests, re
import os
log = Log(__name__)
ROLE_COLOR_MAP = {
'compute': 'blue',
'switch': 'green',
'enclosure': 'orange',
'pdu': 'red'
}
def sanitize_slug(str):
for c in ['.', ',', '#', '+', '?', '$', '%', '!', '@', '\\', '/', '\"']:
str = str.replace(c, '')
return str.replace(' ', '-').lower()
class Netbox(object):
def __init__(self, *args, **kwargs):
# create session and login
