diff --git a/runhelper/db.py b/runhelper/db.py index 13faa23203a6f1053332b5ce522f56eeda75421d..5ef5d7cfd86b1bd2052b264ffb527a58fb9bd442 100644 --- a/runhelper/db.py +++ b/runhelper/db.py @@ -41,7 +41,14 @@ class Database(object): pass def __init__(self, file): - self.backend = blitzdb.FileBackend(file) + if file.startswith("mongo://"): + from pymongo import MongoClient + dbName = file[len("mongo://"):] + c = MongoClient() + self.backend = blitzdb.MongoBackend(c[dbName]) + else: + self.backend = blitzdb.FileBackend(file) + self.backend.autocommit = True @staticmethod