Synnefo » snf-cloudcms

import os

import glob

import logging

from django import forms

from django.conf import settings

    return MediaCategory.objects.get(slug=slug)

class RstZipImportForm(forms.Form):

        return data

    def clean_existing_data(self):

        Question.objects.all().delete()

        UserGuideEntry.objects.all().delete()

        MediaFile.objects.filter(categories__slug__in=[self.FAQ_MEDIA_CATEGORY, \

            self.GUIDE_MEDIA_CATEGORY]).delete()

    def save(self, user, use_dir=None):

        dry_run = self.cleaned_data.get('dry_run')

        clean_data = self.cleaned_data.get('clean_data')

        import_file = self.cleaned_data.get('import_file')

                subdirs[0])) and subdirs[0] != 'source':

            zipdir = os.path.join(zipdir, subdirs[0])

        if clean_data:

            try:

                self.clean_existing_data()

            except Exception, e:

        ret = ""

        try:

            for data_type, rst, category, slug, title, html_content, \

                    images, stderr in generate_rst_contents_from_dir(zipdir):

                if stderr:

                service = service_from_filename(rst)

                if not service:

                    logger.info("Skipping entry for file '%s'. No category found" % rst)

                    continue

                # first save media files

                newimages = []

                if data_type == 'userguide':

                    cat = get_media_category(self.GUIDE_MEDIA_CATEGORY)

                    cat = get_media_category(self.FAQ_MEDIA_CATEGORY)

                if not cat:

                    continue

                for imgname, imgpath, imgabspath in images:

                    newalt, newpath = create_or_update_media_file(cat, \

                            imgname, imgabspath)

                # now html contains correct image links, we are ready to save

                # the faq/guide content

                if data_type == 'faq':

                    cat = add_or_update_faq_category(category[0], category[1])

                    question = add_or_update_faq_question(user, service, cat, slug, \

                            title, html_content)

                if data_type == 'userguide':

                    guide_entry = add_or_update_guide_entry(user, service, slug, \

                            title, html_content)

        except Exception, e:

            logger.exception("RST import failed")

        if dry_run:

        else:

def create_or_update_media_file(category, name, path):

    name = category.title + " " + name

    try:

        # TODO: Check language ?????

    except MediaFile.DoesNotExist:

        mf = MediaFile()

        mf.file = File(open(path))

        mf.save()

        mf.translations.create(caption=name)

    # TODO: Check language ?????

    return mf.translations.all()[0].caption, mf.get_absolute_url()

def add_or_update_faq_category(slug, name):

    try:

        category = FaqCategory.objects.get(translations__slug=slug)

    except FaqCategory.DoesNotExist:

        category = FaqCategory()

        category.save()

        category.translations.create(slug=slug, title=name)

    return category

def add_or_update_faq_question(author, service, category, slug,\

        title, html_content):

    try:

        q = Question.objects.get(slug=slug)

    except:

        q = Question()

    q.author = author

    return q

def add_or_update_guide_entry(author, service, slug, title, html_content):

    try:

        guide = UserGuideEntry.objects.get(slug=slug)

    except:

        guide = UserGuideEntry()

    guide.author = author

    return guide