#@+leo-ver=4-thin
#@+node:Zoomq.20051028115650:@thin dict4ini.py
#@verbatim
#@ignore
#@verbatim
#@language python
#@<< dict4ini declarations >>
#@+node:Zoomq.20051028115650.1:<< dict4ini declarations >>
#coding=utf-8
# dump python dict to ini format file
# Author: limodou (limodou@gmail.com)
# Copyleft GPL
# $Revision: 1.1 $
# you can see http://wiki.woodpecker.org.cn/moin/Dict4Ini for more details
# 
# Updates:
#   2005/10/16
#     Saving the order of the items
#     Adding float format
#

__version__ = '0.2'

import sys
import locale
import os.path
import re

r_float = re.compile('\d*\.\d+')
section_delimeter = '/'

#@-node:Zoomq.20051028115650.1:<< dict4ini declarations >>
#@nl
#@+others
#@+node:Zoomq.20051028115650.3:class DictNode
class DictNode(object):
    #@	@+others
    #@+node:Zoomq.20051028115650.4:__init__
    def __init__(self, values, encoding=None, root=None, section=[], orders=[]):
        self._items = values
        self._orders = orders
        self._encoding = encoding
        self._root = root
        self._section = section
    #@-node:Zoomq.20051028115650.4:__init__
    #@+node:Zoomq.20051028115650.5:__getitem__
    def __getitem__(self, name):
        if self._items.has_key(name):
            value = self._items[name]
            if isinstance(value, dict):
                return DictNode(value, self._encoding, self._root, self._section + [name])
            else:
                return value
        else:
            self._items[name] = {}
            self._root.setorder(self.get_full_keyname(name))
            return DictNode(self._items[name], self._encoding, self._root, self._section + [name])
    #@-node:Zoomq.20051028115650.5:__getitem__
    #@+node:Zoomq.20051028115650.6:__setitem__
    def __setitem__(self, name, value):
        if section_delimeter in name:
            sec = name.split(section_delimeter)
            obj = self._items
            
            _s = self._section[:]
            for i in sec[:-1]:
                _s.append(i)
                if obj.has_key(i):
                    if isinstance(obj[i], dict):
                        obj = obj[i]
                    else:
                        obj[i] = {} #may lost some data
                        obj = obj[i]
                else:
                    obj[i] = {}
                    self._root.setorder(section_delimeter.join(_s))
                    obj = obj[i]
            obj[sec[-1]] = value
            self._root.setorder(section_delimeter.join(_s + [sec[-1]]))
        else:
            self._items[name] = value
            self._root.setorder(self.get_full_keyname(name))
    #@-node:Zoomq.20051028115650.6:__setitem__
    #@+node:Zoomq.20051028115650.7:__delitem__
    def __delitem__(self, name):
        if self._items.has_key(name):
            del self._items[name]
    #@-node:Zoomq.20051028115650.7:__delitem__
    #@+node:Zoomq.20051028115650.8:__repr__
    def __repr__(self):
        return repr(self._items)
    #@-node:Zoomq.20051028115650.8:__repr__
    #@+node:Zoomq.20051028115650.9:__getattr__
    def __getattr__(self, name):
        return self.__getitem__(name)
    #@-node:Zoomq.20051028115650.9:__getattr__
    #@+node:Zoomq.20051028115650.10:__setattr__
    def __setattr__(self, name, value):
        if name.startswith('_'):
            if name == '_comment':
                self._root._comments[section_delimeter.join(self._section)] = value
            else:
                self.__dict__[name] = value
        else:
            self.__setitem__(name, value)
    #@-node:Zoomq.20051028115650.10:__setattr__
    #@+node:Zoomq.20051028115650.11:comment
    def comment(self, name, comment):
        if name:
            self._root._comments[section_delimeter.join(self._section + [name])] = comment
        else:
            self._root._comments[section_delimeter.join(self._section)] = comment
    #@-node:Zoomq.20051028115650.11:comment
    #@+node:Zoomq.20051028115650.12:__delattr__
    def __delattr__(self, name):
        if self._items.has_key(name):
            del self._items[name]
    #@-node:Zoomq.20051028115650.12:__delattr__
    #@+node:Zoomq.20051028115650.13:__str__
    def __str__(self):
        return repr(self._items)
    #@-node:Zoomq.20051028115650.13:__str__
    #@+node:Zoomq.20051028115650.14:__len__
    def __len__(self):
        return len(self._items)
    #@-node:Zoomq.20051028115650.14:__len__
    #@+node:Zoomq.20051028115650.15:has_key
    def has_key(self, name):
        return self._items.has_key(name)
    #@-node:Zoomq.20051028115650.15:has_key
    #@+node:Zoomq.20051028115650.16:items
    def items(self):
        return self._items.items()
    #@-node:Zoomq.20051028115650.16:items
    #@+node:Zoomq.20051028115650.17:setdefault
    def setdefault(self, name, value):
        return self._items.setdefault(name, value)
    #@-node:Zoomq.20051028115650.17:setdefault
    #@+node:Zoomq.20051028115650.18:get
    def get(self, name, default=None):
        return self._items.get(name, default)
    #@-node:Zoomq.20051028115650.18:get
    #@+node:Zoomq.20051028115650.19:keys
    def keys(self):
        return self._items.keys()
    #@-node:Zoomq.20051028115650.19:keys
    #@+node:Zoomq.20051028115650.20:values
    def values(self):
        return self._items.values()
    #@-node:Zoomq.20051028115650.20:values
    #@+node:Zoomq.20051028115650.21:get_full_keyname
    def get_full_keyname(self, key):
        return section_delimeter.join(self._section + [key])
    #@-node:Zoomq.20051028115650.21:get_full_keyname
    #@-others
