Wifi Recon

parse_wificollections.py

import sys


def parse_column(line):
    cols = line.split('\t')
    col_count = len(cols)

    if col_count == 13:
        return cols
    elif col_count == 12:
        return [None] + cols
    else:
        raise ValueError('Invalid column count {}'.format(col_count))


def parse(fin):
    """
    :param fin: File input stream
    """
    for line in fin:
        print(parse_column(line))


if __name__ == '__main__':
    parse(sys.stdin)