af1f9fd365
* IMGUR API is no longer used * --skip now accepts file types instead of domain * --skip-domain added * --no-download added * --no-dupe now supports YouTube * Duplicates of older posts will not be dowloaded if --no-dupe and --downloaded-posts options are given together * Invalid characters in MacOS and Linux platforms are removed from filenames * Bug fixes
132 lines
4 KiB
Python
132 lines
4 KiB
Python
import os
|
|
import socket
|
|
import webbrowser
|
|
import random
|
|
|
|
from src.reddit import Reddit
|
|
from src.jsonHelper import JsonFile
|
|
from src.utils import nameCorrector
|
|
|
|
class Config():
|
|
|
|
def __init__(self,filename):
|
|
self.filename = filename
|
|
self.file = JsonFile(self.filename)
|
|
|
|
def generate(self):
|
|
self._validateCredentials()
|
|
self._readCustomFileName()
|
|
self._readCustomFolderPath()
|
|
self._readDefaultOptions()
|
|
return self.file.read()
|
|
|
|
def setCustomFileName(self):
|
|
print("""
|
|
IMPORTANT: Do not change the filename structure frequently.
|
|
If you did, the program could not find duplicates and
|
|
would download the already downloaded files again.
|
|
This would not create any duplicates in the directory but
|
|
the program would not be as snappy as it should be.
|
|
|
|
Type a template file name for each post.
|
|
|
|
You can use SUBREDDIT, REDDITOR, POSTID, TITLE, UPVOTES, FLAIR, DATE in curly braces
|
|
The text in curly braces will be replaced with the corresponding property of an each post
|
|
|
|
For example: {FLAIR}_{SUBREDDIT}_{REDDITOR}
|
|
|
|
Existing filename template:""", None if "filename" not in self.file.read() else self.file.read()["filename"])
|
|
|
|
filename = nameCorrector(input(">> ").upper())
|
|
self.file.add({
|
|
"filename": filename
|
|
})
|
|
|
|
def _readCustomFileName(self):
|
|
content = self.file.read()
|
|
|
|
if not "filename" in content:
|
|
self.file.add({
|
|
"filename": "{REDDITOR}_{TITLE}_{POSTID}"
|
|
})
|
|
content = self.file.read()
|
|
|
|
if not "{POSTID}" in content["filename"]:
|
|
self.file.add({
|
|
"filename": content["filename"] + "_{POSTID}"
|
|
})
|
|
|
|
def setCustomFolderPath(self):
|
|
print("""
|
|
Type a folder structure (generic folder path)
|
|
|
|
Use slash or DOUBLE backslash to separate folders
|
|
|
|
You can use SUBREDDIT, REDDITOR, POSTID, TITLE, UPVOTES, FLAIR, DATE in curly braces
|
|
The text in curly braces will be replaced with the corresponding property of an each post
|
|
|
|
For example: {REDDITOR}/{SUBREDDIT}/{FLAIR}
|
|
|
|
Existing folder structure""", None if "folderpath" not in self.file.read() else self.file.read()["folderpath"])
|
|
|
|
folderpath = nameCorrector(input(">> ").strip("\\").strip("/").upper())
|
|
|
|
self.file.add({
|
|
"folderpath": folderpath
|
|
})
|
|
|
|
def _readCustomFolderPath(self,path=None):
|
|
content = self.file.read()
|
|
if not "folderpath" in content:
|
|
self.file.add({
|
|
"folderpath": "{SUBREDDIT}"
|
|
})
|
|
|
|
def setDefaultOptions(self):
|
|
print("""
|
|
Type options to be used everytime script runs
|
|
|
|
For example: --no-dupes --quit --limit 100 --skip youtube.com
|
|
|
|
Existing default options:""", None if "options" not in self.file.read() else self.file.read()["options"])
|
|
|
|
options = input(">> ").strip("")
|
|
|
|
self.file.add({
|
|
"options": options
|
|
})
|
|
|
|
def _readDefaultOptions(self,path=None):
|
|
content = self.file.read()
|
|
if not "options" in content:
|
|
self.file.add({
|
|
"options": ""
|
|
})
|
|
|
|
def _validateCredentials(self):
|
|
"""Read credentials from config.json file"""
|
|
|
|
try:
|
|
content = self.file.read()["credentials"]
|
|
except:
|
|
self.file.add({
|
|
"credentials":{}
|
|
})
|
|
content = self.file.read()["credentials"]
|
|
|
|
if "reddit" in content and len(content["reddit"]) != 0:
|
|
pass
|
|
else:
|
|
Reddit().begin()
|
|
|
|
print()
|
|
|
|
def setDefaultDirectory(self):
|
|
print("""Set a default directory to use in case no directory is given
|
|
Leave blank to reset it. You can use {time} in foler names to use to timestamp it
|
|
For example: D:/archive/BDFR_{time}
|
|
""")
|
|
print("Current default directory:", self.file.read()["default_directory"] if "default_directory" in self.file.read() else "")
|
|
self.file.add({
|
|
"default_directory": input(">> ")
|
|
}) |