Allows user to generate praw and shreddit configs

item-del-count
Scott 8 years ago
parent 0706cbd6d9
commit e889c234e8

@ -25,7 +25,7 @@ setup(
"Programming Language :: Python"],
packages=["shreddit"],
install_requires=["arrow", "backports-abc", "praw>=4", "prawcore", "PyYAML", "requests", "six", "tornado"],
package_data={"shreddit": ["shreddit.yml.example"]},
package_data={"shreddit": ["*.example"]},
entry_points={
"console_scripts": [
"shreddit=shreddit.app:main"

@ -12,14 +12,28 @@ from shreddit.shredder import Shredder
def main():
parser = argparse.ArgumentParser(description="Command-line frontend to the shreddit library.")
parser.add_argument("-c", "--config", help="Config file to use instead of the default shreddit.yml")
parser.add_argument("-g", "--generate-configs", help="Write shreddit and praw config files to current directory.",
action="store_true")
parser.add_argument("-u", "--user", help="User section from praw.ini if not default", default="default")
args = parser.parse_args()
if args.generate_configs:
if not os.path.isfile("shreddit.yml"):
print("Writing shreddit.yml file...")
with open("shreddit.yml", "w") as fout:
fout.write(pkg_resources.resource_string("shreddit", "shreddit.yml.example"))
if not os.path.isfile("praw.ini"):
print("Writing praw.ini file...")
with open("praw.ini", "w") as fout:
fout.write(pkg_resources.resource_string("shreddit", "praw.ini.example"))
return
config_filename = args.config or "shreddit.yml"
if not os.path.isfile(config_filename):
print("No configuration file could be found. Paste the following into a file called \"shreddit.yml\" and " \
"try running shreddit again:\n\n")
print(pkg_resources.resource_string("shreddit", "shreddit.yml.example"))
print("No shreddit configuration file was found or provided. Run this script with -g to generate one.")
return
if not os.path.isfile("praw.ini"):
print("No praw configuration file was found. Run this script with -g to generate one.")
return
with open(config_filename) as fh:

@ -0,0 +1,7 @@
# Credentials go here. Fill out default, or provide one or more names and call shreddit with the -u option to specify
# which set to use.
[default]
client_id=
client_secret=
username=
password=
Loading…
Cancel
Save