#@-node:Zoomq.20051028115650.3:class DictNode
#@+node:Zoomq.20051028115650.22:class DictIni
    
class DictIni(DictNode):
    #@	@+others
    #@+node:Zoomq.20051028115650.23:__init__
    def __init__(self, inifile=None, values=None, encoding=None, commentdelimeter='#'):
        self._items = {}
        self._inifile = inifile
        self._root = self
        self._section = []
        self._commentdelimeter = commentdelimeter
        if values is not None:
            self._items = values
        self._comments = {}
        self._orders = {}
        self._ID = 1
        self._encoding = getdefaultencoding(encoding)
        
        if self._inifile and os.path.exists(self._inifile):
            self.read(self._inifile, self._encoding)
    #@-node:Zoomq.20051028115650.23:__init__
    #@+node:Zoomq.20051028115650.24:setfilename
    def setfilename(self, filename):
        self._inifile = filename
    #@-node:Zoomq.20051028115650.24:setfilename
    #@+node:Zoomq.20051028115650.25:getfilename
    def getfilename(self):
        return self._inifile
    #@-node:Zoomq.20051028115650.25:getfilename
    #@+node:Zoomq.20051028115650.26:save
    def save(self, inifile=None, encoding=None):
        if inifile is None:
            inifile = self._inifile
            
        if isinstance(inifile, (str, unicode)):
            f = file(inifile, 'w')
        elif isinstance(inifile, file):
            f = inifile
        else:
            f = inifile
            
        if not f:
            f = sys.stdout
            
        if encoding is None:
            encoding = self._encoding
            
        f.write(self._savedict([], self._items, encoding))
        if isinstance(inifile, (str, unicode)):
            f.close()
    #@-node:Zoomq.20051028115650.26:save
    #@+node:Zoomq.20051028115650.27:_savedict
    def _savedict(self, section, values, encoding):
        if values:
            buf = []
            default = []
            for key, value in self._getorderitems(values.items()):
                if isinstance(value, dict):
                    sec = section[:]
                    sec.append(key)
                    buf.append(self._savedict(sec, value, encoding))
                else:
                    c = self._comments.get(section_delimeter.join(section + [key]), '')
                    if c:
                        lines = c.splitlines()
                        default.append('\n'.join(['%s %s' % (self._commentdelimeter, x) for x in lines]))
                        
                    default.append("%s = %s" % (key, uni_prt(value, encoding)))
            if default:
                buf.insert(0, '\n'.join(default))
                buf.insert(0, '[%s]' % section_delimeter.join(section))
                c = self._comments.get(section_delimeter.join(section), '')
                if c:
                    lines = c.splitlines()
                    buf.insert(0, '\n'.join(['%s %s' % (self._commentdelimeter, x) for x in lines]))
            return '\n'.join(buf + [''])
        else:
            return ''
    #@-node:Zoomq.20051028115650.27:_savedict
    #@+node:Zoomq.20051028115650.28:read
    def read(self, inifile=None, encoding=None):
        if inifile is None:
            inifile = self._inifile
            
        if isinstance(inifile, (str, unicode)):
            try:
                f = file(inifile, 'r')
            except:
                return  #may raise Exception is better
        elif isinstance(inifile, file):
            f = inifile
        else:
            f = inifile
            
        if not f:
            f = sys.stdin
            
        if encoding is None:
            encoding = self._encoding
            
        comments = []
        section = ''
        for line in f.readlines():
            line = line.strip()
            if not line: continue
            if line.startswith(self._commentdelimeter):
                comments.append(line[1:].lstrip())
                continue
            if line.startswith('['):    #section
                section = line[1:-1]
                #if comment then set it
                if comments:
                    self.comment(section, '\n'.join(comments))
                    comments = []
                continue
            key, value = line.split('=', 1)
            key = key.strip()
            value = process_value(value.strip(), encoding)
            if section:
                self.__setitem__(section + section_delimeter + key, value)
                #if comment then set it
                if comments:
                    self.__getitem__(section).comment(key, '\n'.join(comments))
                    comments = []
            else:
                self.__setitem__(key, value)
                #if comment then set it
                if comments:
                    self.comment(key, '\n'.join(comments))
                    comments = []
        if isinstance(inifile, (str, unicode)):
            f.close()
    #@-node:Zoomq.20051028115650.28:read
    #@+node:Zoomq.20051028115650.29:setorder
    def setorder(self, key):
        if not self._orders.has_key(key):
            self._orders[key] = self._ID
            self._ID += 1
    #@-node:Zoomq.20051028115650.29:setorder
    #@+node:Zoomq.20051028115650.30:_getorderitems
    def _getorderitems(self, values):
        s = []
        for key, value in values:
            s.append((self._orders.get(key, 99999), key, value))
        s.sort()
        return [(x, y) for z, x, y in s]
    #@-node:Zoomq.20051028115650.30:_getorderitems
    #@-others
