Pastable version of .ini file parser

<div style="overflow:auto;border:solid 1px #ccc;background:#000;color:#F8F8F8">
          <div class="section">
            <span style="margin:0;padding:4px 6px;font-size:12px;float:right;">######iniparser.py</span>
            <pre style="float:left;margin:0 10px;border-right:0;color:#666;">1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60</pre>
            <pre class="sunburst">
<span style="color:#E28964;">import</span> sys
<span style="color:#E28964;">import</span> re

inifile <span style="color:#E28964;">=</span> sys.argv[<span style="color:#3387CC;">1</span>]
quotes <span style="color:#E28964;">=</span> [<span style="color:#65B042;"><span style="color:#65B042;">&quot;</span>'<span style="color:#65B042;">&quot;</span></span>, <span style="color:#65B042;"><span style="color:#65B042;">'</span><span style="color:#DDF2A4;">\&quot;</span><span style="color:#65B042;">'</span></span>]
f <span style="color:#E28964;">=</span> <span style="color:#9B859D;">file</span>(inifile)
d <span style="color:#E28964;">=</span> {}

<span style="color:#99CF50;">def</span> <span style="color:#89BDFF;">get_quote_char</span>(<span style="color:#3E87E3;">line</span>):
    <span style="color:#E28964;">for</span> char <span style="color:#E28964;">in</span> line:
        <span style="color:#E28964;">if</span> char <span style="color:#E28964;">in</span> quotes:
            <span style="color:#E28964;">return</span> char

<span style="color:#99CF50;">def</span> <span style="color:#89BDFF;">getkey</span>(<span style="color:#3E87E3;">line</span>):
    <span style="color:#AEAEAE;font-style:italic;"><span style="color:#AEAEAE;font-style:italic;">#</span>swallow everything up to the =</span>
    <span style="color:#E28964;">return</span> line[ : line.find(<span style="color:#65B042;"><span style="color:#65B042;">'</span>=<span style="color:#65B042;">'</span></span>) ].strip()

<span style="color:#99CF50;">def</span> <span style="color:#89BDFF;">getval</span>(<span style="color:#3E87E3;">line</span>):
    <span style="color:#AEAEAE;font-style:italic;"><span style="color:#AEAEAE;font-style:italic;">#</span>swallow everything after the =</span>
    line  <span style="color:#E28964;">=</span> line[ line.find(<span style="color:#65B042;"><span style="color:#65B042;">'</span>=<span style="color:#65B042;">'</span></span>) <span style="color:#E28964;">+</span> <span style="color:#3387CC;">1</span> : ].strip()
    q <span style="color:#E28964;">=</span> get_quote_char(line)
    startq <span style="color:#E28964;">=</span> line.find(q)

<span style="color:#AEAEAE;font-style:italic;"><span style="color:#AEAEAE;font-style:italic;">#</span>start scanning the line from the quote onwards</span>
    position <span style="color:#E28964;">=</span> <span style="color:#3387CC;">0</span>
    <span style="color:#E28964;">for</span> char <span style="color:#E28964;">in</span> line[ startq : ]:
        <span style="color:#E28964;">if</span> char <span style="color:#E28964;">not</span> <span style="color:#E28964;">in</span> quotes <span style="color:#E28964;">or</span> line[ position <span style="color:#E28964;">-</span> <span style="color:#3387CC;">1</span> ] <span style="color:#E28964;">==</span> <span style="color:#65B042;"><span style="color:#65B042;">'</span><span style="color:#DDF2A4;">\\</span><span style="color:#65B042;">'</span></span>:
            <span style="color:#E28964;">pass</span>
        <span style="color:#E28964;">else</span>:
            <span style="color:#AEAEAE;font-style:italic;"><span style="color:#AEAEAE;font-style:italic;">#</span>might hit some remote corner-case with this</span>
            <span style="color:#E28964;">if</span> position <span style="color:#E28964;">&gt;</span> <span style="color:#3387CC;">0</span>:
                <span style="color:#E28964;">return</span> line[ startq <span style="color:#E28964;">+</span> <span style="color:#3387CC;">1</span> : position ] 
            
        position<span style="color:#E28964;">+=</span><span style="color:#3387CC;">1</span>

<span style="color:#E28964;">for</span> line <span style="color:#E28964;">in</span> f:
    line <span style="color:#E28964;">=</span> line.strip()
    
    <span style="color:#AEAEAE;font-style:italic;"><span style="color:#AEAEAE;font-style:italic;">#</span>skip comments and empty lines</span>
    <span style="color:#E28964;">if</span> line.startswith(<span style="color:#65B042;"><span style="color:#65B042;">'</span>;<span style="color:#65B042;">'</span></span>) <span style="color:#E28964;">or</span> line<span style="color:#E28964;">==</span><span style="color:#65B042;"><span style="color:#65B042;">'</span><span style="color:#65B042;"><span style="color:#65B042;">'</span></span></span>: <span style="color:#E28964;">pass</span>
    
    <span style="color:#AEAEAE;font-style:italic;"><span style="color:#AEAEAE;font-style:italic;">#</span>store sections as dicts</span>
    <span style="color:#E28964;">elif</span> line.startswith(<span style="color:#65B042;"><span style="color:#65B042;">'</span>[<span style="color:#65B042;">'</span></span>):
        section_name <span style="color:#E28964;">=</span> line[ <span style="color:#3387CC;">1</span> : <span style="color:#DAD085;">len</span>(line) <span style="color:#E28964;">-</span> <span style="color:#3387CC;">1</span> ].strip()
        section_dict <span style="color:#E28964;">=</span> { section_name : {} }
        d.update(section_dict)
    <span style="color:#E28964;">else</span>:
        k <span style="color:#E28964;">=</span> getkey(line)
        v <span style="color:#E28964;">=</span> getval(line)

<span style="color:#AEAEAE;font-style:italic;"><span style="color:#AEAEAE;font-style:italic;">#</span>print k,v</span>

<span style="color:#E28964;">try</span>:         
            d[section_name].update( {k:v} )
        <span style="color:#E28964;">except</span> <span style="color:#9B859D;">TypeError</span>:
            <span style="color:#E28964;">print</span> <span style="color:#65B042;"><span style="color:#65B042;">'</span>The ini file contains invalid characters<span style="color:#65B042;">'</span></span>
            
<span style="color:#E28964;">print</span> d

</pre>
          </div>
        </div>

<div style="overflow:auto;border:solid 1px #ccc;background:#000;color:#F8F8F8">
          <div class="section">
            <span style="margin:0;padding:4px 6px;font-size:12px;float:right;">#########test.ini</span>
            <pre style="float:left;margin:0 10px;border-right:0;color:#666;">1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28</pre>
            <pre class="sunburst">

[foo]
greeting <span style="color:#E28964;">=</span> <span style="color:#65B042;"><span style="color:#65B042;">'</span>hello<span style="color:#65B042;">'</span></span>
;this <span style="color:#E28964;">is</span> a comment
name <span style="color:#E28964;">=</span> <span style="color:#65B042;"><span style="color:#65B042;">'</span>Eddie<span style="color:#65B042;">'</span></span>

[bar]
lastname <span style="color:#E28964;">=</span> <span style="color:#65B042;"><span style="color:#65B042;">'</span>Vedder<span style="color:#65B042;">'</span></span>;that<span style="color:#65B042;"><span style="color:#65B042;">'</span>s another comment;<span style="background-color:#150B15;color:#FD5FF1;"></span></span>

[ malformed section  ]
city<span style="color:#E28964;">=</span><span style="color:#65B042;"><span style="color:#65B042;">'</span>Prague<span style="color:#65B042;">'</span></span>
country<span style="color:#E28964;">=</span><span style="color:#65B042;"><span style="color:#65B042;">&quot;</span><span style="color:#DDF2A4;">\'</span>Czech Republic<span style="color:#DDF2A4;">\'</span><span style="color:#65B042;">&quot;</span></span>
whatever<span style="color:#E28964;">=</span><span style="color:#65B042;"><span style="color:#65B042;">'</span>this ; is nasty<span style="color:#65B042;">'</span></span>

[bad]
dog<span style="color:#E28964;">=</span><span style="color:#65B042;"><span style="color:#65B042;">'</span>bau<span style="color:#65B042;">'</span></span>
cat    <span style="color:#E28964;">=</span>     <span style="color:#65B042;"><span style="color:#65B042;">'</span>miao<span style="color:#65B042;">'</span></span>
mouse <span style="color:#E28964;">=</span> <span style="color:#65B042;"><span style="color:#65B042;">&quot;</span>squeak<span style="color:#65B042;">&quot;</span></span>

[tabbed section]
dogname <span style="color:#E28964;">=</span> <span style="color:#65B042;"><span style="color:#65B042;">'</span>\Oliver<span style="color:#65B042;">'</span></span>
catname <span style="color:#E28964;">=</span> <span style="color:#65B042;"><span style="color:#65B042;">'</span>Barbara<span style="color:#65B042;">'</span></span>

[one more]
appliance<span style="color:#E28964;">=</span><span style="color:#65B042;"><span style="color:#65B042;">'</span>lcd <span style="color:#DDF2A4;">\'</span>monitor<span style="color:#DDF2A4;">\'</span><span style="color:#65B042;">'</span></span>
car <span style="color:#E28964;">=</span> <span style="color:#65B042;"><span style="color:#65B042;">&quot;</span>Alfa <span style="color:#DDF2A4;">\&quot;</span>Romeo<span style="color:#DDF2A4;">\&quot;</span> - Giulietta<span style="color:#65B042;">&quot;</span></span>;<span style="color:#65B042;"><span style="color:#65B042;">&quot;</span>foo<span style="color:#65B042;">&quot;</span></span>

</pre>
          </div>
        </div>

<a href="http://refactormycode.com/codes/143-ini-file-parser" style="color:#fff" title="As seen on RefactorMyCode.com"><img alt="Small_logo" src="http://refactormycode.com/images/small_logo.gif" style="border:0" /></a>

counter free hit invisible