def mongo_upgrade():
"""Migrate mongodb schema and data."""
# Read reporting.conf to fetch mongo configuration.
config = Config(os.path.join("..", "..", "conf", "reporting.conf"))
# Run migration only if mongo is enabled as reporting module.
if config.mongodb.enabled:
host = config.mongodb.get("host", "127.0.0.1")
port = config.mongodb.get("port", 27017)
print "Mongo reporting is enabled, strarting mongo data migration."
# Connect.
try:
conn = Connection(host, port)
db = conn.cuckoo
except TypeError:
print "Mongo connection port must be integer"
sys.exit()
except ConnectionFailure:
print "Cannot connect to MongoDB"
sys.exit()
# Check for schema version and create it.
if "cuckoo_schema" in db.collection_names():
print "Mongo schema version not expected"
sys.exit()
else:
db.cuckoo_schema.save({"version": mongo_revision})
else:
print "Mongo reporting module not enabled, skipping mongo migration."
def mongo_upgrade():
"""Migrate mongodb schema and data."""
# Read reporting.conf to fetch mongo configuration.
config = Config(os.path.join("..", "..", "conf", "reporting.conf"))
# Run migration only if mongo is enabled as reporting module.
if config.mongodb.enabled:
host = config.mongodb.get("host", "127.0.0.1")
port = config.mongodb.get("port", 27017)
print "Mongo reporting is enabled, strarting mongo data migration."
# Connect.
try:
conn = Connection(host, port)
db = conn.cuckoo
except TypeError:
print "Mongo connection port must be integer"
sys.exit()
except ConnectionFailure:
print "Cannot connect to MongoDB"
sys.exit()
# Check for schema version and create it.
if "cuckoo_schema" in db.collection_names():
print "Mongo schema version not expected"
sys.exit()
else:
db.cuckoo_schema.save({"version": mongo_revision})
else:
print "Mongo reporting module not enabled, skipping mongo migration."
def mongo_upgrade():
"""Migrate mongodb schema and data."""
# Read reporting.conf to fetch mongo configuration.
config = Config(cfg=os.path.join("..", "..", "conf", "reporting.conf"))
# Run migration only if mongo is enabled as reporting module.
if config.mongodb.enabled:
host = config.mongodb.get("host", "127.0.0.1")
port = config.mongodb.get("port", 27017)
database = config.mongodb.get("db", "cuckoo")
user = config.mongodb.get("user", None)
password = config.mongodb.get("password", None)
print "Mongo reporting is enabled, strarting mongo data migration."
if not port.isnumber():
print "Port must be an integer"
sys.exit()
# Support old Mongo.
try:
from pymongo.connection import Connection
from pymongo.errors import ConnectionFailure
conn = Connection(host, port)
db = conn.cuckoo
done = True
except ImportError:
print "Unable to import pymongo (install with `pip install pymongo`)"
done = False
except ConnectionFailure:
print "Cannot connect to MongoDB"
sys.exit()
try:
if not done:
import pymongo
try:
db = pymongo.MongoClient(host,
port=port,
username=user,
password=password,
authSource=database)[database]
except pymongo.errors.ConnectionFailure:
print "Cannot connect to MongoDB"
sys.exit()
except ImportError:
print "Unable to import pymongo (install with `pip install pymongo`)"
sys.exit()
# Check for schema version and create it.
if "cuckoo_schema" in db.collection_names():
print "Mongo schema version not expected"
sys.exit()
else:
db.cuckoo_schema.save({"version": mongo_revision})
else:
print "Mongo reporting module not enabled, skipping mongo migration."
def mongo_upgrade():
"""Migrate mongodb schema and data."""
# Read reporting.conf to fetch mongo configuration.
config = Config(cfg=os.path.join("..", "..", "conf", "reporting.conf"))
# Run migration only if mongo is enabled as reporting module.
if config.mongodb.enabled:
host = config.mongodb.get("host", "127.0.0.1")
port = config.mongodb.get("port", 27017)
database = config.mongodb.get("db", "cuckoo")
print "Mongo reporting is enabled, strarting mongo data migration."
if not port.isnumber():
print "Port must be an integer"
sys.exit()
# Support old Mongo.
try:
from pymongo.connection import Connection
from pymongo.errors import ConnectionFailure
conn = Connection(host, port)
db = conn.cuckoo
done = True
except ImportError:
print "Unable to import pymongo (install with `pip install pymongo`)"
done = False
except ConnectionFailure:
print "Cannot connect to MongoDB"
sys.exit()
try:
if not done:
import pymongo
try:
db = pymongo.MongoClient(host, port)[database]
except pymongo.errors.ConnectionFailure:
print "Cannot connect to MongoDB"
sys.exit()
except ImportError:
print "Unable to import pymongo (install with `pip install pymongo`)"
sys.exit()
# Check for schema version and create it.
if "cuckoo_schema" in db.collection_names():
print "Mongo schema version not expected"
sys.exit()
else:
db.cuckoo_schema.save({"version": mongo_revision})
else:
print "Mongo reporting module not enabled, skipping mongo migration."
