reddit-image-wall-getter/reddit_imgs/runner.py

#!/usr/bin/env python3
# -*- encoding: utf-8 -*-

import reddit_imgs.sync
import reddit_imgs.fetch
import reddit_imgs.reorganize
import reddit_imgs.wallpapers
import reddit_imgs.thumbnailize
import reddit_imgs.hashit
import reddit_imgs.normalizetobmp
import reddit_imgs.cachedhash

import os
import sys
import shutil
wdir = os.path.abspath('.')

def ensureFolderAvailability():
    if not os.path.exists(os.path.join(wdir,'w')):
        os.makedirs(os.path.join(wdir,'w'))
    if not os.path.exists(os.path.join(wdir,'d')):
        os.makedirs(os.path.join(wdir,'d'))
    if not os.path.exists(os.path.join(wdir,'i')):
        os.makedirs(os.path.join(wdir,'i'))
    if not os.path.exists(os.path.join(wdir,'r')):
        os.makedirs(os.path.join(wdir,'r'))

def managesubreddits():
    i = ''
    while i!='0':
        print('\n'*100)
        print('----------------------------------------------')
        print('              Subreddit Manager               ')
        print('----------------------------------------------')
        print('1) List monitored subreddits')
        print('2) Add monitored subreddit')
        print('3) Remove monitored subreddit')
        print('4) Set as wallpaper source')
        print('5) Unset as wallpaper source')
        print()
        print('0) Back')
        print('----------------------------------------------')
        print()
        print('Enter your choice:')
        i = input()
        i = i.strip()
        print()
        print()
        subreddits_dir = os.path.join(wdir,'r')
        subreddits_isfolder = lambda sr: os.path.isdir(os.path.join(subreddits_dir,sr))
        subreddits = sorted(filter(subreddits_isfolder, os.listdir(subreddits_dir)))
        if i in ['1', '3', '4', '5']:
            print('Subreddits monitored:')
            for sr in subreddits:
                print('/r/{0}'.format(sr),end='')
                if os.path.isfile(os.path.join(subreddits_dir,sr,'wallpaper.flag')):
                    print('\t\t(wallpaper)')
                else:
                    print()
            print()
            if i=='1':
                print('Press enter to continue')
                input()
        if i=='3':
            print('Enter the subreddit you want to get rid of:')
            rem = input('/r/')
            try: shutil.rmtree(os.path.join(subreddits_dir,rem))
            except: pass
            print()
            print('Done.')
            print('Press enter to continue')
            input()
        elif i=='2':
            print('Enter the subreddit you want to add:')
            add = input('/r/')
            try: os.makedirs(os.path.join(subreddits_dir,add))
            except: pass
            print()
            print('Done.')
            print('Press enter to continue')
            input()
        elif i=='4':
            print('Enter the subreddit you want to set as wallpaper source:')
            add = input('/r/')
            try:
                dd = os.path.join(subreddits_dir,add)
                if not os.path.exists(dd):
                    os.makedirs(dd)
                f = open(os.path.join(dd, 'wallpaper.flag'),'w')
                f.write('')
                f.close()
            except: pass
            print()
            print('Done.')
            print('Press enter to continue')
            input()
        elif i=='5':
            print('Enter the subreddit you want to unset as wallpaper source:')
            add = input('/r/')
            try:
                dd = os.path.join(subreddits_dir,add)
                if not os.path.exists(dd):
                    os.makedirs(dd)
                f = open(os.path.join(dd, 'wallpaper.flag'),'w')
                f.write('')
                f.close()
                os.remove(os.path.join(dd, 'wallpaper.flag'))
            except: pass
            print()
            print('Done.')
            print('Press enter to continue')
            input()

def mainmenu():
    i = ''
    while i!='0':
        print('\n'*100)
        print('----------------------------------------------')
        print('            Reddit Image Downloader           ')
        print('----------------------------------------------')
        print('1) Manage subreddits')
        print('2) Get link list to be downloaded from reddit')
        print('3) Download grabbed links')
        print('4) Organize by hashes')
        print('5) Generate thumbnails')
        print('6) Group and put nice names on downloaded data')
        print('7) Sepparate wallpapers')
        print()
        print('0) Quit')
        print('----------------------------------------------')
        print()
        print('Enter your choice:')
        i = input()
        i = i.strip()
        if i=='1':
            managesubreddits()
        elif i=='2':
            reddit_imgs.sync.main()
        elif i=='3':
            reddit_imgs.fetch.main()
        elif i=='4':
            reddit_imgs.hashit.main()
        elif i=='5':
            reddit_imgs.thumbnailize.main()
        elif i=='6':
            reddit_imgs.reorganize.main()
        elif i=='7':
            reddit_imgs.wallpapers.main()

def main():
    ensureFolderAvailability()
    if len(sys.argv)>1:
        cmdline()
    else:
        mainmenu()

def cmdline():
    cmds = sys.argv[1:]
    available_commands = ((
        ('sync', reddit_imgs.sync.main),
        ('fetch', reddit_imgs.fetch.main),
        ('fetchretryingemptyalbuns', reddit_imgs.fetch.retry),
        ('cachedhash', reddit_imgs.cachedhash.main),
        ('hashit', reddit_imgs.hashit.main),
        ('normalizetobmp', reddit_imgs.normalizetobmp.main),
        ('thumbnailize', reddit_imgs.thumbnailize.main),
        ('reorganize', reddit_imgs.reorganize.main),
        ('wallpapers', reddit_imgs.wallpapers.main),
    ))
    available_commands_names = tuple(list(map(lambda a: a[0], available_commands)))
    for cmd in cmds:
        if cmd not in available_commands_names:
            print('Usage {0} [{1}]'.format(sys.argv[0], '/'.join(available_commands_names)))
            return
    for cmd in cmds:
        command_ran = False
        for acmd in available_commands:
            if cmd == acmd[0]:
                command_ran = True
                acmd[1]()
        if not command_ran:
            print('Usage {0} [{1}]'.format(sys.argv[0], '/'.join(available_commands_names)))

if __name__ == '__main__':
    main()