#copyright ReportLab Inc. 2000

#see license.txt for license details

#history http://cvs.sourceforge.net/cgi-bin/cvsweb.cgi/reportlab/lib/utils.py?cvsroot=reportlab

import string, os, sys

from types import *

from reportlab.lib.logger import warnOnce

SeqTypes = (ListType,TupleType)

if os.name == 'mac':

    #with the Mac, we need to tag the file in a special

    #way so the system knows it is a PDF file.

    #This supplied by Joe Strout

    _KNOWN_MAC_EXT = {

        'BMP' : ('ogle','BMP '),

        'EPS' : ('ogle','EPSF'),

        'EPSF': ('ogle','EPSF'),

        'GIF' : ('ogle','GIFf'),

        'JPG' : ('ogle','JPEG'),

        'JPEG': ('ogle','JPEG'),

        'PCT' : ('ttxt','PICT'),

        'PICT': ('ttxt','PICT'),

        'PNG' : ('ogle','PNGf'),

        'PPM' : ('ogle','.PPM'),

        'TIF' : ('ogle','TIFF'),

        'TIFF': ('ogle','TIFF'),

        'PDF': ('CARO','PDF '),

        }

    def markfilename(filename,creatorcode=None,filetype=None,ext='PDF'):

        try:

            if creatorcode is None or filetype is None and ext is not None:

                try:

                    creatorcode, filetype = _KNOWN_MAC_EXT[ext]

                except:

                    return

            macfs.FSSpec(filename).SetCreatorType(creatorcode,filetype)

        except:

            pass

else:

    def markfilename(filename,creatorcode=None,filetype=None):

        pass

def isFileSystemDistro():

    """Attempt to detect if this copy of reportlab is running in a

    file system (as opposed to mostly running in a zip or McMillan

    archive or Jar file).  This is used by test cases, so that

    we can write test cases that don't get activated in a compiled

    distribution."""

    # is this safe enough?

    import reportlab.pdfgen.canvas

    hypothetical_location = reportlab.pdfgen.canvas

    return os.path.isfile(hypothetical_location)

def isCompactDistro():

    return not isFileSystemDistro()

def isSourceDistro():

    import reportlab.pdfgen.canvas

    hypothetical_location = reportlab.pdfgen.canvas

def _checkImportError(errMsg):

    if string.lower(string.strip(str(errMsg)[0:16]))!='no module named': raise

try:

    #raise ImportError

    ### NOTE!  FP_STR SHOULD PROBABLY ALWAYS DO A PYTHON STR() CONVERSION ON ARGS

    ### IN CASE THEY ARE "LAZY OBJECTS".  ACCELLERATOR DOESN'T DO THIS (YET)

    try:

        from _rl_accel import fp_str                # in case of builtin version

    except ImportError, errMsg:

        _checkImportError(errMsg)

        from reportlab.lib._rl_accel import fp_str  # specific

except ImportError, errMsg:

    #_checkImportError(errMsg) # this effectively requires _rl_accel... should not be required

    def fp_str(*a):

        if len(a)==1 and type(a[0]) in SeqTypes: a = a[0]

        s = []

        for i in a:

            s.append('%0.6f' % i)

        return string.join(s)

#hack test for comma users

if ',' in fp_str(0.25):

    _FP_STR = fp_str

    def fp_str(*a):

        return string.replace(apply(_FP_STR,a),',','.')

