Shreddit/shreddit/app.py

"""This module contains script entrypoints for shreddit.
"""
import argparse
import yaml
import logging
from shreddit.oauth import oauth_test
from shreddit.shredder import Shredder


def main():
    parser = argparse.ArgumentParser(description="Command-line frontend to the shreddit library.")
    parser.add_argument("-c", "--config", help="Config file to use instead of the default shreddit.cfg")
    parser.add_argument("-p", "--praw", help="PRAW config (if not ./praw.ini)")
    parser.add_argument("-t", "--test-oauth", help="Perform OAuth test and exit", action="store_true")
    args = parser.parse_args()

    if args.test_oauth:
        oauth_test(args.praw)
        return

    with open(args.config or "shreddit.yml") as fh:
        config = yaml.safe_load(fh)
    if not config:
        raise Exception("No config options passed!")

    shredder = Shredder(config, args.praw)
    shredder.shred()


if __name__ == "__main__":
    main()