Source code for shared.helpers

# encoding: utf-8

import datetime
from urllib import (
    unquote,
    urlencode,
)
from urlparse import (
    urlparse,
    urlunparse,
)

from django.utils import timezone

from app.shared.enums import ChartTimeRange


def create_url(scheme='', url='', path='', params=None):
    """
    Creates URI by adding GET parameters to URL

    :param scheme: url scheme, e.g. 'http'
    :param url: base url, e.g. 'example.com'
    :param path: patch, e.g. '/foo/bar/baz'
    :param params: dict of GET parameters, e.g. {'key': 'val'}
    """
    if not (scheme or url or path):
        return
    params = params or {}

    url = scheme + '://' + url if scheme else url
    url += path
    url_parts = list(urlparse(url))
    url_parts[4] = unquote(urlencode(params))
    url = urlunparse(url_parts)
    return url


def get_date_format(date_type):
    """
    :param date_type: "week", "month", "year" or "all-time"
    :return: date format
    """
    return {
        ChartTimeRange.WEEK: '%Y-%m-%d',
        ChartTimeRange.MONTH: '%Y-%m',
        ChartTimeRange.YEAR: '%Y',
        ChartTimeRange.ALLTIME: None,
    }[date_type]


def unix_time(datetime_):
    """
    :param datetime_: datetime object with or without timezone
    :return: unix time for given datetime
    :rtype: int
    """
    epoch = datetime.datetime.utcfromtimestamp(0)

    if timezone.is_aware(datetime_):
        epoch = timezone.make_aware(epoch, timezone.utc)

    delta = datetime_ - epoch

    return int(delta.total_seconds())


def is_whole(x):
    return (x % 1) == 0