#@-node:Zoomq.20051028115650.22:class DictIni
#@+node:Zoomq.20051028115650.31:process_value
def process_value(value, encoding=None):
    length = len(value)
    t = value
    i = 0
    r = []
    buf = []
    listflag = False
    while i < length:
        if t[i] == '"': #string quote
            buf.append(t[i])
            i += 1
            while t[i] != '"' or (t[i] == '"' and t[i-1] == '\\'):
                buf.append(t[i])
                i += 1
            buf.append(t[i])
            i += 1
        elif t[i] == ',':
            r.append(''.join(buf))
            buf = []
            i += 1
            listflag = True
        elif t[i] == 'u':
            buf.append(t[i])
            i += 1
        else:
            buf.append(t[i])
            i += 1
            while i < length and t[i] != ',':
                buf.append(t[i])
                i += 1
    if buf:
        r.append(''.join(buf))
    result = []
    for i in r:
        if i.isdigit():
            result.append(int(i))
        elif i and i.startswith('u"'):
            result.append(unicode(unescstr(i[1:]), encoding))
        else:
            b = r_float.match(i)
            if b:
                result.append(float(b.group()))
            else:
                result.append(unescstr(i))
            
    if listflag:
        return result
    elif result:
        return result[0]
    else:
        return ''
#@-node:Zoomq.20051028115650.31:process_value
#@+node:Zoomq.20051028115650.32:unescstr
def unescstr(value):
    if value.startswith('"') and value.endswith('"'):
        value = value[1:-1]
        escapechars = [("\\", "\\\\"), ("'", r"\'"), ('\"', r'\"'), ('\b', r'\b'), 
            ('\t', r"\t"), ('\r', r"\r"), ('\n', r"\n")]
        for item in escapechars:
            k, v = item
            value = value.replace(v, k)
    return value
#@-node:Zoomq.20051028115650.32:unescstr
#@+node:Zoomq.20051028115650.33:getdefaultencoding
def getdefaultencoding(encoding):
    if not encoding:
        encoding = locale.getdefaultlocale()[1]
    if not encoding:
        encoding = sys.getfilesystemencoding()
    if not encoding:
        encoding = 'utf-8'
    return encoding
#@-node:Zoomq.20051028115650.33:getdefaultencoding
#@+node:Zoomq.20051028115650.34:uni_prt
def uni_prt(a, encoding=None):
    escapechars = [("\\", "\\\\"), ("'", r"\'"), ('\"', r'\"'), ('\b', r'\b'), 
        ('\t', r"\t"), ('\r', r"\r"), ('\n', r"\n")]
    s = []
    if isinstance(a, (list, tuple)):
        for i, k in enumerate(a):
            s.append(uni_prt(k, encoding))
            s.append(',')
    elif isinstance(a, str):
        t = a
        for i in escapechars:
            t = t.replace(i[0], i[1])
        if ' ' in t or ',' in t or t.isdigit():
            s.append('"%s"' % t)
        else:
            s.append("%s" % t)
    elif isinstance(a, unicode):
        t = a
        for i in escapechars:
            t = t.replace(i[0], i[1])
        s.append('u"%s"' % t.encode(encoding))
    else:
        s.append(str(a))
    return ''.join(s)
#@-node:Zoomq.20051028115650.34:uni_prt
#@-others
        
if __name__ == '__main__':
    #@    << try >>
    #@+node:Zoomq.20051028115650.2:<< try >>
    
    d = DictIni('test.ini')
    #    d._comment = 'Test\nTest2'
    #    d.a = 'b'
    #    d['b'] = 3
    #    d.c.d = (1,2,'b asf aaa')
    #    d['s']['t'] = u'涓�浗'
    #    d['s'].a = 1
    #    d['m/m'] = 'testing'
    #    d.t.m.p = '3'
    
    print d
    #    d.setfilename('test.ini')
    #    d.t.m.comment('p', 'PTesting')
    #    print d.getfilename()
    #    print '---------------------'
    #    
    #    d.save('test.ini')
    #    
    #    a = process_value('1,abc,"aa cc",,"  ,\\"sdf",u"aaa"', 'ascii')
    #    print a
    #    print uni_prt(a, 'utf-8')
    #    
    #    t = DictIni(inifile='test.ini')
    #    print t
    #
    #    t.setfilename('test1.ini')
    #    t.save()
    
    #    t.setfilename('test2.ini')
    #    t.save()
    
    #@-node:Zoomq.20051028115650.2:<< try >>
    #@nl
#@nonl
#@-node:Zoomq.20051028115650:@thin dict4ini.py
#@-leo