OpenOffice.org: utilisation des boîtes de dialogue du système par défaut.

[macros-openoffice.git] / auf-poste-client-reflets / openoffice / CODA.py

#!
# -*- coding: utf-8 -*-
"""
Extension pour faciliter l'intégration du logiciel Coda.
Macro pour faciliter l'intégration des données provenant de Coda dans
une feuille de calcul.

Copyright : Agence universitaire de la Francophonie
Licence : GNU General Public Licence, version 2
Auteur : Jean Christophe André
Date de création : septembre 2009
"""
import re
import datetime
import uno
import unohelper
from com.sun.star.awt import WindowDescriptor
from com.sun.star.awt.WindowClass import MODALTOP
from com.sun.star.awt.VclWindowPeerAttribute import OK, DEF_OK
from com.sun.star.uno import Exception as UnoException, RuntimeException
from com.sun.star.connection import NoConnectException
from com.sun.star.lang import Locale, IllegalArgumentException
from com.sun.star.beans import PropertyValue
#from com.sun.star.util.ParagraphProperties import CENTER
from com.sun.star.util.NumberFormat import CURRENCY, DATE, NUMBER
from com.sun.star.i18n.NumberFormatIndex import NUMBER_1000DEC2
from com.sun.star.container import NoSuchElementException

COLONNES_NUMERIQUES = ('Montant doc','Montant EUR')

##############################################################################

def clipboard_data(ctx, mimetype='text/plain;charset=utf-16'):
    clipboard = ctx.ServiceManager.createInstanceWithContext(
                "com.sun.star.datatransfer.clipboard.SystemClipboard", ctx)
    contents = clipboard.getContents()
    #print "Contents:\n%s\n%s\n%s" % ("-" * 78, "* " + "\n* ".join(dir(contents)), "=" * 78)
    #flavors = contents.getTransferDataFlavors()
    #print "Flavors:\n%s\n%s\n%s" % ("-" * 78, "* " + "\n* ".join([flavor.MimeType for flavor in flavors]), "=" * 78)
    found_flavor = None
    for flavor in contents.getTransferDataFlavors():
        if flavor.MimeType == mimetype:
            found_flavor = flavor
            break
    if not found_flavor:
        raise RuntimeError, u"Erreur : type de données '%s' non disponible.\n" \
            u"\nAvez-vous bien sélectionné puis copié les données dans CODA ?" \
            % mimetype
    data = contents.getTransferData(found_flavor)
    #print "Data:\n", "-" * 78, "\n", data
    return data

##############################################################################

