Skip to content

Commit

Permalink
Merge pull request #4 from DuckBoss/DuckBoss-patch-1
Browse files Browse the repository at this point in the history
HOTFIX : Added missing script.
  • Loading branch information
DuckBoss authored Jun 25, 2018
2 parents b0d2dea + 7766e03 commit 2705e24
Showing 1 changed file with 33 additions and 0 deletions.
33 changes: 33 additions & 0 deletions search_setup.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,33 @@
import configparser

class ConfigReader:
search_terms = [""]
search_subreddit = ""
output_directory = ""
file_extensions = [".jpg, .png, .gif"]
url_types = [""]
search_limit = 0
allow_nsfw = False
allow_stream = False

def __init__(self):
print("Config Reader Module Loaded!")

def read_config(self):
cp = configparser.ConfigParser()
cp.read('config.ini')

# Main Config
self.search_terms = cp['Main']['search_terms'].replace(" ", "").split(',')
self.search_subreddit = cp['Main']['search_subreddit'].split()[0]
self.output_directory = cp['Main']['output_directory'].split()[0]

# File Extensions/URL Types
self.file_extensions = cp['File Extensions']['file_extensions'].replace(" ", "").split(',')
self.url_types = cp['URL Types']['url_types'].replace(" ", "").split(',')

# Optional
self.search_limit = int(cp['Optional']['search_limit'])

self.allow_nsfw = cp.getboolean('Optional', 'allow_nsfw')
self.allow_stream = cp.getboolean('Optional', 'allow_stream')

0 comments on commit 2705e24

Please sign in to comment.