"""
Awk like module

Usage:

    for al in awk_iter_open("some filename"):
        print al._nr   # AWK NR
        print al._nf   # AWK NF
        print al._0r  # AWK $0
        print al._1   # AWK strip($1)
        print al._1r  # AWK $1 , raw string
        print al._1i  # AWK int($1)
        print al._1f  # AWK float($1)
"""

import re
def match(s, pattern):
    return re.match(pattern, s)

class AwkLine(object):
    def __init__(self, line, linenumber=0, sep = None, *args, **kw):
        self.line = line.rstrip()
        self.fields = line.split(sep)
        self.nr = linenumber
        super(AwkLine, self).__init__(*args, **kw)
    
    def __getattr__(self, name):
        if name in ["nr,", "_nr", "linenumber"]:
            return self.nr
            
        if name in [ "_nf", "_n", "n", "nf"]:
            return len(self.fields)
            
        if not name.startswith("_"):
            raise AttributeError  
            
        if name == "_0":
            return self.line.strip()
        if name == "_0r":
            return self.line
        
        if name.endswith("i"):
            return int(self.__getattr__(name[:-1]))
        if name.endswith("f"):
            return float(self.__getattr__(name[:-1]))
        if name.endswith("r"): #1r, 2r, 3r
            try:
                index = int(name[1:-1])
                return self.fields[index-1]
            except:
                raise AttributeError
        else: # 1, 2, 3
            try:
                index = int(name[1:])
                return self.fields[index-1].strip()
            except:
                raise AttributeError

def awk_iter(thefile, sep = None):
    for i, line in enumerate(thefile):
        yield AwkLine(line, i, sep)

def awk_iter_open(filename, sep=None):
    thefile = open(filename, "rt")
    for i, line in enumerate(thefile):
        yield AwkLine(line, i, sep)
    thefile.close()
#
if __name__ == "__main__":
    x =  AwkLine("hello, this is a test for int 100 and float 3.3") 
    print x._nr
    print x._nf
    print x._0
    print x._1
    print x._8i
    print x._11f
    for al in awk_iter_open("awk.py"): # this file itself
        print al.nr, al._0r