Produits Entr'ouvert » w.c.s. » Au quotidien

import base64

import datetime

import urllib2

try:

    import json

except ImportError:

    import simplejson as json

import time

import vobject

from qommon import get_cfg

from qommon.misc import format_time

from qommon.form import *

from wcs.data_sources import register_data_source_function

from wcs.formdata import Evolution

from wcs.forms.common import FormStatusPage

from wcs.workflows import Workflow, WorkflowStatusItem, register_item_class

def get_clicrdv_req(url):

    misc_cfg = get_cfg('misc', {})

    url = 'https://%s/api/v1/%s' % (

                misc_cfg.get('aq-clicrdv-server', 'sandbox.clicrdv.com'), url)

    if '?' in url:

       url = url + '&apikey=%s&format=json' % misc_cfg.get('aq-clicrdv-api-key')

    else:

       url = url + '?apikey=%s&format=json' % misc_cfg.get('aq-clicrdv-api-key')

    req = urllib2.Request(url)

    username = misc_cfg.get('aq-clicrdv-api-username')

    password = misc_cfg.get('aq-clicrdv-api-password')

    authheader = 'Basic ' + base64.encodestring('%s:%s' % (username, password))[:-1]

    req.add_header('Authorization', authheader)

    return req

def get_json(url):

    return json.load(urllib2.urlopen(get_clicrdv_req(url)))

def as_str(s):

    if type(s) is unicode:

        return s.encode(get_publisher().site_charset)

    return s

def get_all_intervention_sets():

    interventions_set = []

    for interventionset in sorted(get_json('interventionsets').get('records'), 

            lambda x,y: cmp(x['sort'],y['sort'])):

        interventions = []

        for intervention in sorted(get_json('interventionsets/%s/interventions' % interventionset.get('id')).get('records'),

                lambda x,y: cmp(x['sort'], y['sort'])):

            if intervention.get('deleted') == True:

                continue

            name = '%s' % as_str(intervention.get('publicname'))

            if not name:

                name = '%s' % as_str(intervention.get('name'))

            interventions.append((intervention.get('id'), as_str(name)))

        interventions_set.append({

                'id': interventionset.get('id'),

                'group_id': interventionset.get('group_id'),

                'name': as_str(interventionset.get('name')),

                'publicname': as_str(interventionset.get('publicname')) or '',

                'description': as_str(interventionset.get('description')) or '',

                'interventions': interventions

                })

    return interventions_set

def get_all_interventions():

    interventions = []

    for s in get_all_intervention_sets():

        for i, publicname in s['interventions']:

            intervention_label = '%s - %s' % (s['publicname'], publicname)

            interventions.append((i, as_str(intervention_label)))

    return interventions

def get_interventions_in_set(interventionset_id):

    interventions = []

    interventions_json = get_json('interventionsets/%s/interventions' % interventionset_id)

    for intervention in interventions_json.get('records'):

        if intervention.get('deleted') != True:

            name = '%s' % as_str(intervention.get('publicname'))

            if not name:

                name = '%s' % as_str(intervention.get('name'))

            interventions.append((intervention.get('id'), name))

    return interventions

def get_available_timeslots(intervention, date_start=None, date_end=None):

    timeslots = []

    iid = intervention

    gid = get_json('interventions/%s' % iid).get('group_id')

    request_url = 'availabletimeslots?intervention_ids[]=%s&group_id=%s' % (iid, gid)

    if date_start is None:

        date_start = datetime.datetime.today().strftime('%Y-%m-%d')

    if date_end is None:

        date_end = (datetime.datetime.today() + datetime.timedelta(366)).strftime('%Y-%m-%d')

    if date_start:

        request_url = request_url + '&start=%s' % urllib2.quote(date_start)

    if date_end:

        request_url = request_url + '&end=%s' % urllib2.quote(date_end)

    for timeslot in get_json(request_url).get('availabletimeslots'):

        timeslots.append(timeslot.get('start'))

    timeslots.sort()

    return timeslots

def get_available_dates(intervention):

    dates = []

    for timeslot in get_available_timeslots(intervention):

        parsed = time.strptime(timeslot, '%Y-%m-%d %H:%M:%S')

        date_tuple = (time.strftime('%Y-%m-%d', parsed),

                      format_time(parsed, '%(weekday_name)s %(day)0.2d/%(month)0.2d/%(year)s'))

        if date_tuple in dates:

            continue

        dates.append(date_tuple)

    return dates

