python-sqlkit 0.9.5-1ubuntu1 / usr / share / pyshared / sqlkit / misc / utils.py

import re
import os
import subprocess

import sqlalchemy

class Container(object):
    """
    a container for other object that allows iteration and dict-like access
    Iterating on this correspond to iterating on the *values* (not the keys)::

       for el in m.related: print el

    is equivalent to::

       for el in m.related.values(): print el

    Container has the nice feature that it completes when used interactively under ipython
    """
    def __init__(self, cdict=None):
        """
        cdict may be a dictionary that will be used to initialize the Container
        """
        self._attrs = []
        self._n = 0
        if cdict:
            for key, value in cdict.iteritems():
                self.add_element(key, value)

    def add_element(self, key, value):
        setattr(self, key, value)
        if not key in self._attrs:
            self._attrs += [key]

    def keys(self):
        return self._attrs

    def values(self):
        return [self[at] for at in self._attrs]

    def iteritems(self):
        for attr in self._attrs:
            yield attr, self[attr]

    def __contains__(self, key):
        return key in self._attrs
 
    def __getitem__(self, key):
        return getattr(self, key)
    
    def __setitem__(self, key, value):
        return self.add_element(key, value)
    
    def __iter__(self):
        return iter(self.values())

    def __repr__(self):
        return "%s" % self._attrs

    def get(self, key, default=None):
        return getattr(self, key, default)


class DictLike(object):
    """
    simple way to enable %(field_name)s lookup in classes
    used to allow correct use of 'format'. None is casted to empty string
    """
    def __init__(self, obj):
        self.obj = obj
        
    def __getitem__(self, key):
        """
        Cast all None values to '' otherwise the lookup can be scattered with 'None'
        as %(field_name)s would render as 'None' a None value
        """
        return getattr(self.obj, key, '') or ''

    def __getattr__(self, key):
        return getattr(self.obj, key)

class ObjLike(object):
    """
    Simple way to access keys of a dict as attributes
    """
    def __init__(self, dictionary):
        """
        :param dictionary: the dictionary whose keys we want to acces as attributes
        """
        self.dict = dictionary

    def __getattr__(self, key):
        try:
            return self.dict[key]
        except KeyError, e:
            return object.__getattr__(self, key)
            
def check_sqlalchemy_version(req):
    """
    It only works if pkg_resources is present
    """
    SA_VER = sqlalchemy.__version__

    def fatal_exit():
        import sys
        msg = "Sqlkit requires at least version %s, found %s"
        print msg % (req, SA_VER)
        sys.exit(1)

    def warning():
        import warnings
        msg = ("You have %s version of sqlalchemy. Sqlkit is tested with " +
              "version newer that %s. You shouldn't have problems " +
              "but we're not sure" )
        warnings.warn(msg)
    
    try:
        from pkg_resources import parse_version as pv
        if pv(sqlalchemy.__version__) >= pv(req):
            return  
        elif pv(sqlalchemy.__version__) < pv('0.5.4'):
            return fatal_exit()
        else:
            warning()
    except ImportError, e:
        ## don't quit just becouse they don't have pkg_resources
        return True


def str2list(list_or_string):
    """
    input parsing common func
    """
    
    if isinstance(list_or_string, basestring):
        return re.split('[ ,]*', list_or_string.strip())
    return list_or_string or []


def get_viewer(ext):
    """
    return the path of the viewer for extension

    :param ext: extension
    """
    import platform
    import subprocess as sp

    progs = {
        'pdf' : ('xdg-open', 'gnome-open', 'acroread', 'xpdf', 'gv', ),
        'odt' : ('xdg-open', 'gnome-open', 'oowriter', ),
        }
    # another workig way to invoke the correct application is to use
    # webbrowser.open_new() but under linux it may switch you to another desktop.
    if platform.system() == 'Windows' or os.name == "mac":
        # this is good for local file but for remote files *in PrintTool* only, it shows
        # an amaizing delay of 15 seconds that you don't have via webbrowser module
        # I really don't understand why, a simple test in a file doesn't show this delay!
        return 'start'
    else:
        for cmd in progs[ext]:
            path = sp.Popen('which ' + cmd, shell=True, stdout=sp.PIPE).communicate()[0]
            if path:
                return path.strip()

def open_file(filename, viewer=None):
    """
    Open file filename with the default application (ext: .pdf or openoffice family)

    :param filename: the name of the file to be opened
    """
    if os.name in ('posix', 'mac'):
        if filename.endswith('pdf'):
            p =  subprocess.call((viewer or get_viewer('pdf'), filename)) 
        else:
            p =  subprocess.call((viewer or get_viewer('odt'), filename))
    else:
        # I made some test with call(('start', filename)) but in several
        # occasion I coudn't get the start command to find the
        # file. os.startfile() doesn't suffer the same problem.
        os.startfile(filename)


if __name__ == '__main__' :

    c = Container()
    c.add_element('uno', 1)
    print c.keys()
    print 'uno' in c
    c['due'] = 2
    c['tre'] = 3
    c['quattro'] = 4
    print c.due
    for x in c:
        if x>2:
            break
        print x
        
    for x in c:
        print x

    print str2list('uno,due, tre')
    print str2list(['uno','due', 'tre'])