def main(ctx, action):
    data = clipboard_data(ctx)

    numeric_pattern = re.compile('^[+-]?[0-9]+([\. ][0-9][0-9][0-9])*([\.,][0-9]+)?$')
    date_pattern = re.compile('^([0-9]?[0-9])/([0-9]?[0-9])/([0-9]?[0-9]?[0-9]?[0-9])$')

    if action == 'cumuls':
        sheet_name = u'CODA-Cumuls'
        # on découpe le texte reçu en lignes puis en colonnes
        # on ne garde ici que les 10 premières colonnes
        data = [row.split('\t')[:10] for row in data.splitlines()]
        data[0][6] = '' # suppression de la 2nde colonne 'Engagement'
        data[0][8] = '' # suppression de la colonne 'Colonne9'

    elif action == 'details' or action == 'details-encore':
        sheet_name = u'CODA-Détails'
        # on découpe le texte reçu en lignes puis en colonnes
        data = [row.split('\t') for row in data.splitlines()]

    else:
        raise RuntimeError, u"Action inconnue (erreur de programmation)."

    # on efface la dernière ligne pleine de car. nuls, le cas échéant
    last = data[len(data)-1]
    if len(last) == 1 and last[0].strip('\x00') == '':
        del data[len(data)-1]

    # on détermine le type des données, par colonne :
    # - on vérifie si toutes les valeurs de la colonne sont identiques
    # - on vérifie si la colonne ne contient que des dates
    # - on vérifie si la colonne ne contient que des nombres
    column_type = [str] * len(data[0])
    for column in range(len(data[0])-1,-1,-1):
        # quelques colonnes ne doivent pas être testées
        if action in ('details','details-encore') and data[0][column] in ('Num doc','PCG'):
            continue
        # on prend la première valeur, nettoyée un peu pour l'analyse
        first = data[1][column].strip().replace(u'\xa0','')
        flag_date = first and re.match(date_pattern, first)
        flag_numeric = first and first != '#DEV!' and re.match(numeric_pattern, first)
        diff = False
        # parcours des autres valeurs de la colonne
        for row in range(2,len(data)):
            value = data[row][column].strip().replace(u'\xa0','')
            if value:
                if not diff and value != first:
                    diff = True
                if flag_date and not re.match(date_pattern, value):
                    flag_date = False
                if flag_numeric and value != '#DEV!' and not re.match(numeric_pattern, value):
                    flag_numeric = False
        # s'il y a au moins 2 lignes et que les valeurs dans la colonne
        # sont toutes identiques et qu'il n'y a pas d'intitulé sur la
        # colonne alors on supprime toute la colonne
        if len(data) > 2 and not diff and not data[0][column]:
            for row in range(len(data)):
                del data[row][column]
            del column_type[column]
            continue
        # si la colonne contient des dates, conversion des valeurs comme telles
        if flag_date:
            for row in range(1,len(data)):
                value = data[row][column].strip().replace(u'\xa0','')
                if value and value != '#DEV!':
                    m = re.match(date_pattern, value)
                    if m:
                        (day, month, year) = [int(x) for x in m.groups()]
                        if year < 50:
                            year += 2000
                        elif year < 100:
                            year += 1900
                        value = '%02d/%02d/%04d' % (month, day, year)
                    data[row][column] = value
            column_type[column] = datetime.date
            continue
        # si la colonne est numérique, conversion des valeurs comme telles
        if flag_numeric:
            for row in range(1,len(data)):
                value = data[row][column].strip().replace(u'\xa0','')
                if value and value != '#DEV!':
                    data[row][column] = float(value.replace('.','').replace(',','.'))
            column_type[column] = float
            continue

    #print "Data:\n", "-" * 78, "\n", data

    # récupération du document en cours, vérification du type Spreadsheet
    desktop = ctx.ServiceManager.createInstanceWithContext(
                            "com.sun.star.frame.Desktop", ctx)
    document = desktop.getCurrentComponent()
    if not document.supportsService("com.sun.star.sheet.SpreadsheetDocument"):
        raise RuntimeError, u"Ce n'est pas un document de type Spreadsheet (Calc)."

    # basculement vers la feuille de calcul concernée, création au besoin
    sheets = document.getSheets()
    try:
        sheet = sheets.getByName(sheet_name)
        new_sheet = False
    except NoSuchElementException, e:
        sheets.insertNewByName(sheet_name, 0)
        sheet = sheets.getByName(sheet_name)
        new_sheet = True
    controller = document.getCurrentController()
    controller.setActiveSheet(sheet)

    # récupération des formats date, monétaire et nombre en français
    # http://api.openoffice.org/docs/common/ref/com/sun/star/util/XNumberFormatTypes.html
    # http://api.openoffice.org/docs/common/ref/com/sun/star/i18n/NumberFormatIndex.html
    locale_fr_FR = Locale('fr', 'FR', '')
    number_formats = document.getNumberFormats()
    date_format = number_formats.getStandardFormat(DATE, locale_fr_FR)
    currency_format = number_formats.getStandardFormat(CURRENCY, locale_fr_FR)
    number_format = number_formats.getFormatIndex(NUMBER_1000DEC2, locale_fr_FR)
    #format_string = number_formats.generateFormat(
    #                                NUMBER, locale_fr_FR, True, False, 2, 1)
    #format_key = number_formats.queryKey(format_string, locale_fr_FR, True)
    #if format_key < 0:
    #    format_key = number_formats.addNew(format_string, locale_fr_FR)
    #print "NUMBER_1000DEC2=%s key=%s" % (NUMBER_1000DEC2, format_key)
    #number_format = number_formats.getByKey(format_key)

    # détermination de la zone de travail
    if action.endswith('-encore'):
      # localisation de la la zone déjà utilisée
      cursor = sheet.createCursor()
      cursor.gotoEndOfUsedArea(False)
      cursor.gotoStartOfUsedArea(True)
      rangeAddress = cursor.getRangeAddress()
      row = rangeAddress.EndRow
      # FIXME: on devrait comparer ici rangeAddress.EndColumn avec len(data[0])
    else:
      row = 0

    cell_range = sheet.getCellRangeByPosition(
                                0, row, len(data[0])-1, row + len(data)-1)
    cursor = sheet.createCursorByRange(cell_range)

    # insertion des données
    for row in range(1, len(data)):
        for column in range(len(data[row])):
            value = data[row][column]
            cell = cursor.getCellByPosition(column, row)
            if column_type[column] == float and value != '#DEV!':
                cell.Value = value
            else:
                cell.Formula = value

    # insertion des en-têtes, après les données pour un autofit correct
    if new_sheet:
        for column in range(len(data[0])):
            # écriture de l'en-tête de la colonne
            cell = cursor.getCellByPosition(column, 0)
            cell.Formula = data[0][column]
            # formatage par défaut pour la colonne
            sheet_column = sheet.Columns.getByIndex(column)
            if column_type[column] == datetime.date:
                sheet_column.NumberFormat = date_format
            elif column_type[column] == float:
                if action == 'cumuls':
                    sheet_column.NumberFormat = currency_format
                else:
                    if data[0][column] in COLONNES_NUMERIQUES:
                        sheet_column.NumberFormat = number_format
                    else:
                        sheet_column.NumberFormat = currency_format
            # ajustement automatique, une fois toutes les valeurs écrites
            sheet_column.OptimalWidth = True
        # formatage spécifique pour la ligne d'en-têtes
        headers_row = sheet.Rows.getByIndex(0)
        headers_row.ParaAdjust = 3 # CENTER

