django-orchestra/orchestra/contrib/websites/backends/webalizer.py

import os
import textwrap

from django.utils.translation import gettext_lazy as _

from orchestra.contrib.orchestration import ServiceController
from orchestra.settings import NEW_SERVERS

from .. import settings


class WebalizerController(ServiceController):
    """
    Creates webalizer conf file for each time a webalizer webapp is mounted on a website.
    """
    verbose_name = _("Webalizer Content")
    model = 'websites.Content'
    default_route_match = "content.webapp.type == 'webalizer'"
    doc_settings = (settings,
        ('WEBSITES_WEBALIZER_PATH',)
    )
    
    def save(self, content):
        context = self.get_context(content)
        self.append(textwrap.dedent("""\
            mkdir -p %(webalizer_path)s
            if [[ ! -e %(webalizer_path)s/index.html ]]; then
                echo 'Webstats are coming soon' > %(webalizer_path)s/index.html
            fi
            cat << 'EOF' > %(webalizer_conf_path)s
            %(webalizer_conf)s
            EOF
            # chown %(user)s:www-data %(webalizer_path)s
            chown www-data:www-data %(webalizer_path)s
            chmod g+xr %(webalizer_path)s
            """) % context
        )
    
    def delete(self, content):
        context = self.get_context(content)
        delete_webapp = not type(content.webapp).objects.filter(pk=content.webapp.pk).exists()
        if delete_webapp:
            self.append("rm -fr %(webapp_path)s" % context)
        remounted = content.webapp.content_set.filter(website=content.website).exists()
        if delete_webapp or not remounted:
            self.append("rm -fr %(webalizer_path)s" % context)
            self.append("rm -f %(webalizer_conf_path)s" % context)
    
    def get_context(self, content):
        conf_file = "%s.conf" % content.website.unique_name
        context = {
            'site_logs': content.website.get_www_access_log_path(),
            'site_name': content.website.name,
            'webapp_path': content.webapp.get_path(),
            'webalizer_path': os.path.join(content.webapp.get_path(), content.website.name),
            'webalizer_conf_path': os.path.join(settings.WEBSITES_WEBALIZER_PATH, conf_file),
            'user': content.webapp.account.username,
            'banner': self.get_banner(),
            'target_server': content.website.target_server,
        }
        if context.get('target_server').name in NEW_SERVERS:
            context['webalizer_conf'] = textwrap.dedent("""\
                # %(banner)s
                LogFile            %(site_logs)s
                LogType            clf
                OutputDir          %(webalizer_path)s
                HistoryName        awffull.hist
                Incremental        yes
                IncrementalName    awffull.current
                ReportTitle        Stats of
                HostName           %(site_name)s
                """) % context
        else:
            context['webalizer_conf'] = textwrap.dedent("""\
                # %(banner)s
                LogFile            %(site_logs)s
                LogType            clf
                OutputDir          %(webalizer_path)s
                HistoryName        webalizer.hist
                Incremental        yes
                IncrementalName    webalizer.current
                ReportTitle        Stats of
                HostName           %(site_name)s                                                        
                """) % context
        
        context['webalizer_conf'] = context['webalizer_conf'] + textwrap.dedent("""\
                                                                                
            PageType       htm*
            PageType       php*
            PageType       shtml
            PageType       cgi
            PageType       pl
            
            DNSCache       /var/lib/dns_cache.db
            DNSChildren    15
            
            HideURL        *.gif
            HideURL        *.GIF
            HideURL        *.jpg
            HideURL        *.JPG
            HideURL        *.png
            HideURL        *.PNG
            HideURL        *.ra
            
            IncludeURL     *
                
            SearchEngine    google.         q=
            SearchEngine    yahoo.          p=
            SearchEngine    msn.            q=
            SearchEngine    search.aol      query=
            SearchEngine    altavista.      q=
            SearchEngine    lycos.          query=
            SearchEngine    hotbot.         query=
            SearchEngine    alltheweb.      query=
            SearchEngine    infoseek.       qt=
            SearchEngine    webcrawler      searchText=
            SearchEngine    excite          search=
            SearchEngine    netscape.       query=
            SearchEngine    ask.com         q=
            SearchEngine    webwombat.      ix=
            SearchEngine    earthlink.      q=
            SearchEngine    search.comcast. q=
            SearchEngine    search.mywebsearch.     searchfor=
            SearchEngine    reference.com   q=
            SearchEngine    mamma.com       query=
            # Last attempt catch all
            SearchEngine    search.         q=
            
            DumpSites      yes""") % context
        return context