diff --git a/planetmint/__init__.py b/planetmint/__init__.py index a3635f5..c15b269 100644 --- a/planetmint/__init__.py +++ b/planetmint/__init__.py @@ -6,7 +6,6 @@ import copy import logging import os -from planetmint.migrations.chain_migration_election import ChainMigrationElection from planetmint.log import DEFAULT_LOGGING_CONFIG as log_config from planetmint.version import __version__ # noqa @@ -107,6 +106,7 @@ from planetmint.common.transaction import Transaction # noqa from planetmint import models # noqa from planetmint.upsert_validator import ValidatorElection # noqa from planetmint.elections.vote import Vote # noqa +from planetmint.migrations.chain_migration_election import ChainMigrationElection Transaction.register_type(Transaction.CREATE, models.Transaction) Transaction.register_type(Transaction.TRANSFER, models.Transaction) diff --git a/planetmint/backend/localmongodb/connection.py b/planetmint/backend/localmongodb/connection.py index 995bc04..294e4ce 100644 --- a/planetmint/backend/localmongodb/connection.py +++ b/planetmint/backend/localmongodb/connection.py @@ -4,7 +4,7 @@ # Code is Apache-2.0 and docs are CC-BY-4.0 import logging -import bigchaindb +import planetmint from ssl import CERT_REQUIRED import pymongo @@ -42,7 +42,7 @@ class Connection: configuration's ``database`` settings """ - dbconf = bigchaindb.config['database'] + dbconf = planetmint.config['database'] self.host = host or dbconf['host'] self.port = port or dbconf['port'] diff --git a/planetmint/backend/localmongodb/query.py b/planetmint/backend/localmongodb/query.py index 217a775..1d51cb1 100644 --- a/planetmint/backend/localmongodb/query.py +++ b/planetmint/backend/localmongodb/query.py @@ -7,11 +7,11 @@ from pymongo import DESCENDING -from bigchaindb import backend -from bigchaindb.backend.exceptions import DuplicateKeyError -from bigchaindb.backend.utils import module_dispatch_registrar -from bigchaindb.backend.localmongodb.connection import LocalMongoDBConnection -from bigchaindb.common.transaction import Transaction +from planetmint import backend +from planetmint.backend.exceptions import DuplicateKeyError +from planetmint.backend.utils import module_dispatch_registrar +from planetmint.backend.localmongodb.connection import LocalMongoDBConnection +from planetmint.common.transaction import Transaction register_query = module_dispatch_registrar(backend.query)