def recursiveImport(modulename, baseDir=None, noCWD=0, debug=0):

    """Dynamically imports possible packagized module, or raises ImportError"""

    import imp

    parts = string.split(modulename, '.')

    name = parts[0]

    if baseDir is None:

        path = sys.path[:]

    else:

        if type(baseDir) not in SeqTypes:

            path = [baseDir]

        else:

            path = list(baseDir)

    path = filter(None,path)

    if noCWD:

        while '.' in path:

            if debug: print 'removed . from path'

            path.remove('.')

        abspath = os.path.abspath('.')

        while abspath in path:

            if debug: print 'removed "%s" from path' % abspath

            path.remove(abspath)

    else:

        if '.' not in path:

            path.insert(0,'.')

    if debug:

        import pprint

        pp = pprint.pprint

        print 'path=',pp(path)

    #make import errors a bit more informative

    fullName = name

    try:

        (file, pathname, description) = imp.find_module(name, path)

        childModule = parentModule = imp.load_module(name, file, pathname, description)

        if debug: print 'imported module = %s' % parentModule

        for name in parts[1:]:

            fullName = fullName + '.' + name

            if debug: print 'trying part %s' % name

            (file, pathname, description) = imp.find_module(name, [os.path.dirname(parentModule.__file__)])

            childModule = imp.load_module(fullName, file, pathname, description)

            if debug: print 'imported module = %s' % childModule

            setattr(parentModule, name, childModule)

            parentModule = childModule

    except ImportError:

        msg = "cannot import '%s' while attempting recursive import of '%s'" % (fullName, modulename)

        if baseDir:

            msg = msg + " under paths '%s'" % `path`

        raise ImportError, msg

    return childModule

def recursiveGetAttr(obj, name):

    "Can call down into e.g. object1.object2[4].attr"

    return eval(name, obj.__dict__)

def recursiveSetAttr(obj, name, value):

    "Can call down into e.g. object1.object2[4].attr = value"

    #get the thing above last.

    tokens = string.split(name, '.')

    #print 'name=%s, tokens=%s' % (name, tokens)

    if len(tokens) == 1:

        setattr(obj, name, value)

    else:        

        most = string.join(tokens[:-1], '.')

        last = tokens[-1]

        #print 'most=%s, last=%s' % (most, last)

        parent = recursiveGetAttr(obj, most)

        #print 'parent=%s' % parent

        setattr(parent, last, value)

def import_zlib():

    try:

        import zlib

    except ImportError, errMsg:

        _checkImportError(errMsg)

        zlib = None

        from reportlab.rl_config import ZLIB_WARNINGS

        if ZLIB_WARNINGS: warnOnce('zlib not available')

    return zlib

try:

    from PIL import Image

except ImportError, errMsg:

    _checkImportError(errMsg)

    try:

        import Image

    except ImportError, errMsg:

        _checkImportError(errMsg)

        Image = None

PIL_Image = Image

del Image

__StringIO=None

def getStringIO(buf=None):

    '''unified StringIO instance interface'''

    global __StringIO

    if not __StringIO:

        try:

            from cStringIO import StringIO

        except ImportError:

            from StringIO import StringIO

        __StringIO = StringIO

    return buf is not None and __StringIO(buf) or __StringIO()

class ArgvDictValue:

    '''A type to allow clients of getArgvDict to specify a conversion function'''

    def __init__(self,value,func):

        self.value = value

        self.func = func

def getArgvDict(**kw):

    ''' Builds a dictionary from its keyword arguments with overrides from sys.argv.

        Attempts to be smart about conversions, but the value can be an instance

        of ArgDictValue to allow specifying a conversion function.

    '''

    def handleValue(v,av,func):

        if func:

            v = func(av)

        else:

            t = type(v)

            if t is StringType:

                v = av

            elif t is FloatType:

                v = float(av)

            elif t is IntType:

                v = int(av)

            elif t is ListType:

                v = list(eval(av))

            elif t is TupleType:

                v = tuple(eval(av))

            else:

                raise TypeError, "Can't convert string '%s' to %s" % (av,str(t))

        return v

    A = sys.argv[1:]

    R = {}

    for k, v in kw.items():

        if isinstance(v,ArgvDictValue):

            v, func = v.value, v.func

        else:

            func = None

        handled = 0

        ke = k+'='

        for a in A:

            if string.find(a,ke)==0:

                av = a[len(ke):]

                A.remove(a)

                R[k] = handleValue(v,av,func)

                handled = 1

                break

        if not handled: R[k] = handleValue(v,v,func)

    return R

def getHyphenater(hDict=None):

    try:

        from reportlab.lib.pyHnj import Hyphen

        if hDict is None: hDict=os.path.join(os.path.dirname(__file__),'hyphen.mashed')

        return Hyphen(hDict)

    except ImportError, errMsg:

        if str(errMsg)!='No module named pyHnj': raise

        return None