def get_available_times(intervention, date):

    times = []

    timeslots = get_available_timeslots(intervention,

                    date_start='%s 00:00:00' % date,

                    date_end='%s 23:59:59' % date)

    for timeslot in timeslots:

        parsed = time.strptime(timeslot, '%Y-%m-%d %H:%M:%S')

        time_tuple = (time.strftime('%H:%M:%S', parsed),

                      time.strftime('%Hh%M', parsed))

        times.append(time_tuple)

    times.sort()

    return times

register_data_source_function(get_all_interventions, 'clicrdv_get_all_interventions')

register_data_source_function(get_interventions_in_set, 'clicrdv_get_interventions_in_set')

register_data_source_function(get_available_dates, 'clicrdv_get_available_dates')

register_data_source_function(get_available_times, 'clicrdv_get_available_times')

def form_download_event(self):

    self.check_receiver()

    found = False

    for evo in self.filled.evolution:

        if evo.parts:

            for p in evo.parts:

                if not isinstance(p, AppointmentPart):

                    continue

                cal = vobject.iCalendar()

                cal.add('prodid').value = '-//Entr\'ouvert//NON SGML Publik'

                vevent = vobject.newFromBehavior('vevent')

                vevent.add('uid').value = 'clicrdv-%s' % p.id

                vevent.add('summary').value = p.json_dict.get('group_name')

                vevent.add('dtstart').value = datetime.datetime.strptime(

                                p.json_dict.get('start'), '%Y-%m-%d %H:%M:%S')

                vevent.add('dtend').value = datetime.datetime.strptime(

                                p.json_dict.get('end'), '%Y-%m-%d %H:%M:%S')

                vevent.add('location').value = p.json_dict.get('location')

                cal.add(vevent)

                response = get_response()

                response.set_content_type('text/calendar')

                return cal.serialize()

    raise TraversalError()

class AppointmentPart(object):

    def __init__(self, json_dict):

        self.id = json_dict.get('id')

        self.json_dict = json_dict

    def view(self):

        return htmltext('<p class="appointment"><a href="clicrdvevent">%s</a></p>' % (

                                _('Download Appointment')))

class AppointmentErrorPart(object):

    def __init__(self, msg):

        self.msg = msg

    def view(self):

        return htmltext('<p class="appointment-error">%s</p>' % str(self.msg))

