from newhive import state, config
from newhive.utils import *
import unittest
db = state.Database(config)
def mongo_total(col):
"""return total size of a mongo collection"""
return col.search({}).count()
def mongo_last(col):
return col.search({}).sort([('updated', -1)])
def esdb_last(es_type):
updated = db.esdb.conn.search(match_all_query,
indices=db.esdb.index,
doc_types=es_type,
sort="updated:desc")
return updated
def clear_recent(t):
for i in range(2):
ml = mongo_last(db.Expr)
ml[0].delete()
db.esdb.purge_deleted(t)
import re
from newhive import state
db = state.Database()
from newhive.utils import now, time_u, Apply
import urllib
import os
def migrate():
return Apply.apply_all(add_tag_cloud, db.User.search({}))
def add_tag_cloud(user):
user.calculate_tags()
return True
def dbs(name):
conf = __import__('newhive.config.' + name, fromlist=['newhive', 'config'])
return state.Database(conf)
hive_assets = HiveAssets()
hive_assets.bundle()
jinja_env = jinja2.Environment(
loader=jinja2.FileSystemLoader(os.path.join(config.src_home, 'templates')))
jinja_env.trim_blocks = True
jinja_env.globals.update(asset_bundle=hive_assets.asset_bundle)
jinja_env.filters.update({
'asset_url': hive_assets.url,
'json': extra_json,
'param_esc': quote_plus
})
jinja_env.globals.update({
'colors': colors,
'asset_bundle': hive_assets.asset_bundle
})
db = state.Database(config, hive_assets)
server_env = {
'db': db,
'jinja_env': jinja_env,
'assets': hive_assets,
'config': config,
'controllers': {}
}
jinja_env.globals.update({'media_bucket': db.s3.bucket_url('media')})
controllers = Controllers(server_env)
server_env['controllers'].update(controllers)
