def connect_mongo(self, branch):
"""Connect to mongo db. """
branch_dir = os.path.join(os.path.expanduser('~'), 'active', branch)
if not os.path.isdir(branch_dir):
raise Exception('Branch directory does not exist')
config_dir = os.path.join(branch_dir, 'asdf', 'src', 'utils')
sys.path.insert(0, config_dir)
from config_utility import BranchConfig
self.config = BranchConfig(branch=branch)
# -------------------------------------
# check mongodb connection
if self.config.connection_status != 0:
sys.exit("connection status error: " +
str(self.config.connection_error))
# -------------------------------------
self.client = pymongo.MongoClient(self.config.database)
self.asdf = self.client.asdf.data
self.is_connected = True
def connect_mongo(self, branch):
"""Connect to mongo db. """
from config_utility import BranchConfig
self.config = BranchConfig(branch=branch)
# -------------------------------------
# check mongodb connection
if self.config.connection_status != 0:
sys.exit("connection status error: " +
str(self.config.connection_error))
# -------------------------------------
self.client = pymongo.MongoClient(self.config.database)
self.asdf = self.client.asdf.data
self.is_connected = True
"""
# -----------------------------------------------------------------------------
import sys
import os
branch = sys.argv[1]
utils_dir = os.path.join(
os.path.dirname(os.path.dirname(os.path.abspath(__file__))), 'utils')
sys.path.insert(0, utils_dir)
from config_utility import BranchConfig
config = BranchConfig(branch=branch)
config.test_connection()
# -----------------------------------------------------------------------------
import geo_rasterstats as rs
# -------------------------------------
# initialize mpi
import mpi_utility
job = mpi_utility.NewParallel(capture=True)
# -------------------------------------
# verify good mongo connection on all processors
branch = sys.argv[1]
branch_dir = os.path.join(os.path.expanduser('~'), 'active', branch)
if not os.path.isdir(branch_dir):
raise Exception('Branch directory does not exist')
config_dir = os.path.join(branch_dir, 'asdf', 'src', 'utils')
sys.path.insert(0, config_dir)
from config_utility import BranchConfig
config_attempts = 0
while True:
config = BranchConfig(branch=branch)
config_attempts += 1
if config.connection_status == 0 or config_attempts > 5:
break
# -----------------------------------------------------------------------------
if config.connection_status != 0:
print "mongodb connection error ({0})".format(config.connection_error)
sys.exit()
import errno
# import pymongo
import zipfile
import re
# ----------------------------------------------------------------------------
import sys
import os
branch = sys.argv[1]
utils_dir = os.path.join(
os.path.dirname(os.path.dirname(os.path.abspath(__file__))), 'utils')
sys.path.insert(0, utils_dir)
from config_utility import BranchConfig
config = BranchConfig(branch=branch)
config.test_connection()
# ----------------------------------------------------------------------------
# check mongodb connection
if config.connection_status != 0:
print "error"
# sys.exit("connection status error: " + str(config.connection_error))
# ----------------------------------------------------------------------------
import pymongo
