diff options
| -rw-r--r-- | README.md | 11 | ||||
| -rw-r--r-- | main.py | 31 |
2 files changed, 39 insertions, 3 deletions
@@ -1,4 +1,7 @@ - [](https://gitlab.com/gabrielgio/reddit-nextcloud-importer/-/commits/main) [](https://gitlab.com/gabrielgio/reddit-nextcloud-importer/-/commits/main) + [](https://gitlab.com/gabrielgio/reddit-nextcloud-importer/-/commits/main) + [](https://gitlab.com/gabrielgio/reddit-nextcloud-importer/-/commits/main) # Reddit Nextcloud importer @@ -30,6 +33,12 @@ Nextcloud. * `NEXTCLOUD_PATH`: root folder for where all the images will be saved. +* `FROM_BEGINNING`: it will download all the files since the first availabile + saved post. + +* `LOG_LEVEL`: it will set the log level. It can chosen between: + * info + * error # Running as docker container @@ -1,6 +1,7 @@ import argparse import logging import os +from time import sleep import praw @@ -10,6 +11,15 @@ from praw.models.util import stream_generator from importer.downloader import Downloader from importer.uploader import upload_file, create_folders +levels = { + 'critical': logging.CRITICAL, + 'error': logging.ERROR, + 'warn': logging.WARNING, + 'warning': logging.WARNING, + 'info': logging.INFO, + 'debug': logging.DEBUG +} + parser = argparse.ArgumentParser(description="Monitor saved") parser.add_argument('-c', '--client-id', help="Reddit client id", @@ -35,11 +45,22 @@ parser.add_argument('-o', '--nextcloud-host', parser.add_argument('-d', '--nextcloud-path', help="Nextcloud root folder", default=os.environ.get('NEXTCLOUD_PATH', 'im')) - -logging.basicConfig(level=logging.INFO) +parser.add_argument('-f', '--from-beginning', + dest='from_beginning', + help="it will attempt to download all saved posts from the beginning.", + action='store_true', + default=os.environ.get("FROM_BEGINNING") is not None) +parser.add_argument('-l', '--log-level', + default=os.environ.get('LOG_LEVEL', 'info'), + choices=levels.keys(), + help=f'it will set log level.') if __name__ == "__main__": args = parser.parse_args() + + level = levels.get(args.log_level.lower()) + logging.basicConfig(level=level) + reddit = praw.Reddit(client_id=args.client_id, client_secret=args.client_secret, password=args.reddit_password, @@ -73,6 +94,12 @@ if __name__ == "__main__": logging.error(e) + if args.from_beginning: + logging.info(f"Downloading from the beginning") + for post in redditor.saved(limit=None): + upload(post) + sleep(60) + generator = stream_generator(redditor.saved, attribute_name="name") for post in generator: upload(post) |