main.py

# Copyright (C) 2018  Ahmad A. A. (https://github.com/bbpgrs/)

import common as cmn
import files
import file_lists
import os
import logging
import time
import sys

"""
main module, program will be run through this module
"""


def main():
    """
    Run program
    """
    cmn.make_dir(cmn.LOG_DIR)
    logging.basicConfig(
        # filename=os.path.join(cmn.LOG_DIR, cmn.LOG_FILE_NAME % time.strftime(cmn.LOG_FILE_NAME_TIME)),
        handlers=[
            logging.FileHandler(os.path.join(cmn.LOG_DIR, cmn.LOG_FILE_NAME % time.strftime(cmn.LOG_FILE_NAME_TIME))),
            logging.StreamHandler()
        ],
        level=cmn.LOG_LEVEL,
        format=cmn.LOG_MSG_FORMAT,
        datefmt=cmn.LOG_TIME_FORMAT
    )

    start_time = time.time()
    logging.info("Starting program ...")

    if "-man" in sys.argv:
        logging.info("Detected 'manifest only' command line argument, program will only generate file manifests.\n")
        file_lists.run()
    elif "-dlo" in sys.argv:
        logging.info("Detected 'download only' command line argument, program will use existing manifest lists.\n")
        files.run()
    else:
        logging.warning("No valid command line arguments, program will ignore arguments.\n")
        file_lists.run()
        files.run()

    logging.info("Finished running program.")
    run_time = time.time() - start_time
    logging.info("Total run time: %s.\n" % time.strftime('%H:%M:%S', time.gmtime(run_time)))


if __name__ == "__main__":
    main()