def __init__(self):
client = pymongo.MongoClient(
host=Config.mongo_host(),
port=Config.mongo_port(),
)
self.graphs = Graphs(client)
self.scores = Scores(client)
self.nodesets = NodeSets(client)
self.runs = Runs(client)
self.subgraphs = Subgraphs(client)
def main():
app = connexion.App(__name__, specification_dir='./swagger/')
app.app.json_encoder = encoder.JSONEncoder
app.add_api('swagger.yaml', arguments={'title': 'DeRegNet REST API'})
app.app.config['ELASTIC_APM'] = {
# Set required service name. Allowed characters:
# a-z, A-Z, 0-9, -, _, and space
'SERVICE_NAME': 'deregnet-rest',
# Set custom APM Server URL (default: http://localhost:8200)
'SERVER_URL': 'http://apm-server:8200',
'DEBUG': DEBUG,
}
apm = ElasticAPM(app.app)
app.run(host=Config.host(),
port=Config.port(),
debug=DEBUG,
server=Config.server_backend())
def __init__(self, name):
self.redis = StrictRedis(
port=Config.redis_port(),
host=Config.redis_host(),
)
self._name = name
from pymongo import MongoClient
from deregnet_rest.config import Config
database = MongoClient(
host=Config.mongo_host(),
port=Config.mongo_port(),
)
node_id_attr,
run_id,
x_consumer_id,
)
self.runs.update_one(
filter={
'id': run_id,
'X-Consumer-ID': x_consumer_id,
},
update={
'$set': {
'done': True,
'subgraph_ids': subgraph_ids
},
},
)
REDIS_URI = 'redis://{}:{}/1'.format(Config.redis_host(), Config.redis_port())
celery = Celery('main', broker=REDIS_URI)
@celery.task(name='find-subgraph')
def find_subgraphs(run_id):
runner = Runner()
runner.run(run_id)
if __name__ == '__main__':
celery.start()
from pymongo import MongoClient
from deregnet_rest.config import Config
database = MongoClient(
host=Config.mongo_host(),
port=Config.mongo_port(),
username=Config.mongo_user(),
password=Config.mongo_password(),
)