counter.py

import sys
import traceback

from domain.document import Document
from domain.errors import ConfigError
from interface.walker import Walker

if __name__ == "__main__":
    try:
        if len(sys.argv) < 2:
            raise ConfigError(
                "Usage: py counter.py <path> [<filename>]\n\n"
                "Omitting <filename> will scan all files in the path."
            )
        path = str(sys.argv[1])

        filename = None
        if len(sys.argv) > 2:
            filename = str(sys.argv[2])

        if filename:  # single file
            print("Single file mode initiated for", filename, "at", path)
            docs = [Document(filename, path)]
        else:  # all files under the path
            print("Multiple file mode initiated for path", path)
            walker = Walker(path)
            docs = walker.walk()
            print("Files found: ", len(docs))

        print("Scan completed, starting to process...")
        for doc in docs:
            # todo: Database retrieval for comparison
            words = doc.words
            print(doc.file_name + ":", str(words), "word" + ("" if words == 1 else "s"))

        # todo Actual saving to database
        print("Data saved")
    except ConfigError as e:
        print(str(e))
        sys.exit(1)
    except Exception as e:
        print(str(e))
        traceback.print_exc()  # todo Configurable error output?
        sys.exit(1)