def _className(self):

    '''Return a shortened class name'''

    try:

        name = self.__class__.__name__

        i=string.rfind(name,'.')

        if i>=0: return name[i+1:]

        return name

    except AttributeError:

        return str(self)

def open_for_read(name,mode='b'):

    '''attempt to open a file or URL for reading'''

    import urllib

    try:

        t, o = urllib.splittype(name)

        if not t or t=='file': raise ValueError

        o = urllib.urlopen(name)

        return getStringIO(o.read())

    except:

        return open(name,'r'+mode)

class DebugMemo:

    '''Intended as a simple report back encapsulator

    Typical usages

    1) To record error data

        dbg = DebugMemo(fn='dbgmemo.dbg',myVar=value)

        dbg.add(anotherPayload='aaaa',andagain='bbb')

        dbg.dump()

    2) To show the recorded info

        dbg = DebugMemo(fn='dbgmemo.dbg',mode='r')

        dbg.load()

        dbg.show()

    3) To re-use recorded information

        dbg = DebugMemo(fn='dbgmemo.dbg',mode='r')

            dbg.load()

        myTestFunc(dbg.payload('myVar'),dbg.payload('andagain'))

    in addition to the payload variables the dump records many useful bits

    of information which are also printed in the show() method.

    '''

    def __init__(self,fn='rl_dbgmemo.dbg',mode='w',getScript=1,modules=(),**kw):

        import time, socket

        self.fn = fn

        if mode!='w': return

        self.store = store = {}

        if sys.exc_info() != (None,None,None):

            import traceback

            s = getStringIO()

            traceback.print_exc(None,s)

            store['__traceback'] = s.getvalue()

        cwd=os.getcwd()

        lcwd = os.listdir(cwd)

        exed = os.path.abspath(os.path.dirname(sys.argv[0]))

        store.update({  'gmt': time.asctime(time.gmtime(time.time())),

                        'platform': sys.platform,

                        'version': sys.version,

                        'executable': sys.executable,

                        'prefix': sys.prefix,

                        'path': sys.path,

                        'argv': sys.argv,

                        'cwd': cwd,

                        'hostname': socket.gethostname(),

                        'lcwd': lcwd,

                        })

        if exed!=cwd: store.update({'exed': exed,

                                    'lexed': os.listdir(exed),

                                    })

        if hasattr(os,'uname'):

            store.update({

                'uname': os.uname(),

                'ctermid': os.ctermid(),

                'getgid': os.getgid(),

                'getuid': os.getuid(),

                'getegid': os.getegid(),

                'geteuid': os.geteuid(),

                'getlogin': os.getlogin(),

                'getgroups': os.getgroups(),

                'getpgrp': os.getpgrp(),

                'getpid': os.getpid(),

                'getppid': os.getppid(),

                })

        if getScript:

            fn = os.path.abspath(sys.argv[0])

            if os.path.isfile(fn):

                store['__script'] = open(fn,'r').read()

        module_versions = {}

        for n,m in sys.modules.items():

            if n=='reportlab' or n=='rlextra' or n[:10]=='reportlab.' or n[:8]=='rlextra.':

                v = getattr(m,'__version__',None)

                if v: module_versions[n] = v

        store['__module_versions'] = module_versions

        self.store['__payload'] = {}

        self._add(kw)

    def _add(self,D):

        payload = self.store['__payload']

        for k, v in D.items():

            payload[k] = v

    def add(self,**kw):

        self._add(kw)

    def dump(self):

        import pickle

        pickle.dump(self.store,open(self.fn,'wb'))

    def load(self):

        import pickle

        self.store = pickle.load(open(self.fn,'rb'))

    def _show_module_versions(self,k,v):

        print k[2:]

        K = v.keys()

        K.sort()

        for k in K:

            vk = v[k]

            try:

                m = recursiveImport(k,sys.path[:],1)

                d = getattr(m,'__version__',None)==vk and 'SAME' or 'DIFFERENT'

            except:

                m = None

                d = '??????unknown??????'

            print '  %s = %s (%s)' % (k,vk,d)

    def _banner(self,k,what):

        print '###################%s %s##################' % (what,k[2:])