##############################################################################

# Show a message box with the UNO based toolkit
def messageBox(ctx, message):
    document = XSCRIPTCONTEXT.getDocument()
    window = document.CurrentController.Frame.ContainerWindow

    aDescriptor = WindowDescriptor()
    aDescriptor.Type = MODALTOP
    aDescriptor.WindowServiceName = "infobox"
    aDescriptor.ParentIndex = -1
    aDescriptor.Parent = window
    #aDescriptor.Bounds = Rectangle()
    aDescriptor.WindowAttributes = OK

    tk = window.getToolkit()
    msgbox = tk.createWindow(aDescriptor)
    msgbox.setCaptionText("Collage d'interrogation CODA")
    msgbox.setMessageText(unicode(message))

    return msgbox.execute()

def coller_cumuls(event=False):
    u"""Coller une interrogation des cumuls CODA."""
    ctx = uno.getComponentContext()
    try:
        main(ctx, 'cumuls')
    except Exception, e:
        messageBox(ctx, e)
    return None

def coller_details(event=False):
    u"""Coller une interrogation des détails CODA."""
    ctx = uno.getComponentContext()
    try:
        main(ctx, 'details')
    except Exception, e:
        messageBox(ctx, e)
    return None

def coller_details_encore(event=False):
    u"""Coller une interrogation des détails CODA supplémentaire."""
    ctx = uno.getComponentContext()
    try:
        main(ctx, 'details-encore')
    except Exception, e:
        messageBox(ctx, e)
    return None

g_exportedScripts = (coller_cumuls, coller_details, coller_details_encore)

##############################################################################

from com.sun.star.task import XJobExecutor

class CollerCumulsJob(unohelper.Base, XJobExecutor):
    def __init__(self, context):
        self._context = context

    def trigger(self, args):
        try:
            main(self._context, 'cumuls')
        except Exception, e:
            messageBox(self._context, e)

class CollerDetailsJob(unohelper.Base, XJobExecutor):
    def __init__(self, context):
        self._context = context

    def trigger(self, args):
        try:
            main(self._context, 'details')
        except Exception, e:
            messageBox(self._context, e)

class CollerDetailsEncoreJob(unohelper.Base, XJobExecutor):
    def __init__(self, context):
        self._context = context

    def trigger(self, args):
        try:
            main(self._context, 'details-encore')
        except Exception, e:
            messageBox(self._context, e)

g_ImplementationHelper = unohelper.ImplementationHelper()
g_ImplementationHelper.addImplementation( \
    CollerCumulsJob, "org.auf.openoffice.CODA.CollerCumuls", \
    ("com.sun.star.task.Job",),)
g_ImplementationHelper.addImplementation( \
    CollerDetailsJob, "org.auf.openoffice.CODA.CollerDetails", \
    ("com.sun.star.task.Job",),)
g_ImplementationHelper.addImplementation( \
    CollerDetailsEncoreJob, "org.auf.openoffice.CODA.CollerDetailsEncore", \
    ("com.sun.star.task.Job",),)