Produits Entr'ouvert » Calebasse

import os

from django.contrib.auth.models import Group

from django.conf import settings

from datetime import timedelta, datetime

from .middleware.request import get_request

__EPOCH = datetime(day=5,month=1,year=1970)

def __date_to_datetime(date):

    return datetime(date.year, date.month, date.day)

def weeks_since_epoch(date):

    days_since_epoch = (__date_to_datetime(date) - __EPOCH).days

    return days_since_epoch // 7

def weekday_ranks(date):

    '''Returns n so that if date occurs on a certain weekday, this is the

       n-th weekday of the month counting from the first. Also return -n-1 if

       this is the n-th weekday of the month counting from the last.

    '''

    n = 0

    month = date.month

    i = date - timedelta(days=7)

    while i.month == month:

        n += 1

        i = i - timedelta(days=7)

    m = -1

    i = date + timedelta(days=7)

    while i.month == month:

        m -= 1

        i = i + timedelta(days=7)

    return n, m

def is_super_user(user):

    if not user or not user.is_authenticated():

        return False

    if user.is_superuser:

        return True

    super_group = None

    try:

        super_group = Group.objects.get(name='Super utilisateurs')

    except:

        return False

    if super_group in user.groups.all():

        return True

    return False

def is_validator(user):

    if is_super_user(user):

        return True

    if not user or not user.is_authenticated():

        return False

    validator_group = None

    try:

        validator_group = Group.objects.get(name='Administratifs')

    except:

        return False

    if validator_group in user.groups.all():

        return True

    return False

def get_nir_control_key(nir):

    try:

        # Corse dpt 2A et 2B

        minus = 0

        if nir[6] in ('A', 'a'):

            nir = [c for c in nir]

            nir[6] = '0'

            nir = ''.join(nir)

            minus = 1000000

        elif nir[6] in ('B', 'b'):

            nir = [c for c in nir]

            nir[6] = '0'

            nir = ''.join(nir)

            minus = 2000000

        nir = int(nir) - minus

        return (97 - (nir % 97))

    except:

        return None

def get_service_setting(setting_name, default_value=None):

    from .cbv import HOME_SERVICE_COOKIE

    request = get_request()

    if not request:

        return None

    service = request.COOKIES.get(HOME_SERVICE_COOKIE)

    if not service:

        return None

    if not hasattr(settings, 'SERVICE_SETTINGS'):

        return None

    return settings.SERVICE_SETTINGS.get(service, {}).get(setting_name) or default_value

def get_last_file(path, prefix=None, suffix=None):

    '''

    A filename is of format year-mont-day-hour-min-sec.xml or

    prefix_year-mont-day-hour-min-secsuffix

    If suffix is None, only file *_year-mont-day-hour-min-sec are treated.

    If prefix is None, all files *_year-mont-day-hour-min-secsuffix are

    treated.

    '''

    if not path or not os.path.isdir(path):

        return None

    cv_files = [(f, f) for f in os.listdir(path) \

            if os.path.isfile(os.path.join(path, f)) ]

    if prefix:

        cv_files = [(f1, f2[len(prefix) + 1:])

                for f1, f2 in cv_files if f2.startswith(prefix)]

    else:

        cv_files = [(f1, f2.rsplit('_', 1)[-1])

                for f1, f2 in cv_files]

    if suffix:

        cv_files = [(f1, f2[:-len(suffix)])

                for f1, f2 in cv_files if f2.endswith(suffix)]

    fmt = '%Y-%d-%m-%H-%M-%S'

    cv_files_and_dates = list()

    for f1, f2 in cv_files:

        try:

            d = datetime.strptime(f2, fmt)

        except:

            pass

        else:

            cv_files_and_dates.append((f1, d))

    if len(cv_files_and_dates):

        cv_files_and_dates = sorted(cv_files_and_dates,

                key=lambda x:x[1], reverse=True)

        return os.path.join(path, cv_files_and_dates[0][0])

    else:

        return None