Synnefo

# Copyright 2011-2012 GRNET S.A. All rights reserved.

#

# Redistribution and use in source and binary forms, with or

# without modification, are permitted provided that the following

# conditions are met:

#

#   1. Redistributions of source code must retain the above

#      copyright notice, this list of conditions and the following

#      disclaimer.

#

#   2. Redistributions in binary form must reproduce the above

#      copyright notice, this list of conditions and the following

#      disclaimer in the documentation and/or other materials

#      provided with the distribution.

#

# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS

# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED

# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR

# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,

# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT

# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF

# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED

# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT

# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN

# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE

# POSSIBILITY OF SUCH DAMAGE.

#

# The views and conclusions contained in the software and

# documentation are those of the authors and should not be

# interpreted as representing official policies, either expressed

# or implied, of GRNET S.A.

import calendar

import datetime

import math

from collections import defaultdict

from django import template

from django.core.paginator import Paginator, EmptyPage

from django.db.models.query import QuerySet

from astakos.im.settings import PAGINATE_BY, RESOURCES_PRESENTATION_DATA

from astakos.im.models import RESOURCE_SEPARATOR

register = template.Library()

DELIM = ','

@register.filter

def monthssince(joined_date):

    now = datetime.datetime.now()

    date = datetime.datetime(

        year=joined_date.year, month=joined_date.month, day=1)

    months = []

    month = date.month

    year = date.year

    timestamp = calendar.timegm(date.utctimetuple())

    while date < now:

        months.append((year, month, timestamp))

        if date.month < 12:

            month = date.month + 1

            year = date.year

        else:

            month = 1

            year = date.year + 1

        date = datetime.datetime(year=year, month=month, day=1)

        timestamp = calendar.timegm(date.utctimetuple())

    return months

@register.filter

def to_unicode(s):

    return unicode(s)

@register.filter

def to_string(s):

    return str(s)

@register.filter

def lookup(d, key):

    try:

        return d.get(str(key))

    except:

        return

@register.filter

def lookup_uni(d, key):

    return d.get(unicode(key))

@register.filter

def dkeys(d):

    return d.keys()

@register.filter

def month_name(month_number):

    return calendar.month_name[month_number]

@register.filter

def todate(value, arg=''):

    secs = int(value) / 1000

    return datetime.datetime.fromtimestamp(secs)

# @register.filter

# def rcut(value, chars='/'):

#     return value.rstrip(chars)

@register.filter

def paginate(l, args):

    l = l or []

    page, delim, sorting = args.partition(DELIM)

    if sorting:

        if isinstance(l, QuerySet):

            l = l.order_by(sorting)

        elif isinstance(l, list):

            default = ''

            if sorting.endswith('_date'):

                default = datetime.datetime.utcfromtimestamp(0)

            l.sort(key=lambda i: getattr(i, sorting)

                   if getattr(i, sorting) else default)

    paginator = Paginator(l, PAGINATE_BY)

    try:

        paginator.len

    except AttributeError:

        paginator._count = len(list(l))

    try:

        page_number = int(page)

    except ValueError:

        if page == 'last':

            page_number = paginator.num_pages

        else:

            page_number = 1

    try:

        page = paginator.page(page_number)

    except EmptyPage:

        page = paginator.page(1)

    return page

@register.filter

def concat(str1, str2):

    if not str2:

        return str(str1)

    return '%s%s%s' % (str1, DELIM, str2)

@register.filter

def items(d):

    if isinstance(d, defaultdict):

        return d.iteritems()

    return d

@register.filter

def get_value_after_dot(value):

    return value.split(".")[1]

# @register.filter

# def strip_http(value):

#     return value.replace('http://','')[:-1]

from math import log

unit_list = zip(['bytes', 'kB', 'MB', 'GB', 'TB', 'PB'], [0, 0, 0, 0, 0, 0])

@register.filter

def sizeof_fmt(num):

    """Human friendly file size"""

    try:

        num = float(num)

    except:

        return

    else:

        if math.isinf(num):

            return 'Unlimited'

        if num > 1:

            exponent = min(int(log(num, 1024)), len(unit_list) - 1)

            quotient = float(num) / 1024**exponent

            unit, num_decimals = unit_list[exponent]

            format_string = '{0:.%sf} {1}' % (num_decimals)

            return format_string.format(quotient, unit)

        if num == 0:

            return '0 bytes'

        if num == 1:

            return '1 byte'

        else:

           return '0'

@register.filter

def isinf(v):

    if math.isinf(v):

        return 'Unlimited'

    else:

        return v

@register.filter

def truncatename(v, max=18, append="..."):

    length = len(v)

    if length>max:

        return v[:max] + append

    else:

        return v

@register.filter

def resource_groups(project_definition):

    try:

        grants = project_definition.projectresourcegrant_set

        return grants.values_list('resource__group', flat=True)

    except:

        return ()

@register.filter

def resource_grants(project_definition):

    try:

        grants = project_definition.projectresourcegrant_set

        grants = grants.values_list(

            'resource__name',

            'resource__service__name',

            'member_capacity'

        )

        return dict((RESOURCE_SEPARATOR.join([e[1], e[0]]), e[2]) for e in grants)

    except:

        return {}