class ClicRdvCreateAppointment(WorkflowStatusItem):

    description = N_('Create a ClicRDV Appointment')

    key = 'clicrdv-create'

    category = ('aq-clicrdv', N_('ClicRDV'))

    endpoint = False

    var_firstname = None

    var_lastname = None

    var_email = None

    var_firstphone = None

    var_secondphone = None

    var_datetime = None

    var_intervention_id = None

    status_on_success = None

    status_on_failure = None

    def init(cls):

        FormStatusPage._q_extra_exports.append('clicrdvevent')

        FormStatusPage.clicrdvevent = form_download_event

    init = classmethod(init)

    def is_available(self, workflow=None):

        return get_publisher().has_site_option('clicrdv')

    is_available = classmethod(is_available)

    def render_as_line(self):

        return _('Create an appointment in ClicRDV')

    def get_parameters(self):

        return ('var_firstname', 'var_lastname', 'var_email', 'var_firstphone',

                'var_secondphone', 'var_datetime', 'var_intervention_id',

                'status_on_success', 'status_on_failure')

    def add_parameters_widgets(self, form, parameters, prefix='', formdef=None):

        parameter_labels = {

            'var_firstname': N_('First Name'),

            'var_lastname': N_('Last Name'),

            'var_email': N_('Email'),

            'var_firstphone': N_('Phone (1st)'),

            'var_secondphone':  N_('Phone (2nd)'),

            'var_datetime': N_('Date/time'),

            'var_intervention_id': N_('Intervention Id'),

        }

        for parameter in self.get_parameters():

            if not parameter in parameter_labels:

                continue

            if parameter in parameters:

                form.add(StringWidget, '%s%s' % (prefix, parameter),

                         title=_(parameter_labels.get(parameter)),

                         value=getattr(self, parameter),

                         required=False)

        if 'status_on_success' in parameters:

            form.add(SingleSelectWidget, '%sstatus_on_success' % prefix,

                    title=_('Status On Success'), value=self.status_on_success,

                    options = [(None, '---')] + [(x.id, x.name) for x in self.parent.parent.possible_status])

        if 'status_on_failure' in parameters:

            form.add(SingleSelectWidget, '%sstatus_on_failure' % prefix,

                    title=_('Status On Failure'), value=self.status_on_failure,

                    options = [(None, '---')] + [(x.id, x.name) for x in self.parent.parent.possible_status])

    def perform(self, formdata):

        args = {}

        for parameter in self.get_parameters():

            args[parameter] = self.compute(getattr(self, parameter))

            if not args.get(parameter):

                del args[parameter]

        message = {'appointment':

                      {'fiche': {'firstname': args.get('var_firstname', '-'),

                                 'lastname': args.get('var_lastname', '-'),

                                 'email': args.get('var_email'),

                                 'firstphone': args.get('var_firstphone'),

                                 'secondphone': args.get('var_secondphone'),

                                },

                       'date': args.get('var_datetime'),

                       'intervention_ids': [int(args.get('var_intervention_id'))],

                       # 'comments': '-',

                       'websource': 'Publik'}

                  }

        req = get_clicrdv_req('appointments')

        req.add_data(json.dumps(message))

        req.add_header('Content-Type', 'application/json')

        try:

            fd = urllib2.urlopen(req)

        except urllib2.HTTPError, e:

            success = False

            try:

                msg = json.load(e.fp)[0].get('error')

            except:

                msg = _('unknown error')

            if formdata.evolution:

                evo = formdata.evolution[-1]

            else:

                formdata.evolution = []

                evo = Evolution()

                evo.time = time.localtime()

                evo.status = formdata.status

                formdata.evolution.append(evo)

            evo.add_part(AppointmentErrorPart(msg))

        else:

            success = True

            response = json.load(fd)

            appointment_id = response.get('records')[0].get('id')

            # add a message in formdata.evolution

            if formdata.evolution:

                evo = formdata.evolution[-1]

            else:

                formdata.evolution = []

                evo = Evolution()

                evo.time = time.localtime()

                evo.status = formdata.status

                formdata.evolution.append(evo)

            evo.add_part(AppointmentPart(response.get('records')[0]))

        formdata.store()

        if (success and self.status_on_success) or (success is False and self.status_on_failure):

            if success:

                formdata.status = 'wf-%s' % self.status_on_success

            else:

                formdata.status = 'wf-%s' % self.status_on_failure

register_item_class(ClicRdvCreateAppointment)

class ClicRdvCancelAppointment(WorkflowStatusItem):

    description = N_('Cancel a ClicRDV Appointment')

    key = 'clicrdv-cancel'

    category = ('aq-clicrdv', N_('ClicRDV'))

    endpoint = False

    status_on_success = None

    status_on_failure = None

    def get_parameters(self):

        return ('status_on_success', 'status_on_failure')

    def add_parameters_widgets(self, form, parameters, prefix='', formdef=None):

        if 'status_on_success' in parameters:

            form.add(SingleSelectWidget, '%sstatus_on_success' % prefix,

                    title=_('Status On Success'), value=self.status_on_success,

                    options = [(None, '---')] + [(x.id, x.name) for x in self.parent.parent.possible_status])

        if 'status_on_failure' in parameters:

            form.add(SingleSelectWidget, '%sstatus_on_failure' % prefix,

                    title=_('Status On Failure'), value=self.status_on_failure,

                    options = [(None, '---')] + [(x.id, x.name) for x in self.parent.parent.possible_status])

    def is_available(self, workflow=None):

        return get_publisher().has_site_option('clicrdv')

    is_available = classmethod(is_available)

    def render_as_line(self):

        return _('Cancel an appointment in ClicRDV')

    def perform(self, formdata):

        success = True

        for evo in [evo for evo in formdata.evolution if evo.parts]:

            for part in [part for part in evo.parts if isinstance(part, AppointmentPart)]:

                appointment_id = part.id

                try:

                    req = get_clicrdv_req('appointments/%s' % appointment_id)

                    req.get_method = (lambda: 'DELETE')

                    fd = urllib2.urlopen(req)

                    none = fd.read()

                except urllib2.URLError:

                    # clicrdv will return a "Bad Request" (HTTP 400) response

                    # when it's not possible to remove an appointment

                    # (for example because it's too late)

                    success = False

        if (success and self.status_on_success) or (success is False and self.status_on_failure):

            if success:

                formdata.status = 'wf-%s' % self.status_on_success

            else:

                formdata.status = 'wf-%s' % self.status_on_failure

register_item_class(ClicRdvCancelAppointment)