from django.utils.encoding import smart_str

from kitsune.products.models import Product
from kitsune.taggit.models import Tag
from kitsune.wiki.models import Document

tags_to_migrate = {
    # source tag -> product
    'firefox': ['firefox'],
    'sync': ['firefox', 'mobile'],
    'persona': ['firefox'],
    'desktop': ['firefox'],
    'fxhome': ['firefox', 'mobile'],
    'firefox-10': ['firefox'],
    'firefox-602': ['firefox'],
    'firefox-50': ['firefox'],
    'android': ['mobile'],
    'mobile': ['mobile']
}


def assert_equals(a, b):
    assert a == b, '%s != %s' % (a, b)


def run():
    # Get all the tags to migrate.
    tags = list(Tag.objects.filter(slug__in=tags_to_migrate.keys()))

    total_affected = 0

    # For each tag, get the document and add a product for it.
    for tag in tags:
        for product_slug in tags_to_migrate[tag.slug]:
            product = Product.objects.get(slug=product_slug)

            # Assign the product to all the documents tagged with tag.
            for doc in Document.objects.filter(tags__slug=tag.slug):

                doc.products.add(product)

                print 'Added product "%s" to document "%s"' % (
                    smart_str(product.slug), smart_str(doc.title))
                total_affected += 1

    print 'Done! (%d)' % total_affected

from django.utils.encoding import smart_str

from kitsune.products.models import Product
from taggit.models import Tag
from kitsune.wiki.models import Document

tags_to_migrate = {
    # source tag -> product
    'firefox': ['firefox'],
    'sync': ['firefox', 'mobile'],
    'persona': ['firefox'],
    'desktop': ['firefox'],
    'fxhome': ['firefox', 'mobile'],
    'firefox-10': ['firefox'],
    'firefox-602': ['firefox'],
    'firefox-50': ['firefox'],
    'android': ['mobile'],
    'mobile': ['mobile']
}


def assert_equals(a, b):
    assert a == b, '%s != %s' % (a, b)


def run():
    # Get all the tags to migrate.
    tags = list(Tag.objects.filter(slug__in=tags_to_migrate.keys()))

    total_affected = 0

    # For each tag, get the document and add a product for it.
    for tag in tags:
        for product_slug in tags_to_migrate[tag.slug]:
            product = Product.objects.get(slug=product_slug)

            # Assign the product to all the documents tagged with tag.
            for doc in Document.objects.filter(tags__slug=tag.slug):

                doc.products.add(product)

                print 'Added product "%s" to document "%s"' % (
                    smart_str(product.slug), smart_str(doc.title))
                total_affected += 1

    print 'Done! (%d)' % total_affected

Fix wrong module name in migrations
Fix wrong module name in migrations
