modularisation

[restcoda.git] / utils.py

#!/usr/bin/python
# -*- coding: utf-8 -*-

#
# petits utilitaires
#
def dict_fetchone(cursor):
    """Renvoie le resultat d'un fetchone dans un dictionnaire"""
    result = cursor.fetchone()
    if result == None: return None
    result_dict = {}
    for i in range(len(result)):
        if isinstance( result[i], str ):
	    result_dict[cursor.description[i][0]] = result[i].decode('iso-8859-1')
	else:
            result_dict[cursor.description[i][0]] = result[i]
    return result_dict

import re 
p = re.compile('(dem|com)-(...)-(...)',re.IGNORECASE)
def coda2rest(value):
    """Traduit un nom CODA vers la base REST correspodante, 
    par exemple DEM-LOG-AUF en demlog ou COM-ARE-VN3 en comarevn3"""
    m = p.search(value)
    if m == None: return value
    if m.group(3).lower() == 'auf':
        return m.group(1).lower() + m.group(2).lower()
    else:
        return m.group(1).lower() + m.group(2).lower() + m.group(3).lower()