__main__.py

#!/usr/bin/env python2.7
#-*- encoding: utf-8 -*-

# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>

import sys
import getopt
import postprocess as ps

from scan import *
import gui

def print_help():
    print('''Usage: {0}: [OPTION] scan|text|learn file [file...]\
          \nExtract string from image file
          \nvalid argument may be:\
          \n    -h, --help: print this help message\
          \n    -d, --dir: dataset directory\
          \n    scan: scan for valid text in file'''.format(sys.argv[0]))
    sys.exit(0)

if __name__=="__main__":
    optlist, args = getopt.getopt(sys.argv[1:], "hd:", ["help", "dir"])
    direc = "dataset"
    p = ps.postprocess()
    for o, a in optlist:
        if o in ("-h", "--help"):
            print_help()
        if o in ("-d", "--dir"):
            direc = a
    if len(args) >= 2 and args[0] in ("s", "scan"):
        knn = learnLetter()
        for filename in args[1:]:
            scan(knn, filename, p)
    elif len(args) >= 2 and args[0] in ("t", "text"):
        print(direc)
        knn = learnLetter(directory=direc)
        for filename in args[1:]:
            scantext(knn, filename, p)
    elif len(args) >= 1 and args[0] in ("l", "learn"):
        if len(args) >= 2:
            knn = learnLetter(filename)
        else:
            learnLetter()
    elif len(args) >= 1 and args[0] in ("f", "format"):
        if len(args) >= 2:
            for filename in args[1:]:
                splitDataset(filename)
    else:
       gui.Gui(args, p, directory=direc)
    sys.exit(0)