1
0
Fork 0
mirror of https://github.com/ThomasGsp/HyperProxmox.git synced 2025-02-13 03:32:16 +00:00

errors management

This commit is contained in:
Tlams 2018-02-09 18:01:58 +00:00
parent 986a14d643
commit be0f9fac8f

View file

@ -4,6 +4,7 @@ import json
import redis import redis
import time import time
class Redis_wrapper: class Redis_wrapper:
def __init__(self, server="127.0.0.1", port=6379, db=0, password=None): def __init__(self, server="127.0.0.1", port=6379, db=0, password=None):
# DB = # DB =
@ -24,10 +25,9 @@ class Redis_wrapper:
host=self.server, port=self.port, db=self.db, password=self.password, host=self.server, port=self.port, db=self.db, password=self.password,
charset="utf-8", decode_responses=True) charset="utf-8", decode_responses=True)
self.r.client_list() self.r.client_list()
return conn
except BaseException as err: except BaseException as err:
print("Redis connexion error on {0}:{1} ({2})".format(self.server, self.port, err)) print("Redis connexion error on {0}:{1} ({2})".format(self.server, self.port, err))
conn = False
return conn
def insert_instances_queue(self, logtext, expir=3000): def insert_instances_queue(self, logtext, expir=3000):
self.r.set(time.time(), logtext, expir) self.r.set(time.time(), logtext, expir)
@ -39,8 +39,18 @@ class Redis_wrapper:
self.r.set(key, value, expir) self.r.set(key, value, expir)
def get_message(self, key): def get_message(self, key):
return self.r.get(key) try:
result = {
"result": "OK",
"value": self.r.get(key)
}
except BaseException as e:
result = {
"result": "ERROR",
"type": "Redis - Request on get_message",
"value": "Invalid request: {0}".format(e)
}
return result
class MongoDB: class MongoDB:
def __init__(self, server="127.0.0.1", port=27017): def __init__(self, server="127.0.0.1", port=27017):
@ -66,17 +76,15 @@ class MongoDB:
try: try:
conn = MongoClient(self.server + ':' + str(self.port)) conn = MongoClient(self.server + ':' + str(self.port))
conn.server_info() conn.server_info()
return conn
except BaseException as err: except BaseException as err:
print("MongoDB connexion error on {0}:{1} ({2})".format(self.server, self.port, err)) print("MongoDB connexion error on {0}:{1} ({2})".format(self.server, self.port, err))
conn = False
return conn
def authenticate(self, user=None, password=None, mechanism='SCRAM-SHA-1'): def authenticate(self, user=None, password=None, mechanism='SCRAM-SHA-1'):
try: try:
self.client.db.authenticate(user, password, mechanism) self.client.db.authenticate(user, password, mechanism)
except (TypeError, ValueError) as e: except (TypeError, ValueError) as e:
raise("MongoDB authentification error on {0}:{1} ({2})".format(self.server, self.port, e)) print("MongoDB authentification error on {0}:{1} ({2})".format(self.server, self.port, e))
def generalmongosearch(self, collection, id): def generalmongosearch(self, collection, id):
try: try:
@ -87,6 +95,7 @@ class MongoDB:
except BaseException as e: except BaseException as e:
result = { result = {
"result": "ERROR", "result": "ERROR",
"type": "MongoDB - Request on generalmongosearch",
"value": "Invalid request: {0}".format(e) "value": "Invalid request: {0}".format(e)
} }
return result return result
@ -111,15 +120,25 @@ class MongoDB:
except BaseException as serr: except BaseException as serr:
result = { result = {
"result": "ERROR", "result": "ERROR",
"type": "MongoDB - Request on get_clusters",
"value": "MongoDB error on {0}:{1} ({2})".format(self.server, self.port, serr) "value": "MongoDB error on {0}:{1} ({2})".format(self.server, self.port, serr)
} }
return result return result
def insert_clusters(self, data): def insert_clusters(self, data):
try: try:
return self.db[self.collection_clusters].insert(data) result = {
"result": "OK",
"value": self.db[self.collection_clusters].insert(data)
}
except BaseException as serr: except BaseException as serr:
raise ("MongoDB error on {0}:{1} ({2})".format(self.server, self.port, serr)) result = {
"result": "ERROR",
"type": "MongoDB - Request on insert_clusters",
"value": "MongoDB error on {0}:{1} ({2})".format(self.server, self.port, serr)
}
return result
def get_clusters_conf(self, cluster=None): def get_clusters_conf(self, cluster=None):
try: try:
@ -136,6 +155,7 @@ class MongoDB:
except BaseException as e: except BaseException as e:
result = { result = {
"result": "ERROR", "result": "ERROR",
"type": "MongoDB - Request on get_cluster_conf",
"value": "Invalid request: {0}".format(e) "value": "Invalid request: {0}".format(e)
} }
return result return result
@ -150,6 +170,7 @@ class MongoDB:
except BaseException as e: except BaseException as e:
result = { result = {
"result": "ERROR", "result": "ERROR",
"type": "MongoDB - Request on insert_cluster_conf",
"value": "Invalid request: {0}".format(e) "value": "Invalid request: {0}".format(e)
} }
return result return result
@ -164,6 +185,7 @@ class MongoDB:
except BaseException as e: except BaseException as e:
result = { result = {
"result": "ERROR", "result": "ERROR",
"type": "MongoDB - Request on update_cluster_conf",
"value": "Invalid request: {0}".format(e) "value": "Invalid request: {0}".format(e)
} }
return result return result
@ -178,6 +200,7 @@ class MongoDB:
except BaseException as e: except BaseException as e:
result = { result = {
"result": "ERROR", "result": "ERROR",
"type": "MongoDB - Request on delete_cluster_conf",
"value": "Invalid request: {0}".format(e) "value": "Invalid request: {0}".format(e)
} }
return result return result
@ -219,9 +242,17 @@ class MongoDB:
""" NODES MANAGEMENT""" """ NODES MANAGEMENT"""
def insert_nodes(self, data): def insert_nodes(self, data):
try: try:
return self.db[self.collection_nodes].insert(data) result = {
"result": "OK",
"value": self.db[self.collection_nodes].insert(data)
}
except BaseException as serr: except BaseException as serr:
raise ("MongoDB error on {0}:{1} ({2})".format(self.server, self.port, serr)) result = {
"result": "ERROR",
"type": "MongoDB - Request on insert_instances",
"value": "MongoDB error on {0}:{1} ({2})".format(self.server, self.port, serr)
}
return result
def get_nodes(self, date, cluster, node, grata=0): def get_nodes(self, date, cluster, node, grata=0):
try: try:
@ -251,6 +282,7 @@ class MongoDB:
except BaseException as serr: except BaseException as serr:
result = { result = {
"result": "ERROR", "result": "ERROR",
"type": "MongoDB - Request on get_nodes",
"value": "MongoDB error on {0}:{1} ({2})".format(self.server, self.port, serr) "value": "MongoDB error on {0}:{1} ({2})".format(self.server, self.port, serr)
} }
return result return result
@ -258,9 +290,18 @@ class MongoDB:
""" INSTANCE MANAGEMENT""" """ INSTANCE MANAGEMENT"""
def insert_instances(self, data): def insert_instances(self, data):
try: try:
return self.db[self.collection_instances].insert(data) result = {
"result": "OK",
"value": self.db[self.collection_instances].insert(data)
}
except BaseException as serr: except BaseException as serr:
raise ("MongoDB error on {0}:{1} ({2})".format(self.server, self.port, serr)) result = {
"result": "ERROR",
"type": "MongoDB - Request on insert_instances",
"value": "MongoDB error on {0}:{1} ({2})".format(self.server, self.port, serr)
}
return result
def get_instances(self, date, cluster, node, vmid): def get_instances(self, date, cluster, node, vmid):
try: try:
@ -295,6 +336,7 @@ class MongoDB:
except BaseException as serr: except BaseException as serr:
result = { result = {
"result": "ERROR", "result": "ERROR",
"type": "MongoDB - Request on get_instances",
"value": "MongoDB error on {0}:{1} ({2})".format(self.server, self.port, serr) "value": "MongoDB error on {0}:{1} ({2})".format(self.server, self.port, serr)
} }
return result return result
@ -303,9 +345,17 @@ class MongoDB:
""" STORAGE MANAGEMENT""" """ STORAGE MANAGEMENT"""
def insert_storages(self, data): def insert_storages(self, data):
try: try:
return self.db[self.collection_storages].insert(data) result = {
"result": "OK",
"value": self.db[self.collection_storages].insert(data)
}
except BaseException as serr: except BaseException as serr:
raise ("MongoDB error on {0}:{1} ({2})".format(self.server, self.port, serr)) result = {
"result": "ERROR",
"type": "MongoDB - Request on insert_storages",
"value": "MongoDB error on {0}:{1} ({2})".format(self.server, self.port, serr)
}
return result
def get_storages(self, date, cluster, node): def get_storages(self, date, cluster, node):
try: try:
@ -332,6 +382,7 @@ class MongoDB:
except BaseException as serr: except BaseException as serr:
result = { result = {
"result": "ERROR", "result": "ERROR",
"type": "MongoDB - Request on get_storages",
"value": "MongoDB error on {0}:{1} ({2})".format(self.server, self.port, serr) "value": "MongoDB error on {0}:{1} ({2})".format(self.server, self.port, serr)
} }
return result return result
@ -341,9 +392,17 @@ class MongoDB:
""" DISKS MANAGEMENT""" """ DISKS MANAGEMENT"""
def insert_disks(self, data): def insert_disks(self, data):
try: try:
return self.db[self.collection_disks].insert(data) result = {
"result": "OK",
"value": self.db[self.collection_disks].insert(data)
}
except BaseException as serr: except BaseException as serr:
raise ("MongoDB error on {0}:{1} ({2})".format(self.server, self.port, serr)) result = {
"result": "ERROR",
"type": "MongoDB - Request on get_storages",
"value": "MongoDB error on {0}:{1} ({2})".format(self.server, self.port, serr)
}
return result
def get_disks(self, date, cluster, node, vmid): def get_disks(self, date, cluster, node, vmid):
try: try:
@ -378,6 +437,7 @@ class MongoDB:
except BaseException as serr: except BaseException as serr:
result = { result = {
"result": "ERROR", "result": "ERROR",
"type": "MongoDB - Request on get_disks",
"value": "MongoDB error on {0}:{1} ({2})".format(self.server, self.port, serr) "value": "MongoDB error on {0}:{1} ({2})".format(self.server, self.port, serr)
} }
return result return result