rhodecode/lib/dbmigrate/versions/034_version_2_3_0.py

import logging
import datetime

from sqlalchemy import *
from sqlalchemy.exc import DatabaseError
from sqlalchemy.orm import relation, backref, class_mapper, joinedload
from sqlalchemy.orm.session import Session
from sqlalchemy.ext.declarative import declarative_base

from rhodecode.lib.dbmigrate.migrate import *
from rhodecode.lib.dbmigrate.migrate.changeset import *
from rhodecode.lib.utils2 import str2bool

from rhodecode.model.meta import Base
from rhodecode.model import meta
from rhodecode.lib.dbmigrate.versions import _reset_base, notify

log = logging.getLogger(__name__)


def get_by_name(cls, key):
    return cls.query().filter(cls.app_settings_name == key).scalar()


def get_by_name_or_create(cls, key, val='', type='unicode'):
    res = get_by_name(cls, key)
    if not res:
        res = cls(key, val, type)
    return res


def upgrade(migrate_engine):
    """
    Upgrade operations go here.
    Don't create your own engine; bind migrate_engine to your metadata
    """
    _reset_base(migrate_engine)
    from rhodecode.lib.dbmigrate.schema import db_2_3_0_0

    # issue fixups
    fixups(db_2_3_0_0, meta.Session)


def downgrade(migrate_engine):
    meta = MetaData()
    meta.bind = migrate_engine


def fixups(models, _SESSION):
    notify('Fixing existing GA code into new format')
    cur_code = get_by_name(models.RhodeCodeSetting, 'ga_code')
    val = '''
<script>
 // google analytics
 var _gaq_code = '_GACODE_';
 var _gaq = _gaq || [];
 _gaq.push(['_setAccount', _gaq_code]);
 _gaq.push(['_trackPageview']);

 (function() {
  var ga = document.createElement('script'); ga.type = 'text/javascript'; ga.async = true;
  ga.src = ('https:' == document.location.protocol ? 'https://ssl' : 'http://www') + '.google-analytics.com/ga.js';
  var s = document.getElementsByTagName('script')[0]; s.parentNode.insertBefore(ga, s);
 })();

 rhodecode_statechange_callback = function(url, data){
  // ANALYTICS callback on html5 history state changed
  // triggered by file browser, url is the new url,
  // data is extra info passed from the State object
  if ( typeof window._gaq !== 'undefined' ) {
    _gaq.push(['_trackPageview', url]);
  }
 }
</script>'''
    if cur_code and getattr(cur_code, 'app_settings_value', ''):
        cur_val = getattr(cur_code, 'app_settings_value', '')
        val = val.replace('_GACODE_', cur_val)
        notify('Found GA code %s, migrating' % cur_val)
        new = get_by_name_or_create(models.RhodeCodeSetting, 'pre_code', val)
        new.app_settings_value = val
        _SESSION().add(new)
        _SESSION().commit()