188 lines
6.3 KiB
Python
Executable File
188 lines
6.3 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# -*- encoding: utf-8 -*-
|
|
|
|
import reddit_imgs.sync
|
|
import reddit_imgs.fetch
|
|
import reddit_imgs.fetch2
|
|
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.fetch2.main),
|
|
('fetch_old', 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()
|