render.py

"""
Re-render embed pages for storymaps with problematic id chars
"""
import importlib
import re
import os
import sys

# Import settings module
if __name__ == "__main__":
    if not os.environ.get('FLASK_SETTINGS_MODULE', ''):
        os.environ['FLASK_SETTINGS_MODULE'] = 'core.settings.loc'

settings_module = os.environ.get('FLASK_SETTINGS_MODULE')

print 'Importing settings: %s' % settings_module

try:
    importlib.import_module(settings_module)
except ImportError, e:
    raise ImportError("Could not import settings '%s' (Is it on sys.path?): %s" % (settings_module, e))

print 'Connecting to database'

from storymap.connection import _user
from storymap import storage
from api import app, _write_embed_draft, _write_embed_published
from flask import g

# Find all storymaps with ids that contain bad chars

print 'Examining StoryMaps'

regex = re.compile('[^A-Za-z0-9_\-]')

skip = 0
limit = 100
n = _user.count()

while skip < n:
    for user in _user.find({}, skip=skip, limit=limit):
        for id in [id for id in user['storymaps'].keys() if regex.search(id)]:        
            key_prefix = storage.key_prefix(user['uid'], id)  
            
            with app.test_request_context():
                print 'Writing embed draft for %s' % key_prefix                
                _write_embed_draft(key_prefix, user['storymaps'][id])
            
                if user['storymaps'][id].get('published_on'):
                    print 'Writing embed published for %s' % key_prefix                
                    _write_embed_published(key_prefix, user['storymaps'][id])
                            
    skip += limit   
    
sys.exit(0)